from metadata.database import Database from metadata.download import MetadataDownloader import metadata.download import metadata.metadata import download_links import url_to_path import download import logging import os import tempfile logging.basicConfig(level=logging.INFO) TEMP_FOLDER = "music-downloader" DATABASE_FILE = "metadata.db" DATABASE_STRUCTURE_FILE = "database_structure.sql" 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") NOT_A_GENRE = ".", "..", "misc_scripts", "Music", "script", ".git", ".idea" MUSIC_DIR = os.path.expanduser('~/Music') TOR = False logger = logging.getLogger() logger.level = logging.DEBUG temp_dir = os.path.join(tempfile.gettempdir(), TEMP_FOLDER) if not os.path.exists(temp_dir): os.mkdir(temp_dir) database = Database(os.path.join(temp_dir, DATABASE_FILE), os.path.join(temp_dir, DATABASE_STRUCTURE_FILE), DATABASE_LOGGER, reset_anyways=True) 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 search_for_metadata(query: str): search = metadata.metadata.Search(query=query) print(search.options) while True: input_ = input( "q to quit, ok to download, .. for previous options, . for current options, int for this element: ").lower() input_.strip() if input_ == "q": exit(0) if input_ == "ok": return search.current_chosen_option if input_ == ".": print(search.options) continue if input_ == "..": print(search.get_previous_options()) continue if input_.isdigit(): print(search.choose(int(input_))) continue 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): proxies = None if TOR: proxies = { 'http': 'socks5h://127.0.0.1:9150', 'https': 'socks5h://127.0.0.1:9150' } if start_at <= 2: genre = get_genre() logging.info(f"{genre} has been set as genre.") if start_at <= 0: search = search_for_metadata(query=input("initial query: ")) logging.info("Starting Downloading of metadata") metadata_downloader = MetadataDownloader(database, METADATA_DOWNLOAD_LOGGER) metadata_downloader.download(search) if start_at <= 1: logging.info("Fetching Download Links") download_links.Download(database, METADATA_DOWNLOAD_LOGGER, proxies=proxies) if start_at <= 2: logging.info("creating Paths") url_to_path.UrlPath(database, PATH_LOGGER, genre=genre) if start_at <= 3: logging.info("starting to download the mp3's") download.Download(database, DOWNLOAD_LOGGER, proxies=proxies, base_path=MUSIC_DIR) if __name__ == "__main__": cli(start_at=0)