music-kraken-core/src/url_to_path.py

import os.path
import json

from metadata import database


class UrlPath:
    def __init__(self, genre: str, temp: str = "temp", file: str = ".cache3.csv", step_two_file: str = ".cache2.csv"):
        self.temp = temp
        self.file = file

        self.genre = genre

        for row in database.get_tracks_without_filepath():
            file, path = self.get_path_from_row(row)
            database.set_filepath(row['id'], file, path, genre)


    def get_path_from_row(self, row):
        """
        genre/artist/song.mp3

        :param row:
        :return: path:
        """
        return os.path.join(self.get_genre(), self.get_artist(row), self.get_album(row), f"{self.get_song(row)}.mp3"), os.path.join(self.get_genre(), self.get_artist(row), self.get_album(row))

    def escape_part(self, part: str):
        return part.replace("/", " ")

    def get_genre(self):
        return self.escape_part(self.genre)

    def get_album(self, row):
        return self.escape_part(row['album'])

    def get_artist(self, row):
        artists = [artist['name'] for artist in row['artists']]
        return self.escape_part(artists[0])

    def get_song(self, row):
        return self.escape_part(row['title'])


if __name__ == "__main__":
    UrlPath("dsbm")