mirror of https://github.com/testerSunshine/12306
271 lines
12 KiB
Python
271 lines
12 KiB
Python
# -*- coding=utf-8 -*-
|
||
import json
|
||
import time
|
||
|
||
import datetime
|
||
|
||
import re
|
||
import urllib
|
||
|
||
from myUrllib import myurllib2
|
||
|
||
|
||
class select:
|
||
def __init__(self, from_station, to_station):
|
||
self.from_station, self.to_station = self.station_table(from_station, to_station)
|
||
self.order_request_params = {} # 订单提交时的参数
|
||
self.ticketInfoForPassengerForm = {} # 初始化当前页面参数
|
||
self.current_seats = {} # 席别信息
|
||
self.token = ''
|
||
self.set_type = ''
|
||
|
||
def get_order_request_params(self):
|
||
return self.order_request_params
|
||
|
||
def get_ticketInfoForPassengerForm(self):
|
||
return self.ticketInfoForPassengerForm
|
||
|
||
def get_current_seats(self):
|
||
return self.current_seats
|
||
|
||
def get_token(self):
|
||
return self.token
|
||
|
||
def get_set_type(self):
|
||
return self.set_type
|
||
|
||
def station_seat(self, index):
|
||
"""
|
||
获取车票对应坐席
|
||
:param seat_type:
|
||
:return:
|
||
"""
|
||
seat = {32: '商务座 ',
|
||
31: '一等座 ',
|
||
30: '二等座 ',
|
||
25: '特等座 ',
|
||
23: '软卧 ',
|
||
28: '硬卧 ',
|
||
29: '硬座 ',
|
||
26: '无座 '
|
||
}
|
||
return seat[index]
|
||
|
||
def station_table(self, from_station, to_station):
|
||
"""
|
||
读取车站信息
|
||
:param station:
|
||
:return:
|
||
"""
|
||
result = open('station_name.txt')
|
||
info = result.read().split('=')[1].strip("'").split('@')
|
||
del info[0]
|
||
station_name = {}
|
||
for i in range(0, len(info)):
|
||
# print info[i]
|
||
n_info = info[i].split('|')
|
||
station_name[n_info[1]] = n_info[2]
|
||
from_station = station_name[from_station]
|
||
to_station = station_name[to_station]
|
||
return from_station, to_station
|
||
|
||
def time(self):
|
||
today = datetime.date.today()
|
||
tomorrow = today+datetime.timedelta(1)
|
||
return tomorrow.strftime('%Y-%m-%d')
|
||
|
||
def getRepeatSubmitToken(self):
|
||
"""
|
||
获取提交车票请求token
|
||
:return: token
|
||
"""
|
||
initdc_url = 'https://kyfw.12306.cn/otn/confirmPassenger/initDc'
|
||
initdc_result = myurllib2.get(initdc_url)
|
||
token_name = re.compile(r"var globalRepeatSubmitToken = '(\S+)'")
|
||
ticketInfoForPassengerForm_name = re.compile(r'var ticketInfoForPassengerForm=(\{.+\})?')
|
||
order_request_params_name = re.compile(r'var orderRequestDTO=(\{.+\})?')
|
||
# if token_name and ticketInfoForPassengerForm_name and order_request_params_name:
|
||
self.token = re.search(token_name, initdc_result).group(1)
|
||
re_tfpf = re.findall(ticketInfoForPassengerForm_name, initdc_result)
|
||
re_orp = re.findall(order_request_params_name, initdc_result)
|
||
if re_tfpf:
|
||
self.ticketInfoForPassengerForm = json.loads(re_tfpf[0].replace("'", '"'))
|
||
else:
|
||
print('车次获取信息为空')
|
||
if re_orp:
|
||
self.order_request_params = json.loads(re_orp[0].replace("'", '"'))
|
||
else:
|
||
print('订单获取信息为空')
|
||
|
||
def getPassengerDTOs(self):
|
||
"""
|
||
获取乘客信息
|
||
:return:
|
||
"""
|
||
get_passengerDTOs = 'https://kyfw.12306.cn/otn/confirmPassenger/getPassengerDTOs'
|
||
get_data = {
|
||
'_json_att': None,
|
||
'REPEAT_SUBMIT_TOKEN': self.token
|
||
}
|
||
jsonData = json.loads(myurllib2.Post(get_passengerDTOs, get_data))
|
||
if 'data' in jsonData and jsonData['data'] and 'normal_passengers' in jsonData['data'] and jsonData['data'][
|
||
'normal_passengers']:
|
||
return jsonData['data']['normal_passengers']
|
||
else:
|
||
if 'data' in jsonData and 'exMsg' in jsonData['data'] and jsonData['data']['exMsg']:
|
||
print(jsonData['data']['exMsg'])
|
||
elif 'messages' in jsonData and jsonData['messages']:
|
||
print(jsonData['messages'])
|
||
else:
|
||
print("未查找到常用联系人")
|
||
|
||
def submitOrderRequest(self):
|
||
"""
|
||
提交车次信息
|
||
:return: 车次信息，座位号
|
||
"""
|
||
|
||
select_url = 'https://kyfw.12306.cn/otn/leftTicket/query?' \
|
||
'leftTicketDTO.train_date={}&leftTicketDTO.from_station={}' \
|
||
'&leftTicketDTO.to_station={}&purpose_codes=ADULT'.format(self.time(), self.from_station, self.to_station)
|
||
check_user_url = 'https://kyfw.12306.cn/otn/login/checkUser'
|
||
submit_station_url = 'https://kyfw.12306.cn/otn/leftTicket/submitOrderRequest'
|
||
station_ticket = json.loads(myurllib2.get(select_url), encoding='utf-8')
|
||
value = station_ticket['data']
|
||
if value.__len__() is 0:
|
||
print ('当前车次查询为空...')
|
||
else:
|
||
for i in value['result']:
|
||
ticket_info = i.split('|')
|
||
for j in range(20, 33):
|
||
if ticket_info[j] != '' and ticket_info[j] != '无': # 过滤有效目标车次
|
||
print ('车次: ' + ticket_info[3] + ' 始发车站: ' + value['map']['AOH'] + ' 终点站: ' + \
|
||
value['map']['CSQ'] + ' ' + self.station_seat(j) + ': ' + ticket_info[j])
|
||
print ('正在尝试提交订票...')
|
||
data = dict(_json_att=None)
|
||
check_user = json.loads(myurllib2.Post(check_user_url, data), encoding='utf-8')
|
||
check_user_flag = check_user['data']['flag']
|
||
if check_user_flag is True:
|
||
print ('尝试提交订单...')
|
||
# 预定的请求参数，注意参数顺序
|
||
# 注意这里为了防止secretStr被urllib.parse过度编码，在这里进行一次解码
|
||
# 否则调用HttpTester类的post方法将会将secretStr编码成为无效码,造成提交预定请求失败
|
||
data = [('secretStr', urllib.unquote(ticket_info[0])), # 字符串加密
|
||
('train_date', self.time()), # 出发时间
|
||
('back_train_date', self.time()), # 返程时间
|
||
('tour_flag', 'dc'), # 旅途类型
|
||
('purpose_codes', 'ADULT'), # 成人票还是学生票
|
||
('query_from_station_name', self.from_station), # 起始车站
|
||
('query_to_station_name', self.to_station), # 终点车站
|
||
]
|
||
submitResult = json.loads(myurllib2.Post(submit_station_url, data), encoding='utf-8')
|
||
if 'data' in submitResult and submitResult['data']:
|
||
if submitResult['data'] == 'N':
|
||
print ('出票成功')
|
||
return self.station_seat(j) # 车次信息，座位号
|
||
else:
|
||
print ('出票失败')
|
||
elif 'messages' in submitResult and submitResult['messages']:
|
||
print(submitResult['messages'])
|
||
else:
|
||
if check_user['messages']:
|
||
print ('用户检查失败：%s' % check_user['messages'])
|
||
else:
|
||
print ('用户检查失败： %s' % check_user)
|
||
else:
|
||
print ('未查询到有效车次，重新查询中')
|
||
|
||
def getPassengerTicketStr(self, set_type):
|
||
"""
|
||
获取getPassengerTicketStr 提交对应的代号码
|
||
:param str: 坐席
|
||
:return:
|
||
"""
|
||
passengerTicketStr = {
|
||
'一等座': 'M',
|
||
'特等座': 'P',
|
||
'二等座': 'O',
|
||
'商务座': 9,
|
||
'硬座': 1,
|
||
'无座': 1,
|
||
'软卧': 4,
|
||
'硬卧': 3,
|
||
}
|
||
self.set_type = str(passengerTicketStr[set_type.replace(' ', '')])
|
||
|
||
def ticket_type(self):
|
||
"""订单票的类型，目前只考虑成人票，此方法暂时搁置，做备案"""
|
||
ticket_type = {'adult': "1", 'child': "2", 'student': "3", 'disability': "4"}
|
||
return ticket_type
|
||
|
||
def checkOrderInfo(self, user_info):
|
||
"""
|
||
检查支付订单，需要提交REPEAT_SUBMIT_TOKEN
|
||
passengerTicketStr : 座位编号,0,票类型,乘客名,证件类型,证件号,手机号码,保存常用联系人(Y或N)
|
||
oldPassengersStr: 乘客名,证件类型,证件号,乘客类型
|
||
:return:
|
||
"""
|
||
passengerTicketStr = []
|
||
oldPassengerStr = []
|
||
passengerTicketStr.append(user_info[0]['passenger_id_type_code'])
|
||
passengerTicketStr.append(user_info[0]['passenger_name'])
|
||
passengerTicketStr.append(user_info[0]['passenger_type'])
|
||
passengerTicketStr.append(user_info[0]['passenger_id_no'])
|
||
passengerTicketStr.append(user_info[0]['mobile_no'])
|
||
oldPassengerStr.append(user_info[0]['passenger_name'])
|
||
oldPassengerStr.append(user_info[0]['passenger_type'])
|
||
oldPassengerStr.append(user_info[0]['passenger_id_no'])
|
||
oldPassengerStr.append(user_info[0]['passenger_type'])
|
||
|
||
checkOrderInfoUrl = 'https://kyfw.12306.cn/otn/confirmPassenger/checkOrderInfo'
|
||
data = {
|
||
'cancel_flag': 2,
|
||
'bed_level_order_num': 000000000000000000000000000000,
|
||
'passengerTicketStr': self.set_type+',0,'+str(passengerTicketStr)+',N',
|
||
'oldPassengerStr': str(oldPassengerStr)+'_',
|
||
'tour_flag': 'dc',
|
||
'REPEAT_SUBMIT_TOKEN': self.token,
|
||
}
|
||
checkOrderInfo = json.loads(myurllib2.Post(checkOrderInfoUrl, data))
|
||
print (checkOrderInfo)
|
||
if 'data' in checkOrderInfo and checkOrderInfo['data']['submitStatus'] is True:
|
||
print ('车票提交通过，正在尝试排队')
|
||
elif 'messages' in checkOrderInfo and checkOrderInfo['messages']:
|
||
print (checkOrderInfo['messages'])
|
||
|
||
def getQueueCount(self):
|
||
"""
|
||
# 模拟查询当前的列车排队人数的方法
|
||
# 返回信息组成的提示字符串
|
||
:param token:
|
||
:return:
|
||
"""
|
||
getQueueCountUrl = 'https://kyfw.12306.cn/otn/confirmPassenger/getQueueCount'
|
||
data = {
|
||
# 'train_date': 'Sun May 07 2017 00:00:00 GMT+0800',
|
||
'train_date': self.ticketInfoForPassengerForm['queryLeftTicketRequestDTO']['train_date'],
|
||
'train_no': self.ticketInfoForPassengerForm['queryLeftTicketRequestDTO']['train_no'],
|
||
'stationTrainCode': self.ticketInfoForPassengerForm['queryLeftTicketRequestDTO']['station_train_code'],
|
||
'seatType': self.set_type,
|
||
'fromStationTelecode': self.ticketInfoForPassengerForm['queryLeftTicketRequestDTO']['from_station'],
|
||
'toStationTelecode': self.ticketInfoForPassengerForm['queryLeftTicketRequestDTO']['to_station'],
|
||
'leftTicket': self.ticketInfoForPassengerForm['leftTicketStr'],
|
||
'purpose_codes': self.ticketInfoForPassengerForm['purpose_codes'],
|
||
'train_location': self.ticketInfoForPassengerForm['train_location'],
|
||
'REPEAT_SUBMIT_TOKEN': self.token,
|
||
}
|
||
print(data)
|
||
getQueueCountResult = json.loads(myurllib2.Post(getQueueCountUrl, data))
|
||
print(getQueueCountResult)
|
||
|
||
def main(self):
|
||
set_type = self.submitOrderRequest()
|
||
self.getPassengerTicketStr(set_type)
|
||
self.getRepeatSubmitToken()
|
||
user_info = self.getPassengerDTOs()
|
||
self.checkOrderInfo(user_info)
|
||
self.getQueueCount()
|
||
|
||
if __name__ == '__main__':
|
||
a = select('上海', '北京')
|
||
a.main() |