python3 实现火车票查询代码+知识点

一. prettytable

1. 简介：可以将输出内容按照表格方式整齐输出。

2. 安装：pip install prettytable

3. 导入：from prettytable import PrettyTable

4. 示例：

from prettytable import PrettyTable
table = PrettyTable(["id", "name"])
table.add_row([1, "lili"])
table.add_row([2, "rose"])
table.add_row([3, "nancy"])
table.add_row([4, "puppet"])
print(table)

效果：

5. 使用

5.1 创建表

pt = PrettyTable()

5.2 按行/列添加

pt.add_row()       # 行
pt.add_column()   # 列

6. 输出

6.1 直接输出

print(pt)

6.2 无表格框输出

print(pt.get_string())

二. split

1. 作用：切分字符串

2. 示例：

"2:3:4:5".split(":")
"|a|b|c".split("|")

结果

["2", "3", "4", "5"]
["", "a", "b", "c"]

三. \u4e00”和“\u9fa5

1. 简介：判断字符串中是否包含中文

“\u4e00”和“\u9fa5”是unicode编码，是unicode表中的汉字的开始和结束。[\u4e00-\u9fa5]+，[]表示内容，+代表至少出现一次。

四. print和pprint区别

1. 简介：

大致相同，但pprint打印更加完整，每行为一个数据结构，方便阅读，对于较长的数据，print打印都在一行，pprint分行打印

五. colorama

1. 简介：给字体着色

2. 安装：pip install colorama

3. 导入：from colorama import init, Fore, Back, Style

说明：
init（）：初始化，其中可以传入参数，autoreset 。默认值为 False ，如果设置为 True，它就会在每一次输出语句之后自动清空格式。
Fore：前景色（即字体颜色）
Back：背景色
Style：改变字体的显示模式。
三个属性可设置的参数如下：
Fore: BLACK, RED, GREEN, YELLOW, BLUE, MAGENTA, CYAN, WHITE, RESET.
Back: BLACK, RED, GREEN, YELLOW, BLUE, MAGENTA, CYAN, WHITE, RESET.
Style: DIM, NORMAL, BRIGHT, RESET_ALL

4. 示例

from colorama import init, Fore, Back, Styleinit()
print(Fore.LIGHTGREEN_EX + "some red text")
print(Back.RED + "and with a green background")
print(Style.BRIGHT + "and in dim text")
print(Style.RESET_ALL)
print("back to normal now!!")

效果

六. 正文

有了上面的基础铺垫，再看代码会简单很多
tickets.py

"""Train tickets query via command-line.火车票通过命令行查询。
Usage:tickets [-gdtkz] <from> <to> <date>Options:-h,--help   显示帮助菜单-g          高铁-d          动车-t          特快-k          快速-z          直达Example:tickets beijing shanghai 2020-01-25"""
import json
import colorama
from prettytable import PrettyTable
from colorama import init, Fore, Back, Style
from station import station
import requests
from colorama import init, Fore
from docopt import docoptinit()                                  # 使用colorama前的初始化def cli():arguments = docopt(__doc__)from_station = station.get(arguments['<from>'])        # station是个字典to_station = station.get(arguments['<to>'])date = arguments['<date>']# 构建url,输入车站名字，直接从station字典中获得相对应的字母代码url = 'https://kyfw.12306.cn/otn/leftTicket/queryO?leftTicketDTO.train_date={}' \'&leftTicketDTO.from_station={}&leftTicketDTO.to_station={}&purpose_codes=ADULT'.format(date, from_station, to_station)# print("这是网址:", url)options = ''.join([key for key, value in arguments.items() if value is True])headers = {'User-Agent': 'Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, ''like Gecko) Chrome/78.0.3904.87 Safari/537.36','Cookie': 'JSESSIONID=0FCBD07FA01EA1713255920101896AC0;'' route=c5c62a339e7744272a54643b3be5bf64;'' BIGipServerpool_passport=384631306.50215.0000; RAIL_EXPIRATION=1582955512068; ''RAIL_DEVICEID=qUANl2BZU-iRhsSMlPYc2Ut5AU-XbI_''XGAWLMc6T1MVYaIeLtvh3KoZuNsAM4fIIznGMn7Z5bx246WADUQl2C2AZhqvnwJmYH-3DRnXMvWRItWlMtwfayV6YckM8VU7UMhb5''R6cJChWUeBOyEDz-LP_GYdUBPGZ7; ''_jc_save_fromStation=%u5317%u4EAC%2CBJP; _jc_save_toStation=%u4E0A%u6D77%2CSHH; ''_jc_save_toDate=2020-02-26; ''_jc_save_wfdc_flag=dc; BIGipServerotn=2062024970.64545.0000; _jc_save_fromDate=2020-02-27'}r = requests.get(url, verify=False, headers=headers)    # 添加verify=False参数不验证证书available_trains = r.json()['data']['result']           # 返回列表station_map = r.json()['data']['map']                  # 返回的是字典形式，地名和代号一一对应,下面要用到# print(station_map)  # {'AOH': '上海虹桥', 'VNP': '北京南', 'SNH': '上海南', 'BJP': '北京', 'SHH': '上海'}# 做表格，设置表头table = PrettyTable(["车次", "出发站", "到达站", "出发时间", "到达时间", "历时", "特等座", "一等座", "二等座", "软卧","硬卧", "软座", "硬座", "无座"])for data in available_trains:list = data.split("|")        # 按照“|” 切分# print(list)if list[1] == '列车停运':      # 以列车停运开头的那几行均不是continueline_nun = list[3]    # 车次from_sta = list[6]to_sta = list[7]start_time = list[8]stop_time = list[9]cost_time = list[10]TDZ = list[32] or "--"   # 特等座     # 如果没有信息，用--显示YDZ = list[31] or "--"  # 一等座EDZ = list[30] or "--"  # 二等座RW = list[23] or "--"  # 软卧YW = list[28] or "--"  # 硬卧RZ = list[27] or "--"  # 软座YZ = list[29] or "--"  # 硬座WZ = list[26] or "--"  # 无座# 表格添加列 按照表格方式整齐地输出+对输出的每个字符和“|”进行着色table.add_row([line_nun, Fore.LIGHTRED_EX+station_map[from_sta]+Fore.RESET,Fore.LIGHTBLUE_EX + station_map[to_sta]+Fore.RESET,Fore.LIGHTRED_EX+start_time+Fore.RESET, Fore.LIGHTGREEN_EX+stop_time+Fore.RESET,cost_time, TDZ, YDZ, EDZ, RW, YW, RZ, YZ, WZ])print(table)if __name__ == '__main__':cli()

parse_station.py

"""提取地址及其对应的代号"""
import re
import requests
from pprint import pprint# 地址和代号网址
url = 'https://kyfw.12306.cn/otn/resources/js/framework/station_name.js?station_version=1.8955'
# 获取url
response = requests.get(url, verify=False)    # 移除SSL认证
# 正则提取中文和代号  格式：列表 内包含多个元组，元组形式 （‘上海’，‘SHH’）
station = re.findall(u'([\u4e00-\u9fa5]+)\|([A-Z]+)', response.text)   # 解析页面中中文和大写字母
# 定义打印信息 缩进4个空格
pprint(dict(station), indent=4)

1. 其实全程用到两个url
url1：根据输入的“起始站”+“终点站”+“日期”来查询结果的页面，所以需要模拟这个url。
url2：解析地名和地名代号的页面。
2. available_trains的打印结果如下

“列车停运”部分不是我们需要的，黄色框选的地方才是
3.for data in available_trains:
list = data.split("|")
print(list) 将available_trains分割后，打印结果如下：

其余不解释了，代码很简短。
最终效果（都能对的上，配色可以自己调，green/red/blue）

文件下载
https://github.com/Charging01/ticket-query.git