From 236a19a8eff36ad98016298694261f730357dec7 Mon Sep 17 00:00:00 2001 From: lexitronic Date: Sun, 17 Aug 2025 04:19:36 -0400 Subject: [PATCH] Use rapidfuzz to compare strings --- deezspot/deezloader/__init__.py | 4 ++-- setup.py | 3 ++- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/deezspot/deezloader/__init__.py b/deezspot/deezloader/__init__.py index 98e0bad..44ccd6c 100644 --- a/deezspot/deezloader/__init__.py +++ b/deezspot/deezloader/__init__.py @@ -64,14 +64,14 @@ from deezspot.models.callback.album import albumObject as albumCbObject from deezspot.models.callback.playlist import playlistObject as playlistCbObject from deezspot.models.callback.common import IDs from deezspot.models.callback.user import userObject - +from rapidfuzz import fuzz def _sim(a: str, b: str) -> float: a = (a or '').strip().lower() b = (b or '').strip().lower() if not a or not b: return 0.0 - return SequenceMatcher(None, a, b).ratio() + return fuzz.partial_ratio(a, b) / 100 # Clean for searching on Deezer def _remove_parentheses(string: str) -> str: diff --git a/setup.py b/setup.py index c1a039b..a85ef95 100644 --- a/setup.py +++ b/setup.py @@ -27,6 +27,7 @@ setup( "fastapi==0.116.1", "uvicorn[standard]==0.35.0", "spotipy-anon==1.5.2", - "librespot-spotizerr==0.3.0" + "librespot-spotizerr==0.3.0", + "rapidfuzz" ], )