#!/bin/env python
# -*- coding=utf-8 -*-
import random
import json
import re
from time import sleep

from config.ticketConf import _get_yaml
from PIL import Image
from damatuCode.damatuWeb import DamatuApi
from myUrllib import myurllib2

codeimg = 'https://kyfw.12306.cn/otn/passcodeNew/getPassCodeNew?module=login&rand=sjrand&%s' % random.random()


def cookietp():
    stoidinput("获取Cookie")
    Url = "https://kyfw.12306.cn/otn/login/init"
    myurllib2.get(Url)
    # for index, c in enumerate(myurllib2.cookiejar):
    #     stoidinput(c)


def readImg():
    """
    增加手动打码，只是登录接口，完全不用担心提交订单效率
    思路
    1.调用PIL显示图片
    2.图片位置说明，验证码图片中每个图片代表一个下标，依次类推，1，2，3，4，5，6，7，8
    3.控制台输入对应下标，按照英文逗号分开，即可手动完成打码，
    :return:
    """

    global randCode
    stoidinput("下载验证码...")
    img_path = './tkcode'
    result = myurllib2.get(codeimg)
    try:
        open(img_path, 'wb').write(result)
        if _get_yaml()["is_aotu_code"]:
            randCode = DamatuApi(_get_yaml()["damatu"]["uesr"], _get_yaml()["damatu"]["pwd"], img_path).main()
        else:
            img = Image.open('./tkcode')
            img.show()
            codexy()
    except OSError as e:
        print (e)
        pass


def stoidinput(text):
    """
    正常信息输出
    :param text:
    :return:
    """
    print "\033[34m[*]\033[0m %s " % text


def errorinput(text):
    """
    错误信息输出
    :param text:
    :return:
    """
    print "\033[32m[!]\033[0m %s " % text
    return False


def codexy():
    """
    获取验证码
    :return: str
    """

    Ofset = raw_input("[*] 请输入验证码: ")
    select = Ofset.split(',')
    global randCode
    post = []
    offsetsX = 0  # 选择的答案的left值,通过浏览器点击8个小图的中点得到的,这样基本没问题
    offsetsY = 0  # 选择的答案的top值
    for ofset in select:
        if ofset == '1':
            offsetsY = 46
            offsetsX = 42
        elif ofset == '2':
            offsetsY = 46
            offsetsX = 105
        elif ofset == '3':
            offsetsY = 45
            offsetsX = 184
        elif ofset == '4':
            offsetsY = 48
            offsetsX = 256
        elif ofset == '5':
            offsetsY = 36
            offsetsX = 117
        elif ofset == '6':
            offsetsY = 112
            offsetsX = 115
        elif ofset == '7':
            offsetsY = 114
            offsetsX = 181
        elif ofset == '8':
            offsetsY = 111
            offsetsX = 252
        else:
            pass
        post.append(offsetsX)
        post.append(offsetsY)
    randCode = str(post).replace(']', '').replace('[', '').replace("'", '').replace(' ', '')


def login(user, passwd):
    """
    登陆
    :param user: 账户名
    :param passwd: 密码
    :return: 
    """
    login_num = 0
    while True:
        cookietp()
        readImg()
        login_num += 1
        randurl = 'https://kyfw.12306.cn/otn/passcodeNew/checkRandCodeAnsyn'
        logurl = 'https://kyfw.12306.cn/otn/login/loginAysnSuggest'
        surl = 'https://kyfw.12306.cn/otn/login/userLogin'
        randdata = {
            "randCode": randCode,
            "rand": "sjrand"
        }
        logdata = {
            "loginUserDTO.user_name": user,
            "userDTO.password": passwd,
            "randCode": randCode
        }
        ldata = {
            "_json_att": None
        }
        fresult = json.loads(myurllib2.Post(randurl, randdata), encoding='utf8')
        checkcode = fresult['data']['msg']
        if checkcode == 'FALSE':
            errorinput("验证码有误,第{}次尝试重试".format(login_num))
        else:
            stoidinput("验证码通过,开始登录..")
            sleep(1)
            try:
                tresult = json.loads(myurllib2.Post(logurl, logdata), encoding='utf8')
                if 'data' not in tresult:
                    errorinput("登录失败: %s" % tresult['messages'][0])
                # elif "messages" in tresult and tresult["messages"][0].find("密码输入错误") is not -1:
                #     errorinput("登陆失败：{}".format(tresult["messages"][0]))
                #     break
                elif 'messages' in tresult and tresult['messages']:
                    messages = tresult['messages'][0]
                    if messages.find("密码输入错误") is not -1:
                        errorinput("登陆失败：{}".format(tresult["messages"][0]))
                        break
                    else:
                        errorinput("登录失败: %s" % tresult['messages'][0])
                        stoidinput("尝试重新登陆")
                else:
                    stoidinput("登录成功")
                    myurllib2.Post(surl, ldata)
                    getUserinfo()
                    break
            except ValueError as e:
                errorinput(e)
        sleep(1)


def getUserinfo():
    """
    登录成功后,显示用户名
    :return:
    """
    url = 'https://kyfw.12306.cn/otn/modifyUser/initQueryUserInfo'
    data = dict(_json_att=None)
    result = myurllib2.Post(url, data)
    userinfo = result
    name = r'<input name="userDTO.loginUserDTO.user_name" style="display:none;" type="text" value="(\S+)" />'
    try:
        stoidinput("欢迎 %s 登录" % re.search(name, result).group(1))
    except AttributeError:
        pass


def main():
    login(_get_yaml()["set"]["12306count"][0]["uesr"], _get_yaml()["set"]["12306count"][1]["pwd"])


def logout():
    url = 'https://kyfw.12306.cn/otn/login/loginOut'
    result = myurllib2.get(url)
    if result:
        stoidinput("已退出")
    else:
        errorinput("退出失败")


if __name__ == "__main__":
    main()
    # logout()