京急の運行情報を取得するスクリプトをpythonで書く（unko.py）

京急の運行情報を取得するスクリプト

京急を愛してやまない。めったに遅延することがない安定感は素晴らしいものがあります。それ故に、たまに遅延すると結構面食らってしまうことも。

unkou.keikyu.co.jp

というわけで、スクリプトを書きました。ホントはJSONとかで取得できる口があればいいのですが、なさそうなのでHTMLをそのままパースします。

コーディングします

pythonは不慣れなので、クソコードであることを予めご承知おきください。ポイントはdivタグのunko-panelのボデーに運行情報が書かれているので、そこだけ抽出します。

import urllib2
from HTMLParser import HTMLParser

unko = ""

class UnkoParser(HTMLParser):
    def __init__(self):
        HTMLParser.__init__(self)
        self.flag = False
        
    def handle_starttag(self, tag, attrs):
        if tag == "div":
            attrs = dict(attrs)
            if 'class' in attrs:
                className = attrs['class']
                if className == 'unko-panel':
                    self.flag = True
    
    def handle_data(self, data):
        if self.flag:         
            global unko
            unko = unko + data
        
    def handle_endtag(self, tag):
        self.flag = False
    

url = "http://unkou.keikyu.co.jp/"
response = urllib2.urlopen(url)

parser = UnkoParser()
parser.feed(response.read())
print unko.replace("\n", "")

response.close()