看到网上有很多火车票查询的小脚本,参考一下,发现很多都已经不能再运行了,据说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] <from> <to> <date>
- <from> 出发地
- <to> 目的地
- <date> 日期
- [-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>
“< >”中的是参数,如本例中的<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数据中取出需要的字段,欢迎有更好方法的朋友留言,感谢!!