基于python的火车票查询工具

效果图

  • 1.只查询动车票


  • 2.查询所有票


Github链接

代码链接为:https://github.com/happyte/tickets

接口设计

  • 1.查询火车票,需要出发地点,目的点,日期和所乘列类型这几个参数,因此设计出的接口为python3 tickets.py [-gdtkz] <from> <to> <date> , [-gdtkz] 代表查询的火车类型,该参数可叠加,例如-gd代表查询所有的动车和高铁。
  • 2.python的docopt模块可以解析命令行的参数,代码如下:
"""命令行火车票查看器
Usage:
    tickets [-gdtkz] <from> <to> <date>
Options:
    -h,--help   显示帮助菜单
    -g          高铁
    -d          动车
    -t          特快
    -k          快速
    -z          直达
Example:
    tickets 北京 上海 2016-10-10
    tickets -dg 成都 南京 2016-10-10
"""
 from docopt import docent

 def cli():
      arguments = docopt(__doc__)
      print(arguments)

 if __name__ == '__main__':
     cli()
  • 3.在命令行输入命令python3 tickets.py -dg 成都 上海 2017-02-10 输出的结果如下:

获取12306数据

  • 1.用谷歌的chrome浏览器抓取下网页的数据


    从上图中发现请求的url后面跟了4个参数,分别为leftTicketDTO.train_date=2017-02-07leftTicketDTO.from_station=CDWleftTicketDTO.to_station=SHHpurpose_codes=ADULT 四个参数。

  • 2.再抓取下response,结果如下:



    所有的车次信息都在data这个字典中,0-10这个分别包含了每趟车的所有信息。

  • 3.在请求的url中出发点和目的地都是英文缩写,例如上面的CDWSHH,而我们在命令行中输入的是中文,那么需要通过中文查找到对应的英文缩写,这个用chrome抓取到的数据好像没找到。那么查看下网页源代码


    点击进入第一个station_version=1.8994这个文件,进入之后看到该文件是所有车站对应的中文和英文缩写,但是只需要提取中文和英文的大写缩写,编写一个正则表达式提取。另外新建一个文件叫parse_station.py,代码如下:

 # -*- coding:utf-8 -*-
 import re
 import requests
 from pprint import print

 url = 'https://kyfw.12306.cn/otn/resources/js/framework/station_name.js?station_version=1.8994'
response = requests.get(url, verify=False)     # verify=False不验证证书
stations = re.findall(u'([\u4e00-\u9fa5]+)+\|([A-Z]+)', response.text)
pprint(dict(stations), indent=4)       # indent代表缩进
  • 4.在命令行执行python3 parse_station.py > stations.py,就新建了一个stations.py文件里面生成了一个中文和英文缩写对应的字典,如下所示:
    这里写图片描述

获取需要的信息

  • 1.构造url请求,导入上面生成的stations.py文件
 from stations import stations

 def cli():
    arguments = docopt(__doc__)
    from_station = stations.get(arguments['<from>'])
    to_staion = stations.get(arguments['<to>'])
    date = arguments['<date>']
    url = 'https://kyfw.12306.cn/otn/leftTicket/queryZ?leftTicketDTO.train_date={}&leftTicketDTO.' \
          'from_station={}&leftTicketDTO.to_station={}&purpose_codes=ADULT'.\
        format(date, from_station, to_staion)
    response = requests.get(url, verify=False)
  • 2.得到请求的json应答,因为上面说了所有的列车信息都在data字典内部,因此需要的信息在response.json()['data'] 中。
  • 3.创建一个类用于分析需要的数据,prettytable 模块用于创建图形化表格,colorama 模块用于给表格上色。
class TransCollection:
    header = '车次 车站 时间 历时 一等座 二等座 软卧 硬卧 硬座 无座'.split()

    def __init__(self, available_trains, options):
        self.availavle_trains = available_trains
        self.options = options

    def _get_duration(self, train_data):
        duration = train_data.get('lishi').replace(':', '小时')+'分'
        if duration.startswith('00'):
            return duration[4:]
        if duration.startswith('0'):
            return duration[1:]
        return duration

    @property
    def trains(self):
        for train in self.availavle_trains:
            train_data = train['queryLeftNewDTO']
            train_number = train_data['station_train_code'][0].lower()   # 开头转换成小写
            if not self.options or train_number in self.options:
                train = [
                    train_data['station_train_code'],              # 车次
                    '\n'.join([Fore.GREEN+train_data['from_station_name']+Fore.RESET,    # 车站
                               Fore.RED+train_data['to_station_name']+Fore.RESET]),
                    '\n'.join([Fore.GREEN+train_data['start_time']+Fore.RESET,           # 车站
                               Fore.RED+train_data['arrive_time']+Fore.RESET]),
                    self._get_duration(train_data),                # 历时
                    train_data['zy_num'],                          # 一等座
                    train_data['ze_num'],                          # 二等座
                    train_data['rw_num'],                          # 软卧
                    train_data['yw_num'],                          # 硬卧
                    train_data['yz_num'],                          # 硬座
                    train_data['wz_num'],                          # 无座
                ]
                yield train

    def pretty_print(self):
        pt = PrettyTable()
        pt._set_field_names(self.header)
        for train in self.trains:
            pt.add_row(train)
        print(pt)
  • 4.创建一个上面类的对象,调用pretty_print 函数,在cli函数中添加
def cli():
    arguments = docopt(__doc__)
    from_station = stations.get(arguments['<from>'])
    to_staion = stations.get(arguments['<to>'])
    date = arguments['<date>']
    url = 'https://kyfw.12306.cn/otn/leftTicket/queryZ?leftTicketDTO.train_date={}&leftTicketDTO.' \
          'from_station={}&leftTicketDTO.to_station={}&purpose_codes=ADULT'.\
        format(date, from_station, to_staion)
    response = requests.get(url, verify=False)
    options = ''.join([key for key, value in arguments.items() if value is True])
    TransCollection(response.json()['data'], options).pretty_print()

推荐阅读更多精彩内容

  • 04.01 假期快要来了,心里的小算盘三个人一起打了很久,算计着学校怎样给的假期,盘算着坐哪个火车玩的最如意。来回...
    Ding默昕阅读 45评论 0 0
  • 我写这篇文章是为了解决问题,而不是花了很多时间缓解自己的焦虑。也许你满脑子都是要做的工作,那就别读这篇文章了,赶紧...
    刚好要起飞阅读 103评论 0 0
  • 因为图片占用 Github 太多空间,现在已经放弃 GithubPage 了,但是还是把以前的搭建过程记录一下,说...
    iamzhaozekai阅读 149评论 0 0