126 lines
3.6 KiB
Python
126 lines
3.6 KiB
Python
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)
|