基于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-07、leftTicketDTO.from_station=CDW、leftTicketDTO.to_station=SHH、purpose_codes=ADULT 四个参数。

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

  
  
  

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

• 3.在请求的url中出发点和目的地都是英文缩写，例如上面的CDW 和SHH，而我们在命令行中输入的是中文，那么需要通过中文查找到对应的英文缩写，这个用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()