Initialize sources on first update

This commit is contained in:
Tim Van Baak 2019-12-17 10:09:19 -08:00
parent 90b38195ec
commit 3919346cb4
1 changed files with 12 additions and 0 deletions

View File

@ -16,6 +16,18 @@ def update_sources(*source_names):
error.as_item("Error importing source '{}'".format(source_name), traceback.format_exc()) error.as_item("Error importing source '{}'".format(source_name), traceback.format_exc())
continue continue
cell_path = os.path.join(DUNGEON_PATH, source_name)
if not os.path.isdir(cell_path):
try:
logger.info("Creating cell for source '{}'".format(source_name))
os.mkdir(cell_path)
state_path = os.path.join(cell_path, "state")
with open(state_path, 'w', encoding='utf8') as f:
f.write(json.dumps({}))
except Exception as e:
error.as_item("Error initializing source '{}'".format(source_name), traceback.format_exc())
continue
try: try:
logger.info("Updating source '{}'".format(source_name)) logger.info("Updating source '{}'".format(source_name))
new_count, del_count = update_source(source_name, source_module.fetch_new) new_count, del_count = update_source(source_name, source_module.fetch_new)