47 lines
1.1 KiB
Python
47 lines
1.1 KiB
Python
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()
|