美文网首页
python实现12306车票查询

python实现12306车票查询

作者: lexyhp | 来源:发表于2017-05-03 11:06 被阅读0次

    看到网上有很多火车票查询的小脚本,参考一下,发现很多都已经不能再运行了,据说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] <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

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

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

    相关文章

      网友评论

          本文标题:python实现12306车票查询

          本文链接:https://www.haomeiwen.com/subject/jkektxtx.html