__author__ = 'Administrator'
import requests
import xlstr
import json
import urllib.parse


class C12306:
    username = ''
    password = ''
    domain = 'kyfw.12306.cn' #请求域名（真实连接地址）
    host='kyfw.12306.cn' #请求的域名（host）
    http = requests.session()
    stationCode = {}

    def __init__(self,domain=''):
        if domain!='':
            self.domain=domain

        headers={"host":self.host}
        self.http.get("https://" + self.domain + "/otn/", verify=False,headers=headers)
        res = self.http.get('https://'+self.domain+'/otn/login/init', verify=False,headers=headers)
        assert isinstance(res, requests.Response)

        if not '/otn/resources/merged/login_js.js' in res.text:
            raise C12306Error('初始化页面错误')

        dynamic_js_url = xlstr.substr(res.text, "/otn/resources/merged/login_js.js", "\"");
        dynamic_js = self.http.get("https://"+self.domain + dynamic_js_url, verify=False,headers=headers)
        self.load_station_code()

    def load_station_code(self):
        """Load station telcode from 12306.cn

        加载车站电报码，各个请求中会用到
        :raise C12306Error:
        """
        header={"host":self.host}
        res = requests.get('https://'+self.domain+'/otn/resources/js/framework/station_name.js', verify=False,headers=header)
        if res.status_code != 200:
            raise C12306Error('加载车站信息错误,请重开!')

        stationStrs = xlstr.substr(res.text, "'", "'")
        stationList = stationStrs.split('@')
        stationDict = {}

        for stationStr in stationList:
            station = stationStr.split("|")
            if len(station) > 3:
                stationDict[station[1]] = station[2]

        self.stationCode = stationDict


    def login(self, username, password, auth_code):
        self.username = username
        self.password = password

        data = {'loginUserDTO.user_name': self.username, 'userDTO.password': self.password, 'randCode': auth_code}
        headers = {'X-Requested-With': 'XMLHttpRequest','host':self.host}

        res = self.http.post("https://" + self.domain + "/otn/login/loginAysnSuggest", data, verify=False,
                             headers=headers)

        if not 'loginCheck":"Y"},"' in res.text:
            raise C12306Error('登录失败:' + ''.join(res.json()['messages']))

        return True


    def auth_code_img(self, module='passenger'):
        if module=='passenger':
            url="https://" + self.domain + "/otn/passcodeNew/getPassCodeNew.do?module=passenger&rand=randp"
        else:
            url = "https://" + self.domain + "/otn/passcodeNew/getPassCodeNew?module=" + module + "&rand=sjrand"
        res = self.http.get(url, verify=False,headers={"host":self.host})
        assert isinstance(res, requests.Response)
        return res.content


    def search_ticket(self, fromStation, toStation, date):

        headers={'Referer':'https://kyfw.12306.cn/otn/leftTicket/init',"host":self.host}

        url='https://' + self.domain + '/otn/leftTicket/queryT?leftTicketDTO.train_date='+date\
            +"&leftTicketDTO.from_station="+self.stationCode[fromStation]+"&leftTicketDTO.to_station="+\
            self.stationCode[toStation]+"&purpose_codes=ADULT"

        res = self.http.get(url,verify=False,headers=headers)
        ticketInfo=res.json()
        if ticketInfo['status']!=True or ticketInfo['messages']!=[] :
            raise C12306Error('查询错误:'+''.join(ticketInfo['messages']))

        if len(ticketInfo['data'])<=0:
            raise C12306Error('查询错误:'+'没有找到直达车次，请重新查询!')

        return ticketInfo['data']

    def submit_order(self,ticket):
        """

        :param ticket: Ticket
        :return: :raise C12306Error:
        """
        data='secretStr='+ticket.secret_str+'&train_date='+ticket.train_date+'&back_train_date='+ticket.train_date+\
              '&tour_flag=dc&purpose_codes=ADULT&query_from_station_name='+ticket.from_station_name+\
              '&query_to_station_name='+ticket.to_station_name+'&undefined'
        headers={'Referer':'https://kyfw.12306.cn/otn/leftTicket/init','X-Requested-With':'XMLHttpRequest'\
            ,'Content-Type':'application/x-www-form-urlencoded; charset=UTF-8',"host":self.host}

        data=data.encode()

        res=self.http.post('https://'+self.domain+'/otn/leftTicket/submitOrderRequest',data,verify=False,headers=headers)

        orderInfo=res.json()
        if orderInfo['status']!=True:
            raise C12306Error('提交订单错误:'+''.join(orderInfo['messages']))

        res=self.http.post('https://'+self.domain+'/otn/confirmPassenger/initDc',{'_json_att':''},verify=False,headers=headers)
        pageText=res.text

        self.Token=xlstr.substr(pageText,"globalRepeatSubmitToken = '","'")
        self.keyCheck=xlstr.substr(pageText,"'key_check_isChange':'","'")
        self.leftTicketStr=xlstr.substr(pageText,"leftTicketStr':'","'")
        self.trainLocation=xlstr.substr(pageText,"train_location':'","'")

        if  len(self.Token)!=32 :
            raise C12306Error('预定页面获取失败!')

        return True


    def check_order(self,Ticket,passengerList,randCode):
        if len(passengerList)<1 :
            raise C12306Error('没有勾选乘客，无法购票!')

        ticketInfo=[]
        oldPassengerInfo=[]
        for passenger in passengerList:
            ticketStr=Ticket.seat_type+",0,"+passenger['ticketType']+","+passenger['name']+',1,'+passenger['idcard']+','+\
                      passenger['telephone']+',N'
            ticketInfo.append(ticketStr)
            oldPStr=passenger['name']+',1,'+passenger['idcard']+','+passenger['ticketType']
            oldPassengerInfo.append(oldPStr)
        ticketStrs=urllib.parse.quote_plus('_'.join(ticketInfo))
        oldPassengerStrs=urllib.parse.quote_plus('_'.join(oldPassengerInfo))

        pstr="cancel_flag=2&bed_level_order_num=000000000000000000000000000000&passengerTicketStr="+ticketStrs\
             +"&oldPassengerStr="+oldPassengerStrs+"&tour_flag=dc&randCode="+randCode+\
             "&_json_att=&REPEAT_SUBMIT_TOKEN="+self.Token

        pstr=pstr.encode()

        res=self.http.post('https://'+self.domain+'/otn/confirmPassenger/checkOrderInfo',pstr,headers={"host":self.host})

        orderInfo=res.json()
        if orderInfo['status']!=True:
            raise C12306Error('提交订单错误:'+''.join(orderInfo['messages']))

        if orderInfo['data']['submitStatus'] ==False and  '验证码' in orderInfo['data']['errMsg']:
            return False

        if orderInfo['data']['submitStatus'] ==True:
            pstr="train_date="+urllib.parse.quote_plus(Ticket.train_date_utc)+"&train_no="+Ticket.train_no+"&stationTrainCode="+\
                 Ticket.station_train_code+"&seatType="+Ticket.seat_type+"&fromStationTelecode="+\
                 Ticket.from_station_telecode+"&toStationTelecode="+Ticket.to_station_telecode+"&leftTicket="+\
                 Ticket.yp_info+"&purpose_codes=00&_json_att=&REPEAT_SUBMIT_TOKEN="+self.Token

            pstr=pstr.encode()
            res=self.http.post('https://'+self.domain+'/otn/confirmPassenger/getQueueCount',pstr,headers={"host":self.host}).json()

            if res['status']!=True:
                raise C12306Error('查询排队队列错误:'+''.join(res['messages']))

            if res['data']['op_2']!='false':
                raise C12306Error('排队人数已满，取消操作!')
            else:
                self.passengerStr="passengerTicketStr="+ticketStrs+"&oldPassengerStr="+oldPassengerStrs
                return True
        else:
            raise C12306Error('订单错误:'+''.join(orderInfo['data']['errMsg']))

    def confirm_order(self,randCode):
        pstr=self.passengerStr+"&randCode="+randCode+"&purpose_codes=00"+\
             "&key_check_isChange="+self.keyCheck+"&leftTicketStr="+self.leftTicketStr+\
             "&train_location="+self.trainLocation+"&_json_att=&REPEAT_SUBMIT_TOKEN="+self.Token
        pstr=pstr.encode()

        res=self.http.post("https://"+self.domain+"/otn/confirmPassenger/confirmSingleForQueue",pstr,headers={"host":self.host}).json()

        if res['status']==True :
            return True
        elif '请重试' in res['messages']:
            return self.confirm_order(randCode)
        else:
            raise C12306Error('确认订单出现错误:'+res['messages'])

    def keep_online(self):
        print("keep online")
        self.http.get("https://"+self.domain+"/otn/queryOrder/initNoComplete",headers={"host":self.host})


class C12306Error(Exception):
    def __init__(self, val):
        self.value = val

    def __str__(self):
        str(self.value)