python实现12306车票查询

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

环境

  • Mac osx
  • python3.6
  • pycharm

效果图

预览效果

编码

  1. 安装脚本用到的模块

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

这是个对终端输出的文本进行着色的模块,直接上代码图(官方示例):

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数据中取出需要的字段,欢迎有更好方法的朋友留言,感谢!!

展开阅读全文

没有更多推荐了,返回首页