利用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 相关文章推荐
Python中使用ConfigParser解析ini配置文件实例
Aug 30 Python
Python实现Const详解
Jan 27 Python
python中for语句简单遍历数据的方法
May 07 Python
Python 绘图和可视化详细介绍
Feb 11 Python
Python 调用Java实例详解
Jun 02 Python
python根据unicode判断语言类型实例代码
Jan 17 Python
python编程测试电脑开启最大线程数实例代码
Feb 09 Python
对Python3 pyc 文件的使用详解
Feb 16 Python
基于Pytorch SSD模型分析
Feb 18 Python
Python如何把Spark数据写入ElasticSearch
Apr 18 Python
Selenium自动化测试工具使用方法汇总
Jun 12 Python
基于Django集成CAS实现流程详解
Nov 28 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
如何将数据从文本导入到mysql
2006/10/09 PHP
php中如何同时使用session和cookie来保存用户登录信息
2013/07/05 PHP
WordPres对前端页面调试时的两个PHP函数使用小技巧
2015/12/22 PHP
golang、python、php、c++、c、java、Nodejs性能对比
2017/03/12 NodeJs
Django 标签筛选的实现代码(一对多、多对多)
2018/09/05 PHP
php7 list()、session及其他模块的修改实例分析
2020/05/25 PHP
JS判断客服QQ号在线还是离线状态的方法
2015/01/13 Javascript
Javascript中使用parseInt函数需要注意的问题
2015/04/02 Javascript
JQuery给网页更换皮肤的方法
2015/05/30 Javascript
JavaScript实现动态删除列表框值的方法
2015/08/12 Javascript
详解AngularJS中的filter过滤器用法
2016/01/04 Javascript
一个用jquery写的判断div滚动条到底部的方法【推荐】
2016/04/29 Javascript
深入理解事件冒泡(Bubble)和事件捕捉(capture)
2016/05/28 Javascript
js实现砖头在页面拖拉效果
2020/11/20 Javascript
Node.js命令行/批处理中如何更改Linux用户密码浅析
2018/07/22 Javascript
微信小程序使用GoEasy实现websocket实时通讯
2020/05/19 Javascript
python实现简单的计时器功能函数
2015/03/14 Python
Google开源的Python格式化工具YAPF的安装和使用教程
2016/05/31 Python
Python的SQLalchemy模块连接与操作MySQL的基础示例
2016/07/11 Python
Python3 replace()函数使用方法
2018/03/19 Python
Python爬虫工程师面试问题总结
2018/03/22 Python
python itchat给指定联系人发消息的方法
2019/06/11 Python
Python:Numpy 求平均向量的实例
2019/06/29 Python
Django多数据库的实现过程详解
2019/08/01 Python
Python生态圈图像格式转换问题(推荐)
2019/12/02 Python
基于PyTorch的permute和reshape/view的区别介绍
2020/06/18 Python
高考考python编程是真的吗
2020/07/20 Python
python线程优先级队列知识点总结
2021/02/28 Python
7款设计巧妙的css3飘带状3D立体效果的导航菜单和表单窗口
2013/02/04 HTML / CSS
美国女孩服装购物网站:Justice
2017/03/04 全球购物
英国排名第一的礼品体验公司:Red Letter Days
2018/08/16 全球购物
Burt’s Bees英国官网:世界领先的天然个人护理品牌
2020/08/17 全球购物
函授毕业自我鉴定
2013/12/19 职场文书
2015年元旦活动总结
2014/05/09 职场文书
群众路线领导班子四风对照检查材料
2014/09/27 职场文书
初中语文教学随笔
2015/08/15 职场文书