利用Python实现命令行版的火车票查看器


Posted in Python onAugust 05, 2016

接口设计

一个应用写出来最终是要给人使用的,哪怕只是给你自己使用。所以,首先应该想想你希望怎么使用它?让我们先给这个小应用起个名字吧,既然及查询票务信息,那就叫它tickets好了。我们希望用户只要输入出发站,到达站以及日期就让就能获得想要的信息,所以tickets应该这样被使用:

$ tickets from to date

另外,火车有各种类型,高铁、动车、特快、快速和直达,我们希望可以提供选项只查询特定的一种或几种的火车,所以,我们应该有下面这些选项:

     -g 高铁

     -d 动车

     -t 特快

     -k 快速

     -z 直达

这几个选项应该能被组合使用,所以,最终我们的接口应该是这个样子的:

$ tickets [-gdtkz] from to date

接口已经确定好了,剩下的就是实现它了。

开发环境

写Python程序的一个良好实践是使用virtualenv这个工具建一个虚拟的环境。我们的程序使用Python3开发,下面在你的工作目录下建一个文件夹tickets,进去创建一个虚拟环境:

$ virtualenv -p /usr/bin/python3 venv

通过下面的命令激活它:

$ . venv/bin/activate

解析参数

Python有很多写命令行应用的工具,如argparse, docopt, options…这里,我们选用docopt这个简单易用的工具,我们先安装它:

$ pip3 install docopt

docopt可以按我们在文档字符串中定义的格式来解析参数,在tickets.py中:

# coding: utf-8

"""Train tickets query via command-line.

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

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

Example:
  tickets 南京 北京 2016-07-01
  tickets -dg 南京 北京 2016-07-01
"""
from docopt import docopt

def cli():
  """command-line interface""" 
  arguments = docopt(__doc__)
  print(arguments)

if __name__ == '__main__': 
  cli()

下面我们运行一下这个程序:

$ python3 tickets.py 上海 北京 2016-07-01

我们得到下面的参数解析结果:

{‘-d': False, ‘-g': False, ‘-k': False, ‘-t': False, ‘-z': False, ‘<date>': ‘2016-07-01', ‘<from>': ‘上海', ‘<to>': ‘北京'}

获取数据

参数已经解析好了,下面就是如何获取数据了,这也是最主要的部分。首先我们打开12306,进入余票查询页面,如果你使用chrome,那么按F12打开开发者工具,选中Network一栏,在查询框钟我们输入上海到北京,日期2016-07-01, 点击查询,我们在调试工具发现,查询系统实际上请求了这个URL

https://kyfw.12306.cn/otn/lcxxcx/query?purpose_codes=ADULT&queryDate=2016-07-01&from_station=SHH&to_station=BJP

并且返回的是Json格式的数据!接下来问题就简单了,我们只需要构建请求URL然后解析返回的Json数据就可以了。但是我们发现,URL里面from_stationto_station并不是汉字,而是一个代号,而用户输入的是汉字,我们要如何获取代号呢?我们打开网页源码看看有没有什么发现。

啊哈!果然,我们在网页里面找到了这个链接:点我。这里面貌似是包含了所有车站的中文名,拼音,简写和代号等信息, 我们在项目目录下将它保存为stations.html。但是这些信息挤在一起,而我们只想要中文名和大写字母的代号信息,怎么办呢?

BINGO!正则表达式,我们写个小脚本来匹配提取出想要的信息吧, 在parse.py中:

# coding: utf-8
import re
from pprint import pprint

with open('stations.html', 'r') as f:
  text = f.read()
  stations = re.findall(u'([\u4e00-\u9fa5]+)\|([A-Z]+)', text)
  pprint(dict(stations), indent=4)

我们运行这个脚本,它将以字典的形式返回所有车站和它的大写字母代号, 我们将结果重定向到stations.py中,

$ python3 parse.py > stations.py

我们为这个字典加名字,stations, 最终,stations.py文件是这样的:

stations = {  
  '一间堡': 'YJT',
  '一面坡': 'YPB',
   ... 
  '龙镇': 'LZA',
  '龙骨甸': 'LGM'
}

现在,用户输入车站的中文名,我们就可以直接从这个字典中获取它的字母代码了:

...
from stations import stations

def cli():
  arguments = docopt(__doc__)
  from_staion = stations.get(arguments['<from>'])
  to_station = stations.get(arguments['<to>'])
  date = arguments['<date>']
  # 构建URL
  url = 'https://kyfw.12306.cn/otn/lcxxcx/query?purpose_codes=ADULT&queryDate={}&from_station={}&to_station={}'.format(
    date, from_staion, to_station
)

万事俱备,下面我们来请求这个URL获取数据吧!这里我们使用requests这个库, 先安装它:

$ pip3 install requests

它提供了非常简单易用的接口,

...
import requests

def cli():
  ...
  # 添加verify=False参数, 不验证证书
  r = requests.get(url, verify=False)
  print(r.json())

从结果中,我们可以观察到,与车票有关的信息需要进一步提取:

def cli():
  ...
  r = requsets.get(url);
  rows = r.json()['data']['datas']

显示结果

数据已经获取到了,剩下的就是提取我们要的信息并将它显示出来。prettytable这个库可以让我们它像MySQL数据库那样格式化显示数据。

$ pip3 install prettytable

这样使用它:

...
from prettytable import PrettyTable

def cli():
  ...
  headers = '车次 车站 时间 历时 商务 一等 二等 软卧 硬卧 软座 硬座 无座'.split()
  pt = PrettyTable()
  pt._set_field_names(headers)
  for row in rows:
  # 从row中根据headers过滤信息, 然后调用pt.add_row()添加到表中
    ...
  print(pt)

总结

以上就是用Python写一个命令行版的火车票查看器的全部内容,对于Python学习是个不错的例子,希望能对大家学习Python有所帮助。

Python 相关文章推荐
pycharm 使用心得(七)一些实用功能介绍
Jun 06 Python
django 自定义用户user模型的三种方法
Nov 18 Python
Python中pip安装非PyPI官网第三方库的方法
Jun 02 Python
Python中使用urllib2模块编写爬虫的简单上手示例
Jan 20 Python
详解python之配置日志的几种方式
May 22 Python
如何优雅地处理Django中的favicon.ico图标详解
Jul 05 Python
解决安装python库时windows error5 报错的问题
Oct 21 Python
python 实现手机自动拨打电话的方法(通话压力测试)
Aug 08 Python
Python 用三行代码提取PDF表格数据
Oct 13 Python
Python安装tar.gz格式文件方法详解
Jan 19 Python
Python批量修改xml的坐标值全部转为整数的实例代码
Nov 26 Python
python字符串的多行输出的实例详解
Jun 08 Python
Python处理JSON数据并生成条形图
Aug 05 #Python
用Python解决计数原理问题的方法
Aug 04 #Python
快速入手Python字符编码
Aug 03 #Python
Python采用Django制作简易的知乎日报API
Aug 03 #Python
利用Python实现图书超期提醒
Aug 02 #Python
Python正规则表达式学习指南
Aug 02 #Python
Python实现SMTP发送邮件详细教程
Mar 02 #Python
You might like
PHP正则的Unknown Modifier错误解决方法
2010/03/02 PHP
YII2自动登录Cookie总是失效的解决方法
2017/06/28 PHP
Yii框架 session 数据库存储操作方法示例
2019/11/18 PHP
TP5框架实现的数据库备份功能示例
2020/04/05 PHP
漂亮的widgets,支持换肤和后期开发新皮肤
2007/04/23 Javascript
简单的JS多重继承示例
2008/03/13 Javascript
JavaScript中通过闭包解决只能取得包含函数中任何变量最后一个值的问题
2010/08/12 Javascript
Bootstrap实现input控件失去焦点时验证
2016/08/04 Javascript
nodejs入门教程三:调用内部和外部方法示例
2017/04/24 NodeJs
Vue.js数据绑定之data属性
2017/07/07 Javascript
让bootstrap的carousel支持滑动滚屏的实现代码
2017/11/27 Javascript
为nuxt项目写一个面包屑cli工具实现自动生成页面与面包屑配置
2019/09/29 Javascript
关于引入vue.js 文件的知识点总结
2020/01/28 Javascript
[05:56]第十六期——新进3大C之小兔基
2014/06/24 DOTA
图文详解WinPE下安装Python
2016/05/17 Python
Python中列表和元组的使用方法和区别详解
2020/12/30 Python
在Pycharm terminal中字体大小设置的方法
2019/01/16 Python
python控制nao机器人身体动作实例详解
2019/04/29 Python
用uWSGI和Nginx部署Flask项目的方法示例
2019/05/05 Python
教你一步步利用python实现贪吃蛇游戏
2019/06/27 Python
python Django编写接口并用Jmeter测试的方法
2019/07/31 Python
详解Python Opencv和PIL读取图像文件的差别
2019/12/27 Python
django 前端页面如何实现显示前N条数据
2020/03/16 Python
python:解析requests返回的response(json格式)说明
2020/04/30 Python
Python 列表中的修改、添加和删除元素的实现
2020/06/11 Python
如何解决安装python3.6.1失败
2020/07/01 Python
python能做哪些生活有趣的事情
2020/09/09 Python
利用Python如何画一颗心、小人发射爱心
2021/02/21 Python
Maxpeedingrods美国:高性能汽车零件
2020/02/14 全球购物
机电一体化自荐信
2013/12/10 职场文书
计算机专业学生求职信分享
2013/12/15 职场文书
领导干部群众路线教育实践活动个人对照检查材料
2014/09/23 职场文书
2015年统计员个人工作总结
2015/07/23 职场文书
从np.random.normal()到正态分布的拟合操作
2021/06/02 Python
Java 实现限流器处理Rest接口请求详解流程
2021/11/02 Java/Android
MYSQL如何查看进程和kill进程
2022/03/13 MySQL