"""
Usage:test [-gdtkz] <from> <to> <date>Options:-h,--help   显示帮助菜单-g          高铁-d          动车-t          特快-k          快速-z          直达Example:tickets -gdt beijing shanghai 2016-08-25
"""import docoptargs = docopt.docopt(__doc__)
print(args)# 上面 """ """ 包含中的：

#Usage:
#   test [-gdtkz] <from> <to> <date>#是必须要的 test 是可以随便写的，不影响解析

 1 #coding=utf-8
 2
 3 import requests
 4 import re
 5 from pprint import pprint
 6
 7
 8 def get_stations():
 9     # 7@cqn|重庆南|CRW|chongqingnan|cqn|
10     url = 'https://kyfw.12306.cn/otn/resources/js/framework/station_name.js?station_version=1.8968'
11     r = requests.get(url,verify=False)
12     patter = re.compile('([A-Z]+)\|([a-z]+)')
13     items = dict(re.findall(patter,r.text))
14     stations = dict(zip(items.values(),items.keys()))
15     # for key in stations:
16     #     print("{0}-->{1}".format(key,stations[key]))
17     pprint(stations,indent=4)
18
19 if __name__ == '__main__':
20     get_stations()

#coding=utf-8from prettytable import PrettyTableclass TrainCollection(object):"""解析列车信息"""# 显示车次、出发/到达站、 出发/到达时间、历时、一等坐、二等坐、软卧、硬卧、硬座header = '序号 车次 出发站/到达站 出发时间/到达时间 历时 商务座 一等座 二等座 软卧 硬卧 硬座 无座'.split()def __init__(self,rows,traintypes):self.rows = rowsself.traintypes = traintypesdef _get_duration(self,row):"""获取车次运行的时间"""duration = row.get('lishi').replace(':','小时') + '分'if duration.startswith('00'):return duration[4:]elif duration.startswith('0'):return duration[1:]return duration@propertydef trains(self):result = []flag = 0for row in self.rows:if row['station_train_code'][0] in self.traintypes:flag += 1train = [# 序号
                    flag,# 车次row['station_train_code'],# 出发、到达站点'/'.join([row['from_station_name'],row['to_station_name']]),# 成功、到达时间'/'.join([row['start_time'],row['arrive_time']]),# duration 时间
                    self._get_duration(row),# 商务座row['swz_num'],# 一等座row['zy_num'],# 二等座row['ze_num'],# 软卧row['rw_num'],# 硬卧row['yw_num'],# 硬座row['yz_num'],# 无座row['wz_num']]result.append(train)return resultdef print_pretty(self):"""打印列车信息"""pt = PrettyTable()pt._set_field_names(self.header)for train in self.trains:pt.add_row(train)print(pt)if __name__ == '__main__':t = TrainCollection()

 1 """Train tickets query via command-line.
 2
 3 Usage:
 4     tickets [-gdtkz] <from> <to> <date>
 5
 6 Options:
 7     -h,--help   显示帮助菜单
 8     -g          高铁
 9     -d          动车
10     -t          特快
11     -k          快速
12     -z          直达
13
14 Example:
15     tickets -gdt beijing shanghai 2016-08-25
16 """
17 import requests
18 from docopt import docopt
19 from stations import stations
20 # from pprint import pprint
21 from myprettytable import TrainCollection
22
23 class SelectTrain(object):
24
25     def __init__(self):
26         """
27         获取命令行输入的参数
28         """
29         self.args = docopt(__doc__)#这个是获取命令行的所有参数，返回的是一个字典
30
31
32     def cli(self):
33         """command-line interface"""
34         # 获取 出发站点和目标站点
35         from_station = stations.get(self.args['<from>']) #出发站点
36         to_station = stations.get(self.args['<to>']) # 目的站点
37         leave_time = self._get_leave_time()# 出发时间
38
39         url = 'https://kyfw.12306.cn/otn/lcxxcx/query?purpose_codes=ADULT&queryDate={0}&from_station={1}&to_station={2}'.format(
40             leave_time,from_station,to_station)# 拼接请求列车信息的Url
41
42         # 获取列车查询结果
43         r = requests.get(url,verify=False)
44         traindatas = r.json()['data']['datas'] # 返回的结果，转化成json格式，取出datas，方便后面解析列车信息用
45
46         # 解析列车信息
47         traintypes = self._get_traintype()
48         views = TrainCollection(traindatas,traintypes)
49         views.print_pretty()
50
51     def _get_traintype(self):
52         """
53         获取列车型号，这个函数的作用是的目的是：当你输入 -g 是只是返回 高铁，输入 -gd 返回动车和高铁，当不输参数时，返回所有的列车信息
54         """
55         traintypes = ['-g','-d','-t','-k','-z']
56         # result = []
57         # for traintype in traintypes:
58         #     if self.args[traintype]:
59         #         result.append(traintype[-1].upper())
60
61         trains = [traintype[-1].upper() for traintype in traintypes if self.args[traintype]]
62         if trains:
63             return trains
64         else:
65             return ['G','D','T','K','Z']
66
67     def _get_leave_time(self):
68         """
69         获取出发时间，这个函数的作用是为了：时间可以输入两种格式：2016-10-05、20161005
70         """
71         leave_time = self.args['<date>']
72         if len(leave_time) == 8:
73             return '{0}-{1}-{2}'.format(leave_time[:4],leave_time[4:6],leave_time[6:])
74
75         if '-' in leave_time:
76             return leave_time
77
78
79 if __name__ == '__main__':
80     cli = SelectTrain()
81     cli.cli()