import requests
from lxml import html
import pyperclip


headers = {
      'User-Agent': "Mozilla/5.0 (Linux; U; Android 2.3.6; zh-cn; GT-S5660 Build/GINGERBREAD) AppleWebKit/533.1 "
                    "(KHTML, like Gecko) Version/4.0 Mobile Safari/533.1 MicroMessenger/4.5.255"
}


# 获取验证码
def get_valid_code():

    url = "http://idea.javatiku.cn/gencode.action"
    response = requests.get(url=url, headers=headers)
    selector = html.fromstring(response.text)
    number = selector.xpath('//div/div/h1/text()')
    return number[0]


# 获取激活码
def get_key(nums):
    data = {
        'validCode': nums,
    }
    url = "http://idea.javatiku.cn/rpc/getkey.action"
    cookies = requests.get(url="http://idea.javatiku.cn/", headers=headers, verify=False).cookies.get_dict()
    response = requests.post(url=url, headers=headers, data=data, verify=False, cookies=cookies).json()
    return response["data"]


if __name__ == '__main__':
    pyperclip.copy(get_key(get_valid_code()))
    spam = pyperclip.paste()
    alert = input("激活码已经写入剪切板啦~")