Function enumerator

aura/typos.py:191–241

Iterate over the list of package pairs as generated by `generator`. These package pairs are usually combinations/product of a list of packages A given `method` is then applied to package pair that acts as a filter, usually a levenshtein distance or similar metric

(
        generator: Generator[Tuple[str, str], None, None],
        method: Callable[[str, str], int],
        extended: bool = False,
)

def enumerator(
        generator: Generator[Tuple[str, str], None, None],
        method: Callable[[str, str], int],
        extended: bool = False,
) -> Generator[Tuple[str, str], None, None]:
    """
    Iterate over the list of package pairs as generated by `generator`.
    These package pairs are usually combinations/product of a list of packages
    A given `method` is then applied to package pair that acts as a filter, usually a levenshtein distance or similar metric
    """

    pkg_cache = {}
    pkg_score_cache = {}

    for (orig, typo) in generator:
        res = method(orig, typo)
        if res and res < 2:
            if orig not in pkg_cache:
                orig_pkg = package.PypiPackage.from_cached(orig)
                pkg_cache[orig] = orig_pkg
            else:
                orig_pkg = pkg_cache[orig]

            if orig not in pkg_score_cache:
                orig_score = package.PackageScore(orig, orig_pkg, fetch_github=extended)
                pkg_score_cache[orig] = orig_score
            else:
                orig_score = pkg_score_cache[orig]

            if typo not in pkg_cache:
                typo_pkg = package.PypiPackage.from_cached(typo)
                pkg_cache[typo] = typo_pkg
            else:
                typo_pkg = pkg_cache[typo]

            if typo not in pkg_score_cache:
                typo_score = package.PackageScore(typo, typo_pkg, fetch_github=extended)
                pkg_score_cache[typo] = typo_score
            else:
                typo_score = pkg_score_cache[typo]

            data = {
                "original": orig,
                "typo": typo,
                "orig_pkg": orig_pkg,
                "orig_score": orig_score,
                "typo_pkg": typo_pkg,
                "typo_score": typo_score,
            }

            yield data


def check_name(name: str, full_list: bool=False, download_threshold: Optional[int]=None) -> List[str]: