diff --git a/src/actual_donwload.py b/src/actual_donwload.py index 8d446be..5a0ce7f 100644 --- a/src/actual_donwload.py +++ b/src/actual_donwload.py @@ -52,7 +52,7 @@ def real_download(): if __name__ == "__main__": music_kraken.cli(genre="test", command_list=[ - "#a Ghost Bath", - "1", + "#a Molchat Doma", + "0", "ok" ]) diff --git a/src/music_kraken/objects/source.py b/src/music_kraken/objects/source.py index 901a623..25b17ad 100644 --- a/src/music_kraken/objects/source.py +++ b/src/music_kraken/objects/source.py @@ -3,55 +3,12 @@ from enum import Enum from typing import List, Dict, Tuple, Optional from urllib.parse import urlparse +from ..utils.enums.source import SourcePages, SourceTypes from .metadata import Mapping, Metadata from .parents import DatabaseObject from .collection import Collection -class SourceTypes(Enum): - SONG = "song" - ALBUM = "album" - ARTIST = "artist" - LYRICS = "lyrics" - - -class SourcePages(Enum): - YOUTUBE = "youtube" - MUSIFY = "musify" - GENIUS = "genius" - MUSICBRAINZ = "musicbrainz" - ENCYCLOPAEDIA_METALLUM = "encyclopaedia metallum" - BANDCAMP = "bandcamp" - DEEZER = "deezer" - SPOTIFY = "spotify" - - # This has nothing to do with audio, but bands can be here - WIKIPEDIA = "wikipedia" - INSTAGRAM = "instagram" - FACEBOOK = "facebook" - TWITTER = "twitter" # I will use nitter though lol - MYSPACE = "myspace" # Yes somehow this ancient site is linked EVERYWHERE - - @classmethod - def get_homepage(cls, attribute) -> str: - homepage_map = { - cls.YOUTUBE: "https://www.youtube.com/", - cls.MUSIFY: "https://musify.club/", - cls.MUSICBRAINZ: "https://musicbrainz.org/", - cls.ENCYCLOPAEDIA_METALLUM: "https://www.metal-archives.com/", - cls.GENIUS: "https://genius.com/", - cls.BANDCAMP: "https://bandcamp.com/", - cls.DEEZER: "https://www.deezer.com/", - cls.INSTAGRAM: "https://www.instagram.com/", - cls.FACEBOOK: "https://www.facebook.com/", - cls.SPOTIFY: "https://open.spotify.com/", - cls.TWITTER: "https://twitter.com/", - cls.MYSPACE: "https://myspace.com/", - cls.WIKIPEDIA: "https://en.wikipedia.org/wiki/Main_Page" - } - return homepage_map[attribute] - - class Source(DatabaseObject): """ create somehow like that @@ -62,18 +19,27 @@ class Source(DatabaseObject): """ COLLECTION_ATTRIBUTES = tuple() SIMPLE_ATTRIBUTES = { - "type_enum": None, "page_enum": None, - "url": None + "url": None, + "referer_page": None, + "audio_url": None } - def __init__(self, page_enum: SourcePages, url: str, id_: str = None, type_enum=None) -> None: + def __init__( + self, + page_enum: SourcePages, + url: str, + id_: str = None, + referer_page: SourceTypes = None, + adio_url: str = None + ) -> None: DatabaseObject.__init__(self, id_=id_) - self.type_enum = type_enum self.page_enum = page_enum + self.referer_page = page_enum if referer_page is None else referer_page self.url = url + self.audio_url = adio_url @classmethod def match_url(cls, url: str) -> Optional["Source"]: diff --git a/src/music_kraken/utils/enums/__init__.py b/src/music_kraken/utils/enums/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/src/music_kraken/utils/enums/source.py b/src/music_kraken/utils/enums/source.py new file mode 100644 index 0000000..b536e5c --- /dev/null +++ b/src/music_kraken/utils/enums/source.py @@ -0,0 +1,45 @@ +from enum import Enum + + +class SourceTypes(Enum): + SONG = "song" + ALBUM = "album" + ARTIST = "artist" + LYRICS = "lyrics" + + +class SourcePages(Enum): + YOUTUBE = "youtube" + MUSIFY = "musify" + GENIUS = "genius" + MUSICBRAINZ = "musicbrainz" + ENCYCLOPAEDIA_METALLUM = "encyclopaedia metallum" + BANDCAMP = "bandcamp" + DEEZER = "deezer" + SPOTIFY = "spotify" + + # This has nothing to do with audio, but bands can be here + WIKIPEDIA = "wikipedia" + INSTAGRAM = "instagram" + FACEBOOK = "facebook" + TWITTER = "twitter" # I will use nitter though lol + MYSPACE = "myspace" # Yes somehow this ancient site is linked EVERYWHERE + + @classmethod + def get_homepage(cls, attribute) -> str: + homepage_map = { + cls.YOUTUBE: "https://www.youtube.com/", + cls.MUSIFY: "https://musify.club/", + cls.MUSICBRAINZ: "https://musicbrainz.org/", + cls.ENCYCLOPAEDIA_METALLUM: "https://www.metal-archives.com/", + cls.GENIUS: "https://genius.com/", + cls.BANDCAMP: "https://bandcamp.com/", + cls.DEEZER: "https://www.deezer.com/", + cls.INSTAGRAM: "https://www.instagram.com/", + cls.FACEBOOK: "https://www.facebook.com/", + cls.SPOTIFY: "https://open.spotify.com/", + cls.TWITTER: "https://twitter.com/", + cls.MYSPACE: "https://myspace.com/", + cls.WIKIPEDIA: "https://en.wikipedia.org/wiki/Main_Page" + } + return homepage_map[attribute] \ No newline at end of file