修改提交需要验证码逻辑

init/select_ticket_info.py

@@ -368,54 +368,25 @@ class select:
         data['REPEAT_SUBMIT_TOKEN'] = self.token
         checkOrderInfo = json.loads(myurllib2.Post(checkOrderInfoUrl, data, ))
         if 'data' in checkOrderInfo:
-            if checkOrderInfo["data"]["ifShowPassCode"] == "y":
+            if checkOrderInfo["data"]["ifShowPassCode"] == "Y":
-                print("需要验证码，正在使用自动识别验证码功能")
+                is_need_code = True
-                for i in range(3):
+                if self.getQueueCount(train_no, set_type, is_need_code):
-                    codeimg = 'https://kyfw.12306.cn/otn/passcodeNew/getPassCodeNew?module=login&rand=sjrand&%s' % random.random()
+                    return True
-                    result = myurllib2.get(codeimg)
-                    img_path = './tkcode'
-                    open(img_path, 'wb').write(result)
-                    data['pass_code'] = DamatuApi(_get_yaml()["damatu"]["uesr"], _get_yaml()["damatu"]["pwd"], img_path).main()
-                    checkOrderInfo = json.loads(myurllib2.Post(checkOrderInfoUrl, data, ))
-                    if checkOrderInfo['data']['submitStatus'] is True:
-                        print ('车票提交通过，正在尝试排队')
-                        if self.getQueueCount(train_no, set_type):
-                            return True
-                        else:
-                            raise ticketNumOutException("提交订单失败")
-                    else:
-                        print("验证码识别错误，第{0}次重试".format(i))
             if checkOrderInfo['data']['submitStatus'] is True:
                     print ('车票提交通过，正在尝试排队')
-                    if self.getQueueCount(train_no, set_type):
+                    is_need_code = False
+                    if self.getQueueCount(train_no, set_type, is_need_code):
                         return True
             else:
                 if "errMsg" in checkOrderInfo['data'] and checkOrderInfo['data']["errMsg"]:
                     print checkOrderInfo['data']["errMsg"]
-                    if checkOrderInfo['data']["errMsg"].find("验证码") != -1:
+
-                        print("需要验证码，正在使用自动识别验证码功能")
-                        for i in range(3):
-                            codeimg = 'https://kyfw.12306.cn/otn/passcodeNew/getPassCodeNew?module=login&rand=sjrand&%s' % random.random()
-                            result = myurllib2.get(codeimg)
-                            img_path = './tkcode'
-                            open(img_path, 'wb').write(result)
-                            data['pass_code'] = DamatuApi(_get_yaml()["damatu"]["uesr"], _get_yaml()["damatu"]["pwd"],
-                                                          img_path).main()
-                            checkOrderInfo = json.loads(myurllib2.Post(checkOrderInfoUrl, data, ))
-                            if checkOrderInfo['data']['submitStatus'] is True:
-                                print ('车票提交通过，正在尝试排队')
-                                if self.getQueueCount(train_no, set_type):
-                                    return True
-                                else:
-                                    raise ticketNumOutException("提交订单失败")
-                            else:
-                                print("验证码识别错误，第{0}次重试".format(i))
                 else:
                     print checkOrderInfo
         elif 'messages' in checkOrderInfo and checkOrderInfo['messages']:
             print (checkOrderInfo['messages'][0])
 
-    def getQueueCount(self, train_no, set_type):
+    def getQueueCount(self, train_no, set_type, is_need_code):
         """
         # 模拟查询当前的列车排队人数的方法
         # 返回信息组成的提示字符串
@@ -455,7 +426,7 @@ class select:
                         print("当前余票数小于乘车人数，放弃订票")
                     else:
                         print("排队成功, 当前余票还剩余: {0} 张".format(ticket_split))
-                        if self.checkQueueOrder():
+                        if self.checkQueueOrder(is_need_code):
                             return True
                 else:
                     print("当前排队人数:" + str(countT) + "当前余票还剩余:{0} 张，继续排队中".format(ticket_split))
@@ -472,11 +443,12 @@ class select:
             else:
                 print("未知错误 {0}".format("".join(getQueueCountResult)))
 
-    def checkQueueOrder(self):
+    def checkQueueOrder(self, is_node_code=False):
         """
         模拟提交订单是确认按钮，参数获取方法还是get_ticketInfoForPassengerForm 中获取
         :return: 
         """
+
         passengerTicketStrList, oldPassengerStr = self.getPassengerTicketStrListAndOldPassengerStr()
         checkQueueOrderUrl = "https://kyfw.12306.cn/otn/confirmPassenger/confirmSingleForQueue"
         data = {
@@ -493,28 +465,34 @@ class select:
         }
         try:
             for i in range(3):
-                checkQueueOrderResult = json.loads(myurllib2.Post(checkQueueOrderUrl, data))
+                if is_node_code:
-                if checkQueueOrderResult:
+                    print("需要验证码，正在使用自动识别验证码功能")
-                    break
+                    codeimg = 'https://kyfw.12306.cn/otn/passcodeNew/getPassCodeNew?module=login&rand=sjrand&%s' % random.random()
-        except ValueError:
+                    result = myurllib2.get(codeimg)
-            checkQueueOrderResult = {}
+                    img_path = './tkcode'
-        if checkQueueOrderResult:
+                    open(img_path, 'wb').write(result)
-            if "status" in checkQueueOrderResult and checkQueueOrderResult["status"]:
+                    data['randCode'] = DamatuApi(_get_yaml()["damatu"]["uesr"], _get_yaml()["damatu"]["pwd"],
-                c_data = checkQueueOrderResult["data"] if "data" in checkQueueOrderResult else {}
+                                                  img_path).main()
-                if 'submitStatus' in c_data and c_data['submitStatus']:
-                    print("提交订单成功！")
-                    self.queryOrderWaitTime()
                 else:
-                    if 'errMsg' in c_data and c_data['errMsg']:
+                    print("不需要验证码")
-                        print("提交订单失败，" + c_data['errMsg'])
+                checkQueueOrderResult = json.loads(myurllib2.Post(checkQueueOrderUrl, data))
+                if "status" in checkQueueOrderResult and checkQueueOrderResult["status"]:
+                    c_data = checkQueueOrderResult["data"] if "data" in checkQueueOrderResult else {}
+                    if 'submitStatus' in c_data and c_data['submitStatus'] is True:
+                        print("提交订单成功！")
+                        break
                     else:
-                        print(c_data)
+                        if 'errMsg' in c_data and c_data['errMsg']:
-                        print('订票失败!很抱歉,请重试提交预订功能!')
+                            print("提交订单失败，{0}，重新识别中".format(c_data['errMsg']))
-            elif "messages" in checkQueueOrderResult and checkQueueOrderResult["messages"]:
+                        else:
-                print("提交订单失败,错误信息: " + checkQueueOrderResult["messages"])
+                            print(c_data)
-            else:
+                            print('订票失败!很抱歉,请重试提交预订功能!')
-                print("提交订单中，请耐心等待：" + str(checkQueueOrderResult["validateMessages"]))
+                elif "messages" in checkQueueOrderResult and checkQueueOrderResult["messages"]:
-        else:
+                    print("提交订单失败,错误信息: " + checkQueueOrderResult["messages"])
+                else:
+                    print("提交订单中，请耐心等待：" + str(checkQueueOrderResult["validateMessages"]))
+            self.queryOrderWaitTime()
+        except ValueError:
             print("接口 {} 无响应".format(checkQueueOrderUrl))
 
     def queryOrderWaitTime(self):