Chinaunix首页 | 论坛 | 博客
  • 博客访问: 566874
  • 博文数量: 44
  • 博客积分: 5017
  • 博客等级: 大校
  • 技术积分: 508
  • 用 户 组: 普通用户
  • 注册时间: 2006-10-17 16:03
文章分类

全部博文(44)

文章存档

2011年(1)

2010年(1)

2009年(1)

2008年(5)

2007年(22)

2006年(14)

我的朋友

分类: Python/Ruby

2010-02-04 18:48:06

昨天一个帮一个同学找北京到西安的火车票,突然想到旷同学正好写了一个,但是需要指定火车票的是日期,于是就在上面稍微改了下,并增加了打印链接地址。

checkhcp.py:

#!/usr/bin/python
# encoding: utf-8

import urllib2
import urlparse
import mailbox
import os
import sys
import re
from sgmllib import SGMLParser
import time

class URLListName(SGMLParser):
    is_a=""
    name=[]
    link={}
    def start_a(self, attrs):
        self.is_a=1
        self.attrs = attrs

    def end_a(self):
        self.is_a=""

    def handle_data(self, text):
        ret = text.find('发车日期')
        if self.is_a and ret != -1:
            for atname,atvalue in self.attrs:
                if atname == 'href':
                    self.link[text] = atvalue

if len(sys.argv) < 5:
   mesg = """
用法: python checkhcp.py 始发站 终点站 月 日 关键字

例如,查找2月10号北京到西安的卧铺票:
# python checkhcp.py beijing xian 2 10 卧铺

注:始发站和终点站必须用拼音,关键字可以是车次,车票类型,车票张数等。
"
""
   print mesg
   sys.exit(1)

start = sys.argv[1]
end = sys.argv[2]
month = sys.argv[3]
day = sys.argv[4]
key = ''
if len(sys.argv) == 6:
   key = sys.argv[5]

url = "%s-%s/?sort=d1&Cat1=%s月&Cat3=%s日&page=1" % (start, end, month, day)

sequence = 60#60 * 5

request = urllib2.Request(url)
request.add_header('User-Agent', 'Mozilla/5.0')
opener = urllib2.build_opener()
data = opener.open(request).read()

ticket_name = URLListName()
ticket_name.feed(data)
cache=[]

while 1:
    try:
        print "beign retrive"
        data = opener.open(request).read()
        ticket_name.feed(data)
        ticket_name.name.extend(ticket_name.link)
        print "beign scan"
        
        for result in ticket_name.name:
            if result and result.find(key) >= 0:
                if result in cache:
                    pass
                else:
                    print '\033[91m' + "found:" + result + '\033[0m'
                    print ticket_name.link[result]
                    cache.append(result)
        
        print "scan finished, begin sleep " + str(sequence) + " seconds."
        #print ticket_name.link
        time.sleep(sequence)
    except:
        raise


阅读(2957) | 评论(0) | 转发(0) |
给主人留下些什么吧!~~