94 lines
3.5 KiB
Python
94 lines
3.5 KiB
Python
# SPDX-License-Identifier: MIT
|
|
# Copyright (C) 2025 Max Bachmann
|
|
# This file is generated by tools/generate_python.py
|
|
from __future__ import annotations
|
|
|
|
import contextlib
|
|
import os
|
|
|
|
from rapidfuzz._feature_detector import AVX2, SSE2, supports
|
|
|
|
__all__ = ["distance", "normalized_distance", "normalized_similarity", "similarity"]
|
|
|
|
_impl = os.environ.get("RAPIDFUZZ_IMPLEMENTATION")
|
|
if _impl == "cpp":
|
|
imported = False
|
|
if supports(AVX2):
|
|
with contextlib.suppress(ImportError):
|
|
from rapidfuzz.distance.metrics_cpp_avx2 import ( # pyright: ignore[reportMissingImports]
|
|
osa_distance as distance,
|
|
osa_normalized_distance as normalized_distance,
|
|
osa_normalized_similarity as normalized_similarity,
|
|
osa_similarity as similarity,
|
|
)
|
|
|
|
imported = True
|
|
|
|
if not imported and supports(SSE2):
|
|
with contextlib.suppress(ImportError):
|
|
from rapidfuzz.distance.metrics_cpp_sse2 import ( # pyright: ignore[reportMissingImports]
|
|
osa_distance as distance,
|
|
osa_normalized_distance as normalized_distance,
|
|
osa_normalized_similarity as normalized_similarity,
|
|
osa_similarity as similarity,
|
|
)
|
|
|
|
imported = True
|
|
|
|
if not imported:
|
|
from rapidfuzz.distance.metrics_cpp import ( # pyright: ignore[reportMissingImports]
|
|
osa_distance as distance,
|
|
osa_normalized_distance as normalized_distance,
|
|
osa_normalized_similarity as normalized_similarity,
|
|
osa_similarity as similarity,
|
|
)
|
|
elif _impl == "python":
|
|
from rapidfuzz.distance.metrics_py import (
|
|
osa_distance as distance,
|
|
osa_normalized_distance as normalized_distance,
|
|
osa_normalized_similarity as normalized_similarity,
|
|
osa_similarity as similarity,
|
|
)
|
|
else:
|
|
imported = False
|
|
if supports(AVX2):
|
|
with contextlib.suppress(ImportError):
|
|
from rapidfuzz.distance.metrics_cpp_avx2 import ( # pyright: ignore[reportMissingImports]
|
|
osa_distance as distance,
|
|
osa_normalized_distance as normalized_distance,
|
|
osa_normalized_similarity as normalized_similarity,
|
|
osa_similarity as similarity,
|
|
)
|
|
|
|
imported = True
|
|
|
|
if not imported and supports(SSE2):
|
|
with contextlib.suppress(ImportError):
|
|
from rapidfuzz.distance.metrics_cpp_sse2 import ( # pyright: ignore[reportMissingImports]
|
|
osa_distance as distance,
|
|
osa_normalized_distance as normalized_distance,
|
|
osa_normalized_similarity as normalized_similarity,
|
|
osa_similarity as similarity,
|
|
)
|
|
|
|
imported = True
|
|
|
|
if not imported:
|
|
with contextlib.suppress(ImportError):
|
|
from rapidfuzz.distance.metrics_cpp import ( # pyright: ignore[reportMissingImports]
|
|
osa_distance as distance,
|
|
osa_normalized_distance as normalized_distance,
|
|
osa_normalized_similarity as normalized_similarity,
|
|
osa_similarity as similarity,
|
|
)
|
|
|
|
imported = True
|
|
|
|
if not imported:
|
|
from rapidfuzz.distance.metrics_py import (
|
|
osa_distance as distance,
|
|
osa_normalized_distance as normalized_distance,
|
|
osa_normalized_similarity as normalized_similarity,
|
|
osa_similarity as similarity,
|
|
)
|