music-kraken-core/src/url_to_path.py

import os.path
import logging


class UrlPath:
    def __init__(self, database, logger: logging.Logger, genre: str):
        self.database = database
        self.logger = logger

        self.genre = genre

        for row in self.database.get_tracks_without_filepath():
            file, path = self.get_path_from_row(row)
            self.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")