add_cookie.py

import json

from utils.config import get_config, get_file
from utils.selenium_browser import get_browser
from selenium.webdriver.support.wait import WebDriverWait
from selenium.webdriver.common.by import By
from selenium.webdriver.support import expected_conditions as EC
from selenium.common.exceptions import WebDriverException

if __name__ == '__main__':
    """
    用于获取手机端cookie
    """
    config = get_config()
    config['headless'] = False
    browser = get_browser(config)
    browser.get("https://plogin.m.jd.com/login/login")
    try:
        wait = WebDriverWait(browser, 135)
        print("请在网页端通过手机号码登录")
        wait.until(EC.presence_of_element_located((By.ID, 'msShortcutMenu')))
        browser.get("https://home.m.jd.com/myJd/newhome.action")
        username = wait.until(EC.presence_of_element_located((By.CLASS_NAME, 'my_header_name'))).text
        cookie = ""
        for _ in browser.get_cookies():
            if _["name"] == "pt_key" or _["name"] == "pt_pin":
                cookie += _["name"] + "=" + _["value"] + ";"
        config['mobile_cookie'] = cookie[0:-1]
        print("获取的cookie是：" + cookie)
        with open(get_file("./config.json"), mode='w', encoding="utf-8") as f:
            json.dump(config, f, indent=4, ensure_ascii=False)
        print("成功添加", username)
    except WebDriverException:
        print("添加失败")
    finally:
        browser.close()