# 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)