traktarr/misc/config.py

import json
import os
import sys

from attrdict import AttrDict

config_path = os.path.join(os.path.dirname(sys.argv[0]), 'config.json')
base_config = {
    'core': {
        'debug': False
    },
    'trakt': {
        'api_key': ''
    },
    'sonarr': {
        'url': 'http://localhost:8989',
        'api_key': '',
        'profile': 'HD-1080p',
        'root_folder': '/tv/',
        'tags': {
            'amzn': ['hbo', 'amc', 'usa network', 'tnt', 'starz', 'the cw', 'fxx', 'fox', 'abc', 'nbc', 'cbs', 'tbs',
                     'amazon', 'syfy', 'cinemax']
        }
    },
    'radarr': {
        'url': 'http://localhost:7878',
        'api_key': '',
        'profile': 'HD-1080p',
        'root_folder': '/movies/'
    },
    'filters': {
        'shows': {
            'blacklisted_genres': ['animation', 'game-show', 'talk-show', 'home-and-garden', 'children', 'reality',
                                   'anime', 'news', 'documentary', 'special-interest'],
            'blacklisted_networks': ['twitch', 'youtube', 'nickelodeon', 'hallmark', 'reelzchannel', 'disney',
                                     'cnn', 'cbbc', 'the movie network', 'teletoon', 'cartoon network', 'espn',
                                     'yahoo!',
                                     'fox sports'],
            'allowed_countries': ['us', 'gb', 'ca'],
            'blacklisted_min_runtime': 15,
            'blacklisted_min_year': 2000
        },
        'movies': {
            'blacklisted_genres': ['documentary', 'music'],
            'blacklisted_min_runtime': 60,
            'blacklisted_min_year': 2000,
            'blacklist_title_keywords': ['untitled', 'barbie'],
            'allowed_countries': ['us', 'gb', 'ca']
        }
    }
}
cfg = None


class AttrConfig(AttrDict):
    """
    Simple AttrDict subclass to return None when requested attribute does not exist
    """

    def __init__(self, config):
        super().__init__(config)

    def __getattr__(self, item):
        try:
            return super().__getattr__(item)
        except AttributeError:
            pass
        # Default behaviour
        return None


def build_config():
    if not os.path.exists(config_path):
        print("Dumping default config to: %s" % config_path)
        with open(config_path, 'w') as fp:
            json.dump(base_config, fp, sort_keys=True, indent=2)
        return True
    else:
        return False


def dump_config():
    if os.path.exists(config_path):
        with open(config_path, 'w') as fp:
            json.dump(cfg, fp, sort_keys=True, indent=2)
        return True
    else:
        return False


def load_config():
    with open(config_path, 'r') as fp:
        return AttrConfig(json.load(fp))


############################################################
# LOAD CFG
############################################################

# dump/load config
if build_config():
    print("Please edit the default configuration before running again!")
    sys.exit(0)
else:
    cfg = load_config()