import json import os import sys from attrdict import AttrDict class Singleton(type): _instances = {} def __call__(cls, *args, **kwargs): if cls not in cls._instances: cls._instances[cls] = super(Singleton, cls).__call__(*args, **kwargs) return cls._instances[cls] 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 class Config(object, metaclass=Singleton): base_config = { 'core': { 'debug': False }, 'notifications': { 'verbose': True }, 'automatic': { 'movies': { 'interval': 20, 'anticipated': 3, 'trending': 3, 'popular': 3, 'boxoffice': 10 }, 'shows': { 'interval': 48, 'anticipated': 10, 'trending': 1, 'popular': 1 } }, 'filters': { 'shows': { 'disabled_for': [], 'allowed_countries': [], 'allowed_languages': [], 'blacklisted_genres': [], 'blacklisted_networks': [], 'blacklisted_min_runtime': 15, 'blacklisted_min_year': 2000, 'blacklisted_max_year': 2019, 'blacklisted_title_keywords': [], 'blacklisted_tvdb_ids': [], }, 'movies': { 'disabled_for': [], 'allowed_countries': [], 'allowed_languages': [], 'blacklisted_genres': [], 'blacklisted_min_runtime': 60, 'blacklisted_min_year': 2000, 'blacklisted_max_year': 2019, 'blacklisted_title_keywords': [], 'blacklisted_tmdb_ids': [], 'rotten_tomatoes': "" } }, 'radarr': { 'api_key': '', 'minimum_availability': 'released', 'quality': 'HD-1080p', 'root_folder': '/movies/', 'url': 'http://localhost:7878/' }, 'sonarr': { 'api_key': '', 'language': 'English', 'quality': 'HD-1080p', 'root_folder': '/tv/', 'tags': { }, 'url': 'http://localhost:8989/' }, 'omdb': { 'api_key': '' }, 'trakt': { 'client_id': '', 'client_secret': '' } } def __init__(self, configfile, cachefile, logfile): """Initializes config""" self.conf = None self.config_path = configfile self.cache_path = cachefile self.log_path = logfile @property def cfg(self): # Return existing loaded config if self.conf: return self.conf # Built initial config if it doesn't exist if self.build_config(): print("Please edit the default configuration before running again!") sys.exit(0) # Load config, upgrade if necessary else: tmp = self.load_config() self.conf, upgraded = self.upgrade_settings(tmp) # Save config if upgraded if upgraded: self.dump_config() print("New config options were added, adjust and restart!") sys.exit(0) return self.conf @property def cachefile(self): return self.cache_path @property def logfile(self): return self.log_path def build_config(self): if not os.path.exists(self.config_path): print("Dumping default config to: %s" % self.config_path) with open(self.config_path, 'w') as fp: json.dump(self.base_config, fp, sort_keys=True, indent=2) return True else: return False def dump_config(self): if os.path.exists(self.config_path): with open(self.config_path, 'w') as fp: json.dump(self.conf, fp, sort_keys=True, indent=2) return True else: return False def load_config(self): with open(self.config_path, 'r') as fp: return AttrConfig(json.load(fp)) def __inner_upgrade(self, settings1, settings2, key=None, overwrite=False): sub_upgraded = False merged = settings2.copy() if isinstance(settings1, dict): for k, v in settings1.items(): # missing k if k not in settings2: merged[k] = v sub_upgraded = True if not key: print("Added %r config option: %s" % (str(k), str(v))) else: print("Added %r to config option %r: %s" % (str(k), str(key), str(v))) continue # iterate children if isinstance(v, dict) or isinstance(v, list): merged[k], did_upgrade = self.__inner_upgrade(settings1[k], settings2[k], key=k, overwrite=overwrite) sub_upgraded = did_upgrade if did_upgrade else sub_upgraded elif settings1[k] != settings2[k] and overwrite: merged = settings1 sub_upgraded = True elif isinstance(settings1, list) and key: for v in settings1: if v not in settings2: merged.append(v) sub_upgraded = True print("Added to config option %r: %s" % (str(key), str(v))) continue return merged, sub_upgraded def upgrade_settings(self, currents): upgraded_settings, upgraded = self.__inner_upgrade(self.base_config, currents) return AttrConfig(upgraded_settings), upgraded def merge_settings(self, settings_to_merge): upgraded_settings, upgraded = self.__inner_upgrade(settings_to_merge, self.conf, overwrite=True) self.conf = upgraded_settings if upgraded: self.dump_config() return AttrConfig(upgraded_settings), upgraded