使用python脚本实现查询火车票工具

Python  /  管理员 发布于 7年前   148

使用python脚本实现查询火车票信息的效果图如下：

实现的代码：

# coding: utf-8"""命令行火车票查看器Usage: tickets [-gdtkz] Options: -h,--help 显示帮助菜单 -g   高铁 -d   动车 -t   特快 -k   快速 -z   直达Example: tickets 北京 上海 2016-10-10 tickets -dg 成都 南京 2016-10-10"""import jsonimport requestsimport prettytablefrom docopt import docoptfrom colorama import init, Foreclass CollectInfo: def __init__(self):  self.qurey_ret = []  self.header = ['车次信息', '发/到时间', '发/到站', '历时', '票价', '余票'] # 获取车次相关的所有信息 def query_html_ret(self, query_args):  url = 'http://api.12306.com/v1/train/trainInfos?arrStationCode={to_station}&deptDate={date}\    &deptStationCode={source_station}&findGD=false'.format(to_station=query_args['to_station'],      source_station=query_args['source_station'],      date=query_args['date'])  row_ret = requests.get(url)  return row_ret.json() # 解析获取到的结果 def paser_ret(self, row_ret):  trains_info = row_ret['data']['trainInfos']  for info in trains_info:   row_info = []   # 获取车次信息   row_info.append('\n' + info['trainCode'])   # 获取车次到站时间信息   row_info.append('\n' + '\n'.join([Fore.GREEN + info['deptTime']+ Fore.RESET,Fore.RED + info['arrTime']+ Fore.RESET]))   # 获取车次站点名称   row_info.append('\n' + '\n'.join([Fore.GREEN + info['deptStationName'] + Fore.RESET,Fore.RED + info['arrStationName']+ Fore.RESET]))   # 获取车次到达站点所需时间   row_info.append('\n' + info['runTime'])   # 获取票价以及余票信息   seat_price = []   seat_num = []   for seat in info['seatList']:    seat_price.append(seat['seatName'] + '：' + seat['seatPrice'])    if int(seat['seatNum']) > 10:     ticknum = Fore.GREEN + seat['seatNum'] + Fore.RESET    else:     ticknum = seat['seatNum']    seat_num.append(ticknum)   row_info.append('\n'.join(seat_price))   row_info.append('\n'.join(seat_num))   self.qurey_ret.append(row_info)   self.qurey_ret.append([' ', ' ', ' ', ' ', ' ', ' '])  return self.qurey_ret def show_with_table(self):  ticket_table = prettytable.PrettyTable()  ticket_table.field_names = self.header  for row in self.qurey_ret:   if len(row) == 0:    continue   ticket_table.add_row(row)  return ticket_tabledef main(): arguments = docopt(__doc__) query_args = {} init() # 获取所有站点信息(stations.txt信息通过 函数获取) # https: // kyfw.12306.cn / otn / resources / js / framework / station_name.js?station_version = 1.8971 f = open('stations.txt', 'r') info = f.read() stations_info = json.loads(info) # 从所有站点信息中获取所要查询站点的代码信息 query_args['to_station'] = stations_info[arguments['']] query_args['source_station'] = stations_info[arguments['']] query_args['date'] = arguments[''] # 向12306查询，得到跟车次相关的所有信息 collect_train = CollectInfo() row_ret = collect_train.query_html_ret(query_args) collect_train.paser_ret(row_ret) table = collect_train.show_with_table() print(table) if __name__ == '__main__':  main()

总结

以上所述是小编给大家介绍的使用python脚本查询火车票工具，希望对大家有所帮助，如果大家有任何疑问请给我留言，小编会及时回复大家的。在此也非常感谢大家对站的支持！