config.py

import json
import logging
import logging.handlers
from pathlib import Path
from typing import Dict, Union


_LOGGER_NAME = 'sorter-gpm'

LOGGER = logging.getLogger(_LOGGER_NAME)
LOGGER.setLevel(logging.DEBUG)

_FH = logging.handlers.RotatingFileHandler(
    f'{_LOGGER_NAME}.log',
    maxBytes=40960,
    backupCount=5,
    )
_FH.setLevel(logging.DEBUG)

_CH = logging.StreamHandler()
_CH.setLevel(logging.WARNING)

_FORMATTER = logging.Formatter(
    '%(asctime)s - %(name)s - %(levelname)s - %(message)s'
    )
_FH.setFormatter(_FORMATTER)
_CH.setFormatter(_FORMATTER)

LOGGER.addHandler(_FH)
LOGGER.addHandler(_CH)


_CONFIG_LOAD_ERRORS = (
    FileNotFoundError,
    KeyError,
    TypeError,
    ValueError,
    json.decoder.JSONDecodeError,
    )


_FIELDS = ('Artist', 'Album', 'Title')

CORR: Dict[str, Union[Dict[str, str], Dict]] = {}


def insert_missing_corrections() -> None:
    """Insert empty dictionaries into corrections if not present."""
    for field in _FIELDS:
        if field not in CORR:
            CORR[field] = {}


try:
    with open('config.json', 'r') as f:
        CONFIG = json.load(f)
    FMT = CONFIG['format']
    _TAKEOUT = Path(CONFIG['takeout_dir']).expanduser()
    TRACKS = _TAKEOUT / 'Google Play Music' / 'Tracks'
    DEST = Path(CONFIG['dest_dir']).expanduser()
except _CONFIG_LOAD_ERRORS as e:
    LOGGER.error("config.json doesn't exist or is malformed.")
    LOGGER.error(f'More information: {e}')
    raise e

try:
    with open('corrections.json', 'r') as f:
        CORR = json.load(f)
except _CONFIG_LOAD_ERRORS:
    LOGGER.info("corrections.json doesn't exist. Skipping.")

insert_missing_corrections()