Python3实现火车票查询工具

Python 实现火车票查询工具

一. 实验介绍

　　　　通过python3实现一个简单的命令行版本的火车票查询工具，用实际中的例子会更感兴趣，不管怎么样，既练习了又可以自己使用。

1.  知识点：　

• Python 基础知识的综合运用
• docopt, requests, colorama, prettytable 库的使用
• setuptools 的使用

2. 效果截图：

二. 接口设计

　　接口我们就按照12306官网的查询格式如下：

　   程序名： pickets

　　查询方式： 程序名 出发地 目的地 出发日    

　　火车类型： 

　　　　　　-g　　　　高铁

　　　　　　-d　　　　动车

　　　　　　-t　　　　特快

　　　　　　-k　　　　快速

　　　　　　-z　　　　直达

　　最终组合接口为：

　　　　python  pickets.py  [-gdtkz]  from  to   date

三. 代码实现

　　根据接口设计需要安装的库：

G:Python项目实战pickets>pip3 install requests prettytable docopt colorama

 　　requests:  使用Python访问HTTP资源的必备库　　　　　　　　　　　 

详细使用请访问：http://www.cnblogs.com/python-nameless/p/6013647.html

　　 prettytable: 可以将打印内容格式化为像MySQL一样的输出　　　　　　

详细使用请访问：http://www.cnblogs.com/python-nameless/p/6013698.html

　　 docopt：命令行参数解析工具　　　　　　　　　　　　　　　　　　　  

详细使用请访问：https://github.com/docopt/docopt

  　  colorama: 命令行着色工具

1. 解析参数

　　docopt模块可以按照我们自己在文档字符串定义的格式来解析参数，如下：

#!/usr/bin/env python
# -*- coding: utf-8 -*-
# __Author__: Hong

"""火车票查询工具

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

注意：Usage为固定词汇，其他报错。

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

Example:
    tickets 北京 大同 2016-08-28
    tickets -t 北京 大同 2016-08-28
"""
from docopt import docopt
from requests import request
import prettytable
import colorama


def cli():
    """command-line interface"""

    arguments = docopt(__doc__)
    print(arguments)

if __name__ == '__main__':
    cli()

由上面的程序中，docopt会根据我们在docstring中定义的格式解析参数并返回一个字典（arguments）,例如查询北京到大同的特快如下：

G:Python项目实战pickets>python pickets.py -tk  北京 大同 2016-10-30

 参数解析结果：

G:Python项目实战pickets>python pickets.py -tk  北京 大同 2016-10-30
{'-d': False,
 '-g': False,
 '-k': True,
 '-t': True,
 '-z': False,
 '<date>': '2016-10-30',
 '<from>': '北京',
 '<to>': '大同'}

2. 获取数据

　　参数解析好以后，我们可以通过12306官网获取我们需要的数据，IE浏览器打开12306官网的查票系统，输入出发地，目的地，日期，查询，通过F12开发者模式找到请求地址(刷新一次)：

通过此请求URL，我们可以看到有我们需要的信息，或者直接点击参数查看：

　　可以看到参数为：出发地   查询类型(成人，学生)   日期  目的地

通过正文可以看到：返回一个json格式的数据，即一个字典，里边包括了各个站点的中文名和站点代号，我们可以利用这个接口来获取。

打开此页的源码(在页面右击选择查看源码)：

查看一下具体内容:

复制此链接前边需要加一级域名https://kyfw.12306.cn打开, 如下：

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

station_names是一个很长的字符串，里边包括了所有车站中文名，拼音，简写，代号等，这里可以看到有很多信息，我们提取我们需要的信息中文字母和代号信息，我们用正则表达式来提取：

写一个简单的脚本抓取(parse_station.py)：

#!/usr/bin/env python
# -*- coding: utf-8 -*-
# __Author__: hong


import re
import requests
from pprint import pprint
# pprint: 格式化输出
url = 'https://kyfw.12306.cn/otn/resources/js/framework/station_name.js?station_version=1.8971'
#获取URL
response = requests.get(url, verify=False)
#正则提取中文字母和代号
stations = re.findall(u'([u4e00-u9fa5]+)|([A-Z]+)', response.text)
# indent:定义打印信息的缩进为4个空格
pprint(dict(stations), indent=4)

提出数据为一个dict 重定向到一个文件里(stations.py)定义字典名如下:

注意:

　　字典前边定义函数名为stations，默认使用utf-8中文显示乱码，可以切换为GBK。