music-kraken/music-kraken-core
3
2
Fork 0
Code Issues 10 Pull Requests 1 Packages Projects 2 Releases Wiki Activity

fixed little bug that had to do with youtube putting feature artists in the track name

Browse Source
This commit is contained in:
lars 2022-11-13 16:40:44 +01:00
parent da0bb12190
commit 265e72abad
1 changed files with 18 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
src/utils/phonetic_compares.py
View File

@ -11,8 +11,25 @@ def unify_punctuation(to_unify: str) -> str:
return to_unify return to_unify
def remove_feature_part_from_track(title: str) -> str:
if ")" != title[-1]:
return title
if "(" not in title:
return title
return title[:title.index("(")]
def modify_title(to_modify: str) -> str:
to_modify = to_modify.strip()
to_modify = to_modify.lower()
to_modify = remove_feature_part_from_track(to_modify)
to_modify = unify_punctuation(to_modify)
return to_modify
def match_titles(title_1: str, title_2: str) -> (bool, int): def match_titles(title_1: str, title_2: str) -> (bool, int):
title_1, title_2 = unify_punctuation(title_1).lower(), unify_punctuation(title_2).lower() title_1, title_2 = modify_title(title_1), modify_title(title_2)
distance = jellyfish.levenshtein_distance(title_1, title_2) distance = jellyfish.levenshtein_distance(title_1, title_2)
return distance > TITLE_THRESHOLD_LEVENSHTEIN, distance return distance > TITLE_THRESHOLD_LEVENSHTEIN, distance