44 lines
1.0 KiB
Python
44 lines
1.0 KiB
Python
import asyncio
|
|
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')
|
|
|
|
async def database_save():
|
|
while True:
|
|
with open(CONFIG['db'], 'w') as f:
|
|
json.dump(DATABASE, f)
|
|
await asyncio.sleep(30)
|
|
|
|
|
|
asyncio.ensure_future(database_save())
|