python实现12306车票查询

看到网上有很多火车票查询的小脚本，参考一下，发现很多都已经不能再运行了，据说12306接口返回的数据格式更新比较快，这里自己也写了一个。

环境

Mac osx

python3.6

pycharm

效果图

编码

安装脚本用到的模块

requests， 用于请求12306网站网址

docopt， 解析命令行参数

prettytable， 数据用表格的形式打印在终端

colorama， 为打印在表格中的数据着色

安装方式，直接用pip命令就好：

pip install requests prettytable docopt colorama

下面先来介绍一下prettytable docopt colorama这三个模块

docopt

python命令行参数解析工具有很多，这里参考别的查票脚本用的docopt，为了对这个模块进行了解学习，本篇文章也用了这个模块，首先针对本文，我们要查询火车票信息，肯定要输入出发地点，到达地点，出发日期，以及要查询的票的种类，于是我们需要的命令行模型如下：

python tickets.py [-gdtrkz]

- 出发地

- 目的地

- 日期

- [-gdtrkz] 车票类型（对应）

看一下下面的代码

#coding=utf-8

"""
Usage:
python tickets.py [-gdtrkz] <from> <to> <date>
"""
from docopt import docopt

arguments = docopt(__doc__)
print(arguments)

终端运行上面的代码结果如下：



由上面的测试可以看出，docopt能从注释中的Usage下面的命令解析出一个字典，“[ ]”中的是选项，一般不写代表全部，写了代表查询某一选项对应的数据，

若是选项输入错误，不会抛出异常，只会出现以下提示：

Usage:
python tickets.py [-gdtkz] <from> <to> <date>

“< >”中的是参数，如本例中的,,,这里的参数是不能少的，上面提到的选项少了，是可以查询到数据的，这里的参数少了，虽然不会报错，但是不可能有数据。

为了代码的可读性，一般注释中除了Usage（用法）之外，还有参数说明，使用方式等，比如：

"""
Usage:
python tickets.py [-gdtrkz] <from> <to> <date>

Options:
-h,--help   帮助菜单
-g          高铁
-d          动车
-t          特快
-r          高级软卧
-k          快速
-z          直达

Example:
#查询5月10日北京到上海所有车次
python tickets.py 北京 上海 2017-05-10
#查询5月10日南京到上海的动车和高铁
python tickets.py -dg 南京 上海 2017-05-10
"""

prettytable

这个就不多说了，这里有个对它的用法介绍的很详尽的文章

http://www.cnblogs.com/Mr-Koala/p/6582299.html

colorama

这是个对终端输出的文本进行着色的模块，直接上代码图（官方示例）：



知道这些用法就足够本脚本使用了，模块的介绍到此结束，下面正式进入正题。

网页接口的获取

通过chrome监听点击查询按钮后发送的请求如下



从这个请求我们需要知道，这个一个get请求（嗯，确实是get），请求的地址是：https://kyfw.12306.cn/otn/leftTicket/query ，需要的参数有四个：

- leftTicketDTO.train_date=2017-05-10 车票日期2017-05-10

- leftTicketDTO.from_station=HZH 起点站HZH

- leftTicketDTO.to_station=NJH 终点站NJH

- purpose_codes=ADULT 车票类型ADULT（成人票）

起点站和终点站使用站点的字母缩写表示的



车站对应的字母缩写在哪里？？？？

在刚进入车票查询页面时，在页面加载的js文件中，有下面的链接

https://kyfw.12306.cn/otn/resources/js/framework/station_name.js?station_version=1.9006

这里返回的是所有的车站和对应缩写的数据

import re
import  requests
from pprint import pprint

url = 'https://kyfw.12306.cn/otn/resources/js/framework/station_name.js?station_version=1.9006'
response = requests.get(url,verify=False)
stations = re.findall(u'([\u4e00-\u9fa5]+)\|([A-Z]+)',response.text)
pprint(dict(stations),indent=4)

通过正则表达式，匹配所有的车站和对应的缩写，并转换成字典格式。

因为所有的地点和对应的缩写，变化的可能性不大，将上面获取的字典数据保存为一个文件，备用。

下面继续看查询车票时返回的数据：



通过这个json数据，可以获取到所有车次的信息，关键在于如何对现有的json进行解析，获取到所有车次的信息？ps：网上看了其他人写的一些查询车票信息的脚本，之前的车票信息每个字段对应一个值，标准的json形式：



现在的json数据每列车的信息都在一串字符串中，思来想去，没有找到什么好办法，唯一发现的规律就是每个字段用“|”分割，现在的做法就是将该段字符串以“|”分割成一个列表，从列表中去数需要的字段对应的位置。

主要代码如下：

def trains(self):
for raw_train in self.available_trains:
raw_train_list = raw_train.split('|')
train_no = raw_train_list[3]
initial = train_no[0].lower()
duration = raw_train_list[10]
if initial in self.options:
train = [
train_no,
'\n'.join([Fore.LIGHTGREEN_EX + self.available_place[raw_train_list[6]] + Fore.RESET,
Fore.LIGHTRED_EX + self.available_place[raw_train_list[7]] + Fore.RESET]),
'\n'.join([Fore.LIGHTGREEN_EX + raw_train_list[8] + Fore.RESET,
Fore.LIGHTRED_EX + raw_train_list[9] + Fore.RESET]),
duration,
raw_train_list[-4] if raw_train_list[-4] else '--',
raw_train_list[-5] if raw_train_list[-5] else '--',
raw_train_list[-14] if raw_train_list[-14] else '--',
raw_train_list[-12] if raw_train_list[-12] else '--',
raw_train_list[-7] if raw_train_list[-7] else '--',
raw_train_list[-6] if raw_train_list[-6] else '--',
raw_train_list[-9] if raw_train_list[-9] else '--',
]
yield train

最后附上代码地址https://github.com/lexyhp/12306

关于如何从json数据中取出需要的字段，欢迎有更好方法的朋友留言，感谢！！