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

This commit is contained in:
lars 2022-11-13 16:40:44 +01:00
parent da0bb12190
commit 265e72abad

View File

@ -11,8 +11,25 @@ def unify_punctuation(to_unify: str) -> str:
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):
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)
return distance > TITLE_THRESHOLD_LEVENSHTEIN, distance