ticketfrei/backend.py

#!/usr/bin/env python3

import prepare
import time

import sendmail
from db import DB

from retootbot import RetootBot
# from retweetbot import RetweetBot
# from mailbot import Mailbot
from trigger import Trigger


def get_users(db):
    user_rows = db.get_users()
    users = {}
    for row in user_rows:
        users[row[0]] = []
    return users


def init_bots(config, logger, db, users):
    for uid in users:
        users[uid].append(RetootBot(config, logger, uid, db))
        # users[uid].append(RetweetBot(config, uid, db))
        # users[uid].append(Mailbot(config, uid, db))
    return users


def run():
    config = prepare.get_config()
    logger = prepare.get_logger(config)
    db = DB()

    # set trigger
    trigger = Trigger(config)

    while True:
        # get a dictionary { uid : [ Bot objects ] }
        users = get_users(db)

        # initialize bots
        users = init_bots(config, logger, db, users)

        try:
            for uid in users:
                for bot in users[uid]:
                    reports = bot.crawl()
                    for status in reports:
                        if not trigger.is_ok(status.text):
                            continue
                        for bot2 in users[uid]:
                            if bot == bot2:
                                bot2.repost(status)
                            else:
                                bot2.post(status)
                time.sleep(60)  # twitter rate limit >.<
        except KeyboardInterrupt:
            print("Good bye. Remember to restart the bot!")
        except:
            logger.error('Shutdown', exc_info=True)
            for uid in users:
                for bot in users[uid]:
                    bot.save_last()
            mailer = sendmail.Mailer(config)
            try:
                mailer.send('', config['mail']['contact'],
                            'Ticketfrei Crash Report',
                            attachment=config['logging']['logpath'])
            except:
                logger.error('Mail sending failed', exc_info=True)


if __name__ == '__main__':
    run()