效果圖

• 1.只查詢動車票

  
  

• 2.查詢所有票

  
  

Github鏈接

代碼鏈接為:https://github.com/happyte/tickets

接口設計

• 1.查詢火車票，需要出發(fā)地點，目的點，日期和所乘列類型這幾個參數，因此設計出的接口為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瀏覽器抓取下網頁的數據
  

  
  從上圖中發(fā)現請求的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中出發(fā)點和目的地都是英文縮寫，例如上面的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.在命令行執(zhí)行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.創(chuàng)建一個類用于分析需要的數據，prettytable 模塊用于創(chuàng)建圖形化表格，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.創(chuàng)建一個上面類的對象，調用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()