mirror of https://github.com/l3uddz/traktarr
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
106 lines
3.0 KiB
106 lines
3.0 KiB
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', 'fx', '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()
|