music-kraken-core/src/music_kraken_cli.py

135 lines
3.6 KiB
Python
Raw Normal View History

2022-11-23 07:10:22 +00:00
import musicbrainzngs
import logging
import os
from music_kraken.utils.shared import (
MUSIC_DIR,
NOT_A_GENRE
)
from music_kraken.metadata import (
metadata_search,
metadata_fetch
)
from music_kraken.target import set_target
from music_kraken.audio_source import (
fetch_source,
fetch_audio
)
from music_kraken.lyrics import lyrics
def clear_console():
os.system('cls' if os.name in ('nt', 'dos') else 'clear')
logging.getLogger("musicbrainzngs").setLevel(logging.WARNING)
musicbrainzngs.set_useragent("metadata receiver", "0.1", "https://github.com/HeIIow2/music-downloader")
def get_existing_genre():
valid_directories = []
for elem in os.listdir(MUSIC_DIR):
if elem not in NOT_A_GENRE:
valid_directories.append(elem)
return valid_directories
def help_search_metadata():
msg = """
- - - Type the command you want to execute - - -
.. - Previous Options
(query_string) - Search for songs, albums, bands...
(int) - Select an item from the search results
d - Start the download
h - Help
q - Quit / Exit
"""
print(msg)
def search_for_metadata():
clear_console()
search = metadata_search.Search()
while True:
input_ = input("\"help\" for an overfiew of commands: ")
match (input_.strip().lower()):
case "d" | "ok" | "dl" | "download":
break
case "q" | "quit" | "exit":
clear_console()
exit()
case "h" | "help":
help_search_metadata()
2022-11-23 07:45:33 +00:00
continue
2022-11-23 07:10:22 +00:00
case inp if inp.isdigit():
print()
print(search.choose(int(input_)))
continue
case ".." :
print()
print(search.get_previous_options())
2022-11-23 07:45:33 +00:00
continue
2022-11-23 07:10:22 +00:00
print()
print(search.search_from_query(input_))
print(search.current_option)
return search.current_option
def get_genre():
existing_genres = get_existing_genre()
print("printing available genres:")
for i, genre_option in enumerate(existing_genres):
print(f"{i}: {genre_option}")
genre = input("Input the ID for an existing genre or text for a new one: ")
if genre.isdigit():
genre_id = int(genre)
if genre_id >= len(existing_genres):
logging.warning("An invalid genre id has been given")
return get_genre()
return existing_genres[genre_id]
return genre
def cli(start_at: int = 0, only_lyrics: bool = False, cleare_console: bool = True):
if clear_console:
clear_console()
if start_at <= 2 and not only_lyrics:
genre = get_genre()
logging.info(f"{genre} has been set as genre.")
if start_at <= 0:
search = search_for_metadata()
# search = metadata.search.Option("release", "f8d4b24d-2c46-4e9c-8078-0c0f337c84dd", "Beautyfall")
logging.info("Starting Downloading of metadata")
metadata_downloader = metadata_fetch.MetadataDownloader()
metadata_downloader.download({'type': search.type, 'id': search.id})
if start_at <= 1 and not only_lyrics:
logging.info("creating Paths")
set_target.UrlPath(genre=genre)
if start_at <= 2 and not only_lyrics:
logging.info("Fetching Download Links")
fetch_source.Download()
if start_at <= 3 and not only_lyrics:
logging.info("starting to download the mp3's")
fetch_audio.Download()
if start_at <= 4:
logging.info("starting to fetch the lyrics")
lyrics.fetch_lyrics()
2022-11-15 09:03:49 +00:00
if __name__ == "__main__":
cli()