# https://m.jd.com/
# JD_COOKIE

import time
from playwright.sync_api import sync_playwright


with sync_playwright() as playwright:
    browser = playwright.chromium.launch(headless=False)
    context = browser.new_context()

    page = context.new_page()

    page.goto("https://m.jd.com")

    time.sleep(1.2)

    # 等待登录按钮出现
    page.wait_for_selector("#msShortcutLogin > span")

    # 点击登录按钮
    page.click("#msShortcutLogin > span")

    time.sleep(1.2)

    # 等待同意协议勾选框出现
    page.wait_for_selector("#app > div > p.policy_tip > input")

    # 点击同意协议勾选框
    page.click("#app > div > p.policy_tip > input")

    time.sleep(0.2)

    # 输入手机号, 因为经过上面, 已确定页面已成功加载, 所以不等待
    page.fill(
        "#app > div > div:nth-child(3) > p:nth-child(1) > input", "17620730210")

    # 点击获取验证码按钮
    page.click("#app > div > div:nth-child(3) > p:nth-child(2) > button")

    time.sleep(30)

    page.reload()

    cookies = page.context.cookies()

    pt_key = 'pt_key='
    pt_pin = 'pt_pin='

    for cookie in cookies:
        if cookie['name'] == 'pt_key':
            pt_key += cookie['value']
        if cookie['name'] == 'pt_pin':
            pt_pin += cookie['value']

    print(pt_key+';'+pt_pin+';')

    browser.close()