FUCOROIDフォローカウンター

counter.py

#!/usr/bin/env python
# -*- coding:utf-8 -*-

import config
from TwitterBot import BaseBot
import logging

logger = logging.getLogger("Bot.counter")

class Bot(BaseBot):
    def __init__(self):
        super(Bot, self).__init__(
            config.CONSUMER_KEY,
            config.CONSUMER_SECRET,
            config.ACCESS_KEY,
            config.ACCESS_SECRET)
        self.counter = {}
        self.append_cron('00 00 * * *', Bot.report)

    def on_follow(self, target, source):
        name = source.screen_name
        self.counter[name] = self.counter.get(name, 0) + 1
        logger.info("Follow: %s(%d)", name, self.counter[name])

    def report(self):
        if "FUCOROID" in self.counter:
            self.update_status(u"今日はFUROROIDに%d回フォローされました" % self.counter['FUCOROID'])
        else:
            self.update_status(u"今日はFUROROIDにフォローされませんでした！やったね！")
        self.counter = {}

def main():
    bot = Bot()
    bot.main()

if __name__=='__main__':
    main()