fix: wrong songs downloaded when dots in song name

lfavole · Jan 9, 2025 · 2a92dda · 2a92dda
1 parent 6decb68
commit 2a92dda
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/songs_dl/utils.py b/songs_dl/utils.py
@@ -594,7 +594,9 @@ def order_results(provider: str, best_items: list[Song], results: dict[str, list
     ret: list[tuple[Song, float, list[float]]] = []
 
     def normalize_title(title: str):
-        return re.sub(r"\bst(e?s?)\.?(\s+|$)", r"saint\1\2", title)
+        title = re.sub(r"\.\s*(?=\w\W|\w$|$)", "", title)
+        title = re.sub(r"\bst(e?s?)(\s+|$)", r"saint\1\2", title)
+        return title
 
     for result in results[provider]:
         # check for common word