79 lines
2.8 KiB
Python
79 lines
2.8 KiB
Python
# coding=utf-8
|
||
import base64
|
||
import copy
|
||
import random
|
||
|
||
|
||
def getPassCodeNewOrderAndLogin(session, imgType):
|
||
"""
|
||
下载验证码
|
||
:param session:
|
||
:param imgType: 下载验证码类型，login=登录验证码，其余为订单验证码
|
||
:return:
|
||
"""
|
||
if imgType == "login":
|
||
codeImgUrl = copy.deepcopy(session.urls["getCodeImg"])
|
||
codeImgUrl["req_url"] = codeImgUrl["req_url"].format(random.random())
|
||
else:
|
||
codeImgUrl = copy.deepcopy(session.urls["codeImgByOrder"])
|
||
codeImgUrl["req_url"] = codeImgUrl["req_url"].format(random.random())
|
||
print(u"下载验证码...")
|
||
img_path = './tkcode.png'
|
||
result = session.httpClint.send(codeImgUrl)
|
||
try:
|
||
if isinstance(result, dict):
|
||
print(u"下载验证码失败, 请手动检查是否ip被封，或者重试，请求地址：https://kyfw.12306.cn{}".format(codeImgUrl.get("req_url")))
|
||
return False
|
||
else:
|
||
print(u"下载验证码成功")
|
||
try:
|
||
with open(img_path, 'wb', encoding="utf-8") as img:
|
||
img.write(result)
|
||
except Exception:
|
||
with open(img_path, 'wb') as img:
|
||
img.write(result)
|
||
return result
|
||
except OSError:
|
||
print(u"验证码下载失败，可能ip被封，确认请手动请求: {0}".format(codeImgUrl))
|
||
|
||
|
||
def getPassCodeNewOrderAndLogin1(session, imgType):
|
||
"""
|
||
获取验证码2
|
||
:param session:
|
||
:param imgType:
|
||
:return:
|
||
"""
|
||
if imgType == "login":
|
||
codeImgUrl = copy.deepcopy(session.urls["getCodeImg1"])
|
||
codeImgUrl["req_url"] = codeImgUrl["req_url"].format(random.random())
|
||
else:
|
||
codeImgUrl = copy.deepcopy(session.urls["codeImgByOrder"])
|
||
codeImgUrl["req_url"] = codeImgUrl["req_url"].format(random.random())
|
||
print(u"下载验证码...")
|
||
img_path = './tkcode.png'
|
||
codeImgUrlRsp = session.httpClint.send(codeImgUrl)
|
||
if not isinstance(codeImgUrlRsp, str):
|
||
print("验证码获取失败")
|
||
return
|
||
result = eval(codeImgUrlRsp.split("(")[1].split(")")[0]).get("image")
|
||
try:
|
||
if isinstance(result, dict):
|
||
print(u"下载验证码失败, 请手动检查是否ip被封，或者重试，请求地址：https://kyfw.12306.cn{}".format(codeImgUrl.get("req_url")))
|
||
return False
|
||
else:
|
||
print(u"下载验证码成功")
|
||
try:
|
||
with open(img_path, 'wb', encoding="utf-8") as img:
|
||
img.write(result)
|
||
except Exception:
|
||
with open(img_path, 'wb') as img:
|
||
img.write(base64.b64decode(result))
|
||
return result
|
||
except OSError:
|
||
print(u"验证码下载失败，可能ip被封或者文件写入没权限")
|
||
|
||
|
||
if __name__ == '__main__':
|
||
pass
|