import time import threading import logging import urllib.parse import simplejson as json from sys import exit from . import CONFIG AP_CONFIG = CONFIG['ap'] try: with open(CONFIG['db']) as f: DATABASE = json.load(f) except FileNotFoundError: logging.info('No database was found, making a new one.') DATABASE = {} except json.decoder.JSONDecodeError: logging.info('Invalid JSON in db. Exiting...') exit(1) following = DATABASE.get('relay-list', []) for inbox in following: if urllib.parse.urlsplit(inbox).hostname in AP_CONFIG['blocked_instances']: following.remove(inbox) elif AP_CONFIG['whitelist_enabled'] is True and urllib.parse.urlsplit(inbox).hostname not in AP_CONFIG['whitelist']: following.remove(inbox) DATABASE['relay-list'] = following if 'actors' in DATABASE: DATABASE.pop('actors') def database_save(): global DATABASE time.sleep(60) while True: logging.debug('written database file {}'.format(CONFIG['db'])) with open(CONFIG['db'], 'w') as f: json.dump(DATABASE, f) time.sleep(60) threading.Thread(target=database_save).start()