o 㹏iW@sdZddlZddlZddlmZmZmZmZmZm Z ddl m Z m Z ddl mZdeeege fdedeeeeeeffd ed eeeeff d d Z  ddeded ededeeege fd eeeeff ddZdS)zNModule contains the async interface to match needle against haystack in batch.N)AnyCallableDictListUnioncast) SCORE_INDICES fzy_scorer) HAYSTACKSscorerneedle haystackskeyreturncsZg}|D]}||tt||\}}|durq||||dq|jdddd|S)aCalculate the score for needle against given list of haystacks and rank them. Args: scorer: Scorer to be used to do the calculation. needle: Substring to search. haystacks: List of dictionary containing the haystack to be searched. key: The key within the `haystacks` dictionary that contains the actual string value. Return: Sorted list of haystacks based on the needle score with additional keys for the `score` and `indices`. N)scoreindiceshaystackcS|dSNrxrrG/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/pfzy/match.py(z_rank_task..Trreverse)rrappendsort)r r r rresultrrrrrr _rank_task sr  batch_sizec sdurttD]\}}t|tssd|i|<q s*tdtjdtjfddt dt DIdH}t j |dd d d }g} |D]} | d | d d <| | d qR| S)aFuzzy find the needle within list of haystacks and get matched results with matching index. Note: The `key` argument is optional when the provided `haystacks` argument is a list of :class:`str`. It will be given a default key `value` if not present. Warning: The `key` argument is required when provided `haystacks` argument is a list of :class:`dict`. If not present, :class:`TypeError` will be raised. Args: needle: String to search within the `haystacks`. haystacks: List of haystack/longer strings to be searched. key: If `haystacks` is a list of dictionary, provide the key that can obtain the haystack value to search. batch_size: Number of entry to be processed together. scorer (Callable[[str, str], SCORE_indices]): Desired scorer to use. Currently only :func:`~pfzy.score.fzy_scorer` and :func:`~pfzy.score.substr_scorer` is supported. Raises: TypeError: When the argument `haystacks` is :class:`list` of :class:`dict` and the `key` argument is missing, :class:`TypeError` will be raised. Returns: List of matching `haystacks` with additional key indices and score. Examples: >>> import asyncio >>> asyncio.run(fuzzy_match("ab", ["acb", "acbabc"])) [{'value': 'acbabc', 'indices': [3, 4]}, {'value': 'acb', 'indices': [0, 2]}] Nvalue$z\ missing 1 required argument: 'key', 'key' is required when haystacks is an instance of dictc3s*|]}t||VqdS)N)r ).0offsetr#r rr r rr `s zfuzzy_match..rcSrrrrrrrrjrzfuzzy_match..Trrr)r enumerate isinstancedict TypeError fuzzy_match__name__asynciogatherrangelenheapqmerger) r r rr#r indexrbatchesresultschoices candidaterr(rr.,s0%     r.)r!r"N)__doc__r0r4typingrrrrrr pfzy.scorerr pfzy.typesr strr intr.rrrrsB   %