1
0
Fork 0
mirror of https://github.com/Lynnesbian/FediBooks/ synced 2024-09-27 22:53:04 +00:00
FediBooks/service.py
2019-09-10 12:09:38 +10:00

29 lines
813 B
Python
Executable file

#!/usr/bin/env python3
import MySQLdb
from multiprocessing import Pool
import json
import functions
cfg = json.load(open('config.json'))
print("Establishing DB connection")
db = MySQLdb.connect(
host = cfg['db_host'],
user=cfg['db_user'],
passwd=cfg['db_pass'],
db=cfg['db_name']
)
print("Cleaning up database")
# delete any fedi accounts we no longer need
cursor = db.cursor()
cursor.execute("DELETE FROM fedi_accounts WHERE handle NOT IN (SELECT fedi_id FROM bot_learned_accounts)")
print("Generating posts")
cursor.execute("SELECT handle FROM bots WHERE enabled = TRUE AND TIMESTAMPDIFF(MINUTE, last_post, CURRENT_TIMESTAMP()) > post_frequency")
bots = cursor.fetchall()
with Pool(8) as p:
p.map(functions.make_post, bots)
#TODO: other cron tasks should be done here, like updating profile pictures