music-kraken-core/src/music_kraken/utils/shared.py

46 lines
1.5 KiB
Python
Raw Normal View History

2022-11-11 23:29:07 +00:00
import musicbrainzngs
import logging
import tempfile
import os
2022-11-16 09:15:35 +00:00
from .database import Database
2022-11-11 23:29:07 +00:00
TEMP_FOLDER = "music-downloader"
LOG_FILE = "download_logs.log"
DATABASE_FILE = "metadata.db"
DATABASE_STRUCTURE_FILE = "database_structure.sql"
2022-11-14 14:44:32 +00:00
DATABASE_STRUCTURE_FALLBACK = "https://raw.githubusercontent.com/HeIIow2/music-downloader/master/assets/database_structure.sql"
2022-11-11 23:29:07 +00:00
SEARCH_LOGGER = logging.getLogger("mb-cli")
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")
2022-11-14 14:44:32 +00:00
LYRICS_LOGGER = logging.getLogger("lyrics")
2022-11-11 23:29:07 +00:00
GENIUS_LOGGER = logging.getLogger("genius")
NOT_A_GENRE = ".", "..", "misc_scripts", "Music", "script", ".git", ".idea"
MUSIC_DIR = os.path.expanduser('~/Music')
temp_dir = os.path.join(tempfile.gettempdir(), TEMP_FOLDER)
if not os.path.exists(temp_dir):
os.mkdir(temp_dir)
2022-11-13 14:57:19 +00:00
logging.getLogger("musicbrainzngs").setLevel(logging.WARNING)
2022-11-11 23:29:07 +00:00
musicbrainzngs.set_useragent("metadata receiver", "0.1", "https://github.com/HeIIow2/music-downloader")
database = Database(os.path.join(temp_dir, DATABASE_FILE),
os.path.join(temp_dir, DATABASE_STRUCTURE_FILE),
DATABASE_STRUCTURE_FALLBACK,
DATABASE_LOGGER,
2022-11-13 22:56:43 +00:00
reset_anyways=False)
2022-11-11 23:29:07 +00:00
TOR = False
proxies = {
'http': 'socks5h://127.0.0.1:9150',
'https': 'socks5h://127.0.0.1:9150'
} if TOR else {}