load new defaults from config.json #21

This commit is contained in:
simon 2021-09-20 17:45:01 +07:00
parent 690eb5befd
commit 4b9acd4c51
2 changed files with 56 additions and 11 deletions

View File

@ -12,7 +12,9 @@ from .tasks import app as celery_app
def sync_redis_state(): def sync_redis_state():
""" make sure redis gets the config.json values """ """ make sure redis gets the config.json values """
print('sync redis') print('sync redis')
config = AppConfig().config config_handler = AppConfig()
config_handler.load_new_defaults()
config = config_handler.config
sort_order = config['archive']['sort'] sort_order = config['archive']['sort']
set_message('sort_order', sort_order, expire=False) set_message('sort_order', sort_order, expire=False)
hide_watched = bool(int(config['archive']['hide_watched'])) hide_watched = bool(int(config['archive']['hide_watched']))

View File

@ -21,15 +21,32 @@ class AppConfig:
""" get config from default file or redis if changed """ """ get config from default file or redis if changed """
config = self.get_config_redis() config = self.get_config_redis()
if not config: if not config:
config = self.get_config_file()
config['application'].update(self.get_config_env())
return config
def get_config_file(self):
""" read the defaults from config.json """
with open('home/config.json', 'r', encoding="utf-8") as f: with open('home/config.json', 'r', encoding="utf-8") as f:
config_str = f.read() config_str = f.read()
config = json.loads(config_str) config_file = json.loads(config_str)
config['application']['REDIS_HOST'] = os.environ.get('REDIS_HOST') config_file['application'].update(self.get_config_env())
config['application']['es_url'] = os.environ.get('ES_URL')
config['application']['HOST_UID'] = int(os.environ.get('HOST_UID')) return config_file
config['application']['HOST_GID'] = int(os.environ.get('HOST_GID'))
return config @staticmethod
def get_config_env():
""" read environment application variables """
application = {
'REDIS_HOST': os.environ.get('REDIS_HOST'),
'es_url': os.environ.get('ES_URL'),
'HOST_UID': int(os.environ.get('HOST_UID')),
'HOST_GID': int(os.environ.get('HOST_GID'))
}
return application
@staticmethod @staticmethod
def get_config_redis(): def get_config_redis():
@ -48,13 +65,39 @@ class AppConfig:
if len(to_write): if len(to_write):
if to_write == '0': if to_write == '0':
to_write = False to_write = False
elif to_write == '1':
to_write = True
elif to_write.isdigit(): elif to_write.isdigit():
to_write = int(to_write) to_write = int(to_write)
config_dict, config_value = key.split('.') config_dict, config_value = key.split('.')
config[config_dict][config_value] = to_write config[config_dict][config_value] = to_write
with open('home/config.json', 'w', encoding="utf-8") as f:
f.write(json.dumps(config))
set_message('config', config, expire=False) set_message('config', config, expire=False)
def load_new_defaults(self):
""" check config.json for missing defaults """
default_config = self.get_config_file()
redis_config = self.get_config_redis()
# check for customizations
if not redis_config:
return
needs_update = False
for key, value in default_config.items():
# missing whole main key
if key not in redis_config:
redis_config.update({key: value})
needs_update = True
continue
# missing nested values
for sub_key, sub_value in value.items():
if sub_key not in redis_config[key].keys():
redis_config[key].update({sub_key: sub_value})
needs_update = True
if needs_update:
set_message('config', redis_config, expire=False)