relay/relay/database.py

34 lines
733 B
Python

import asyncio
import logging
import urllib.parse
import simplejson as json
from . import CONFIG
try:
with open(CONFIG['db']) as f:
DATABASE = json.load(f)
except:
logging.info('No database was found, making a new one.')
DATABASE = {}
following = DATABASE.get('relay-list', [])
for inbox in following:
if urllib.parse.urlsplit(inbox).hostname in CONFIG['ap']['blocked_instances']:
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())