mirror of https://github.com/testerSunshine/12306
23 lines
610 B
Python
23 lines
610 B
Python
|
# coding=utf-8
|
|||
|
|
|
|||
|
|
|
|||
|
|
def getPassCodeNewOrderAndLogin(session, imgType):
|
|||
|
|
"""
|
|||
|
|
下载验证码
|
|||
|
|
:param session:
|
|||
|
|
:param imgType: 下载验证码类型，login=登录验证码，其余为订单验证码
|
|||
|
|
:return:
|
|||
|
|
"""
|
|||
|
|
if imgType == "login":
|
|||
|
|
codeImgUrl = session.urls["getCodeImg"]
|
|||
|
|
else:
|
|||
|
|
codeImgUrl = session.urls["codeImgByOrder"]
|
|||
|
|
print (u"下载验证码...")
|
|||
|
|
img_path = './tkcode'
|
|||
|
|
result = session.httpClint.send(codeImgUrl)
|
|||
|
|
try:
|
|||
|
|
print(u"下载验证码成功")
|
|||
|
|
open(img_path, 'wb').write(result)
|
|||
|
|
except OSError as e:
|
|||
|
|
print (e)
|