利用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爬虫入门教程--优雅的HTTP库requests(二)
May 25 Python
Python使用openpyxl读写excel文件的方法
Jun 30 Python
python先序遍历二叉树问题
Nov 10 Python
python使用fork实现守护进程的方法
Nov 16 Python
使用sklearn之LabelEncoder将Label标准化的方法
Jul 11 Python
详解关于Django中ORM数据库迁移的配置
Oct 08 Python
python 阶乘累加和的实例
Feb 01 Python
python set内置函数的具体使用
Jul 02 Python
使用python turtle画高达
Jan 19 Python
在pycharm中实现删除bookmark
Feb 14 Python
Python tkinter之ComboBox(下拉框)的使用简介
Feb 05 Python
Python就将所有的英文单词首字母变成大写
Feb 12 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
IP138 IP地址查询小偷实现代码
2010/02/15 PHP
PHP使用curl函数发送Post请求的注意事项
2016/11/26 PHP
PHP+Ajax实现上传文件进度条动态显示进度功能
2018/06/04 PHP
JS Array对象入门分析
2008/10/30 Javascript
Javascript变量函数浅析
2011/09/02 Javascript
jQuery插件分享之分页插件jqPagination
2014/06/06 Javascript
Node.js中创建和管理外部进程详解
2014/08/16 Javascript
Jquery插件实现点击获取验证码后60秒内禁止重新获取
2015/03/13 Javascript
JS实现弹性漂浮效果的广告代码
2015/09/02 Javascript
Javascript BOM学习小结(六)
2015/11/26 Javascript
javascript实现C语言经典程序题
2015/11/29 Javascript
BootStrap中按钮点击后被禁用按钮的最佳实现方法
2016/09/23 Javascript
微信小程序 登陆流程详细介绍
2017/01/17 Javascript
jQuery选取所有复选框被选中的值并用Ajax异步提交数据的实例
2017/08/04 jQuery
浅谈Koa服务限流方法实践
2017/10/23 Javascript
Vue的实例、生命周期与Vue脚手架(vue-cli)实例详解
2017/12/27 Javascript
vue对storejs获取的数据进行处理时遇到的几种问题小结
2018/03/20 Javascript
JS使用栈判断给定字符串是否是回文算法示例
2019/03/04 Javascript
详解微信小程序之一键复制到剪切板
2019/04/24 Javascript
原生JavaScript实现弹幕组件的示例代码
2020/10/12 Javascript
python 网络编程常用代码段
2016/08/28 Python
Python3.6实现带有简单界面的有道翻译小程序
2019/04/16 Python
PyQt5 QTableView设置某一列不可编辑的方法
2019/06/25 Python
Pandas0.25来了千万别错过这10大好用的新功能
2019/08/07 Python
PyTorch里面的torch.nn.Parameter()详解
2020/01/03 Python
Python换行与不换行的输出实例
2020/02/19 Python
Python基于内置库pytesseract实现图片验证码识别功能
2020/02/24 Python
对象的序列化(serialization)类是面向流的,应如何将对象写入到随机存取文件中
2015/06/22 面试题
实习医生自我评价
2013/09/22 职场文书
职业生涯规划书的格式
2013/12/29 职场文书
班主任经验交流会主持词
2014/04/01 职场文书
2015年出纳工作总结与计划
2015/05/18 职场文书
海底两万里读书笔记
2015/06/26 职场文书
2019个人工作态度自我评价
2019/04/24 职场文书
如何Tomcat中使用ipv6地址
2022/05/06 Servers
Win11 25163.1010更新补丁KB5016904推送,测试服务验证管道(附更新修复汇总)
2022/07/23 数码科技