restructured slightly and addet skript to upload to PyPI

src/music_kraken/utils/__init__.py

@@ -0,0 +1,2 @@
+# tells what exists
+__all__ = ["shared", "object_handeling", "phonetic_compares"]

src/music_kraken/utils/object_handeling.py

@@ -0,0 +1,24 @@
+from datetime import date
+
+
+def get_elem_from_obj(current_object, keys: list, after_process=lambda x: x, return_if_none=None):
+    current_object = current_object
+    for key in keys:
+        if key in current_object or (type(key) == int and key < len(current_object)):
+            current_object = current_object[key]
+        else:
+            return return_if_none
+    return after_process(current_object)
+
+
+def parse_music_brainz_date(mb_date: str) -> date:
+    year = 1
+    month = 1
+    day = 1
+
+    first_release_date = mb_date
+    if first_release_date.count("-") == 2:
+        year, month, day = [int(i) for i in first_release_date.split("-")]
+    elif first_release_date.count("-") == 0 and first_release_date.isdigit():
+        year = int(first_release_date)
+    return date(year, month, day)

src/music_kraken/utils/phonetic_compares.py

@@ -0,0 +1,48 @@
+import jellyfish
+import string
+
+TITLE_THRESHOLD_LEVENSHTEIN = 2
+UNIFY_TO = " "
+
+
+def unify_punctuation(to_unify: str) -> str:
+    for char in string.punctuation:
+        to_unify = to_unify.replace(char, UNIFY_TO)
+    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):
+    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
+
+
+def match_artists(artist_1, artist_2: str):
+    if type(artist_1) == list:
+        distances = []
+
+        for artist_1_ in artist_1:
+            match, distance = match_titles(artist_1_, artist_2)
+            if not match:
+                return match, distance
+
+            distances.append(distance)
+        return True, min(distances)
+    return match_titles(artist_1, artist_2)

src/music_kraken/utils/shared.py

@@ -0,0 +1,45 @@
+import musicbrainzngs
+import logging
+import tempfile
+import os
+
+from ..metadata.database import Database
+
+TEMP_FOLDER = "music-downloader"
+LOG_FILE = "download_logs.log"
+DATABASE_FILE = "metadata.db"
+DATABASE_STRUCTURE_FILE = "database_structure.sql"
+DATABASE_STRUCTURE_FALLBACK = "https://raw.githubusercontent.com/HeIIow2/music-downloader/master/assets/database_structure.sql"
+
+SEARCH_LOGGER = logging.getLogger("mb-cli")
+DATABASE_LOGGER = logging.getLogger("database")
+METADATA_DOWNLOAD_LOGGER = logging.getLogger("metadata-download")
+URL_DOWNLOAD_LOGGER = logging.getLogger("ling-download")
+PATH_LOGGER = logging.getLogger("create-paths")
+DOWNLOAD_LOGGER = logging.getLogger("download")
+LYRICS_LOGGER = logging.getLogger("lyrics")
+GENIUS_LOGGER = logging.getLogger("genius")
+
+NOT_A_GENRE = ".", "..", "misc_scripts", "Music", "script", ".git", ".idea"
+MUSIC_DIR = os.path.expanduser('~/Music')
+
+temp_dir = os.path.join(tempfile.gettempdir(), TEMP_FOLDER)
+if not os.path.exists(temp_dir):
+    os.mkdir(temp_dir)
+
+logging.getLogger("musicbrainzngs").setLevel(logging.WARNING)
+musicbrainzngs.set_useragent("metadata receiver", "0.1", "https://github.com/HeIIow2/music-downloader")
+
+
+database = Database(os.path.join(temp_dir, DATABASE_FILE),
+                    os.path.join(temp_dir, DATABASE_STRUCTURE_FILE),
+                    DATABASE_STRUCTURE_FALLBACK,
+                    DATABASE_LOGGER,
+                    reset_anyways=False)
+
+
+TOR = False
+proxies = {
+    'http': 'socks5h://127.0.0.1:9150',
+    'https': 'socks5h://127.0.0.1:9150'
+} if TOR else {}