kyuko info

kyuko.py

#!/usr/lib/python
#vim:fileencoding=utf-8

import cookielib
import re
import urllib
import urllib2
import time
import twitter
import yaml

class Kyuko:
    def __init__(self, screen_name, api, login, passwd):
        self.api = api 
        self.name = screen_name
    
        cj = cookielib.CookieJar()
        cjhdr = urllib2.HTTPCookieProcessor(cj)
        opener = urllib2.build_opener(cjhdr)

        loginurl = ''
        postdata = {}
        postdata['login'] = login 
        postdata['passwd'] = passwd 
        postdata['mode'] = 'Login'
        postdata['clickcheck'] = '0' 
        params = urllib.urlencode(postdata)

        opener.open(loginurl, params).read()

        self.html = opener.open('').read()

    def parseHtml(self):
        text = re.sub('\n|<.*?>', '', self.html).strip()
        res = re.findall(u'20[0-9]{2}.*?[1-9]{4}', text)
        return res 

    def main(self):
        lists = self.parseHtml()
        tweet = u'' 
        if lists:
            for s in lists:
                tweet += s.decode('utf-8') + u'｜'
        else:
            tweet = u'休講なし' + str(time.time())
        self.api.PostUpdates('D ' + self.name + ' ' + tweet.encode('utf-8'))

if __name__ == '__main__':
    yaml_file = open('bot.yaml', 'r').read()
    yam = yaml.load(yaml_file)
    screen_name = yam['twitter.com']['screenname']
    consumer_key = yam['twitter.com']['consumer_key']
    consumer_secret = yam['twitter.com']['consumer_secret']
    access_token_key = yam['twitter.com']['access_token_key']
    access_token_secret = yam['twitter.com']['access_token_secret']

    login = yam['jp']['login']
    passwd = yam['jp']['passwd']

    api = twitter.Api(consumer_key, consumer_secret, access_token_key, access_token_secret)

    obj = Kyuko(screen_name, api, login, passwd)
    obj.main()