-
Notifications
You must be signed in to change notification settings - Fork 15
/
Copy pathaccess12306.py
294 lines (270 loc) · 14.9 KB
/
access12306.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
# coding: utf8
import time
import tkinter
import copy
import ui.LoginUI, ui.OrderConfirmUI, ui.QueryTrainUI
from core import login, order, query
from common.httpaccess import HttpTester
class AccessTrainOrderNetWork:
def __init__(self):
# 装载代理服务器地址
proxy_info = login.getGoAgentHost()
self.host_address = proxy_info.get('host', '')
self.host_type = proxy_info.get('type', '')
self.ht = HttpTester()
if self.host_address and self.host_type:
self.ht.addProxy(self.host_address, self.host_type)
self.ht.addCookiejar()
# 获取用户帐号信息
self.userInfo = login.getUserInfo()
# 获取用户设置的性能配置信息
self.performanceInfo = login.getPerformanceInfo()
# 装载列车站台编码
self.allStationCodes = {}
# 装载用户设定的常用联系人信息
self.contacts = {}
# 登录验证令牌
self.login_rand = ''
self.randImage = None
self.queryFrame = None
# 记录当前查询的参数值
self.currentSelectedParams = {}
# 车票预定时所填的参数数组
self.orderParams = None
# 读取用户设置的是否进行验证码的检查
self.check_rand_status = self.performanceInfo.get('check_rand', 'Y')
def access(self):
if self.userInfo:
login_result = login.getRandImageUrlAndCodeRand(self.ht)
rand_image_url = login_result.get('url', '')
self.login_rand = login_result.get('rand', '')
if rand_image_url:
self.randImage = ui.LoginUI.LoginFrame(rand_image_url)
self.randImage.loginButton.configure(command=self.processLoginCallBack)
self.randImage.randCode.bind("<Return>", self.processLoginCallBack)
self.randImage.show()
# 处理登录回调
def processLoginCallBack(self, event=None):
if self.randImage:
# 获取前台用户输入的输入证码
randCode = self.randImage.randCode.get()
if randCode:
loginResult = login.login(ht=self.ht, username=self.userInfo[0], password=self.userInfo[1],
randCode=randCode, rand=self.login_rand,
check_rand_status=self.check_rand_status)
if loginResult:
# 登录成功，关闭登框
self.randImage.quit()
# 得到性能配置中设定是否更新车站编码属性，默认是Y，表更新
update_stations = self.performanceInfo.get('update_stations', 'Y')
if update_stations and update_stations == "Y":
# 更新城市编码表
query.updateCityCode(self.ht)
# 装载列车站点编码
self.allStationCodes = query.getAllStationCodes()
# 获取默认的列车查询信息
defaultQueryParams = query.getDefaultQueryParams()
# 载入用户设定的所有联系人信息
self.contacts = order.getAllContacts(self.ht)
# 启动列车查询窗体
self.queryFrame = ui.QueryTrainUI.QueryTrainFrame(initQueryParams=defaultQueryParams)
self.queryFrame.selectButton.configure(command=self.queryTrainsCallBack)
self.queryFrame.show()
else:
print("登录失败，请再来一次")
self.randImage.refreshImg()
self.randImage.randCode.delete(0, tkinter.END)
else:
print("请输入验证码")
# 处理查询回调
def queryTrainsCallBack(self):
if not self.queryFrame:
return
from_station = self.queryFrame.fromStation.get()
to_station = self.queryFrame.toStation.get()
train_date = self.queryFrame.trainDate.get()
start_time = self.queryFrame.getSelectedTrainTime()
end_time = self.queryFrame.endTime.get()
trainClass = self.queryFrame.getSelectedTrainClass()
trainPassType = self.queryFrame.getChoiceTrainPassType()
trainNos = self.queryFrame.trainNo.get()
# 当时间未填时，设置为当前时间
if not train_date or train_date.strip() == '': train_date = time.strftime("%Y-%m-%d", time.localtime())
# 记录当前的查询条件
self.currentSelectedParams = {
'from_station': self.allStationCodes.get(from_station, ''),
'to_station': self.allStationCodes.get(to_station, ''),
'train_date': train_date,
'start_time': start_time,
'end_time': end_time,
'trainNos': trainNos,
'trainClass': trainClass,
'trainPassType': trainPassType,
'justShowCanBuy': self.queryFrame.showCanBuySwitch.get()
}
isAutoQuery = self.queryFrame.autoQuerySwitch.get() # 自动查询开关设置的值
# 执行查询，得到所有满足条件的列车
trains = query.queryTrains(self.ht, query_params=self.currentSelectedParams)
self.queryFrame.infoStartDateLabel.configure(
text="出发日期：%s %s->%s(共 %s 趟列车)" % (train_date, from_station, to_station, len(trains)))
if trains:
print("%s 查询成功!" % time.strftime('%H:%M:%S', time.localtime(time.time())))
self.queryFrame.resultTable.updateResult(trainDatas=trains, orderHandleFuc=self.orderTrainsCallBack)
# 判断查询的结果，如果没有可预订的列车，且开启了自动查询，则进行自动查询循环
canWebBuy = False
for train in trains:
tmp = train.get('canWebBuy', 'N')
if tmp == 'Y':
canWebBuy = True
break
if isAutoQuery and not canWebBuy:
# 获取配置文件中设定的查询间隙,默认是5秒
query_interval = self.performanceInfo.get('query_interval', 5)
self.queryFrame.selectButton.configure(state=tkinter.DISABLED, text='自动查询中')
print('自动查询开启，%s秒后进行下一波查询.' % query_interval)
self.queryFrame.root.after(int(query_interval) * 1000, self.queryTrainsCallBack)
else:
self.queryFrame.selectButton.configure(state=tkinter.NORMAL, text='查询')
# 处理列车预定按钮回调
def orderTrainsCallBack(self, selectStr='', row=0):
if selectStr:
# 对Http请求工具对象进行一次深度拷贝
# ht = copy.deepcopy(self.ht)
ht = HttpTester()
if self.host_address and self.host_type:
ht.addProxy(self.host_address, self.host_type)
ht.setCookiejar(self.ht.getCookiejar())
# 提交预定，获取初始化乘客确认页面内容
submitResult = order.submitOrderRequest(ht, selectStr, queryParams=self.currentSelectedParams)
if submitResult:
# 记录车票预订的HTML解析对象parser
parser = order.ParserConfirmPassengerInitPage(submitResult)
parser.feed(submitResult)
# 列车信息
trainInfo = parser.get_train_info()
# 验证码地址
img_rand_code_url = parser.get_img_code_url()
# 订单选座参数
passenger_params = parser.get_ticketInfoForPassengerForm()
# 判断得到的数据是否合法
if img_rand_code_url and trainInfo and self.contacts and passenger_params:
# self.queryFrame.quit() # 注销查询窗体
# self.queryFrame = None
comfirmFrame = ui.OrderConfirmUI.ConfirmPassengerFrame(contacts=self.contacts,
rand_image_url=img_rand_code_url,
train_info=trainInfo,
passenger_params=passenger_params)
comfirmFrame.backButton.configure(
command=lambda comfirmFrame=comfirmFrame: self.backToTrainQueryCallBack(comfirmFrame))
comfirmFrame.submitButton.configure(command=lambda comfirmFrame=comfirmFrame, httpAccessObj=ht,
parser=parser: self.submitOrderCallBack(
comfirmFrame, parser, httpAccessObj))
comfirmFrame.show()
else:
print('解析车票预订画面失败,详情查看当前目录下的submitResult.html文件.')
else:
print('预订画面获取失败!')
else:
print('非法的请求selectStr为空：%s' % selectStr)
# 处理车票预定窗体的重新选择回调
def backToTrainQueryCallBack(self, comfirmFrame):
# 注销车票预订窗体
comfirmFrame.quit()
# 新建查询窗体
#self.queryFrame = ui.QueryTrainUI.QueryTrainFrame(initQueryParams=self.currentSelectedParams)
#self.queryFrame.selectButton.configure(command=self.queryTrainsCallBack)
#self.queryFrame.show()
# 订单提交回调
def submitOrderCallBack(self, comfirmFrame, parser, httpAccessObj):
# 整合POST上传参数
passenger_info = comfirmFrame.getAllPassengerParams()
count = comfirmFrame.getCustomerCount()
# 订单提交页面隐藏参数
order_request_params = parser.get_order_request_params()
ticketInfoForPassengerForm = parser.get_ticketInfoForPassengerForm()
oldPassengers = []
passengerTickets = []
for i in range(count):
passengerTickets.append(passenger_info['passenger_' + str(i + 1) + '_seat'] + ",0," + passenger_info[
'passenger_' + str(i + 1) + '_ticket'] + "," + passenger_info[
'passenger_' + str(i + 1) + '_name'] + "," + passenger_info[
'passenger_' + str(i + 1) + '_cardtype'] + "," + passenger_info[
'passenger_' + str(i + 1) + '_cardno'] + "," + passenger_info[
'passenger_' + str(i + 1) + '_mobileno'] + ",Y")
# 这里最后一位暂时全部设定为成年人
# adult: "1",
# child: "2",
# student: "3",
# disability: "4"
oldPassengers.append(passenger_info['passenger_' + str(i + 1) + '_name'] + "," + passenger_info[
'passenger_' + str(i + 1) + '_cardtype'] + "," + passenger_info[
'passenger_' + str(i + 1) + '_cardno'] + "," + '1')
passengerTicketStr = "_".join(passengerTickets)
oldPassengersStr = "_".join(oldPassengers) + "_"
image_code = comfirmFrame.getRandCode()
image_code_rand = parser.get_img_code_rand()
# ===================DEBUG=================
# f = open('post_params.txt', 'w')
# f.write(str(self.orderParams))
# f.close()
# =========================================
# 读取用户配置信息，判断是否进行验证码的检查操作，默认设定为Y表示检查
if self.check_rand_status == 'Y':
# 检查用户输入验证码的合法性
checkResult = order.checkOrderImgCode(httpAccessObj, rand=image_code_rand, img_code=image_code,
token=parser.globalRepeatSubmitToken)
else:
checkResult = True
if checkResult:
# 检证用户提交的乘客信息的合法性
checkResult = order.checkOrderInfo(httpAccessObj, randCode=image_code,
passengerTicketStr=passengerTicketStr,
oldPassengersStr=oldPassengersStr, tour_flag='dc',
token=parser.globalRepeatSubmitToken)
if checkResult:
# 参数合法，则显示车票预定对话框
# 排队提示对话框
seat_type = passenger_info['passenger_1_seat']
queueCounParams = [
('train_date', str(time.ctime(order_request_params['train_date']['time'] / 1000))),
('train_no', order_request_params['train_no']),
('stationTrainCode', order_request_params['station_train_code']),
('seatType', seat_type),
('fromStationTelecode', order_request_params['from_station_telecode']),
('toStationTelecode', order_request_params['to_station_telecode']),
('leftTicket', ticketInfoForPassengerForm['queryLeftTicketRequestDTO']['ypInfoDetail']),
('purpose_codes', 'ADULT'),
('isCheckOrderInfo', checkResult['isCheckOrderInfo']),
('REPEAT_SUBMIT_TOKEN', parser.globalRepeatSubmitToken)
]
# 订单提交自动出票参数
orderParams = [
('passengerTicketStr', passengerTicketStr),
('oldPassengerStr', oldPassengersStr),
('randCode', image_code),
('purpose_codes', ticketInfoForPassengerForm['purpose_codes']),
('key_check_isChange', ticketInfoForPassengerForm['key_check_isChange']),
('leftTicketStr', ticketInfoForPassengerForm['leftTicketStr']),
('train_location', ticketInfoForPassengerForm['train_location']),
('REPEAT_SUBMIT_TOKEN', parser.globalRepeatSubmitToken)
]
queue_note = order.getQueueCount(httpAccessObj, queueCounParams, seat_type)
ui.OrderConfirmUI.ConfirmOrderDialog(comfirmFrame.root, queue_note, parser.get_train_info(),
comfirmFrame.getPassengerInfo(), self.comfirmOrderSubmitCallBack,
orderParams, httpAccessObj)
# 用户点击提交订单确认对话框的确认按钮时回调
def comfirmOrderSubmitCallBack(self, orderParams=None, httpAccessObj=None):
if not orderParams or not httpAccessObj:
print('预定参数不能为空')
return
# 检查单程预定允许排队
checkResult = order.checkQueueOrder(httpAccessObj, 'dc', orderParams)
if checkResult:
# 开始排队
timer = order.OrderQueueWaitTime(httpAccessObj, 'dc', order.waitFunc, order.finishMethod)
timer.start()
def main():
temp = AccessTrainOrderNetWork()
temp.access()
if __name__ == "__main__":
main()