Get-JDCookie.py

#!/usr/bin/env python
# -*- coding:utf-8 -*-
import asyncio
from socket import timeout
import pyperclip
import os
from pyppeteer import launch


def find_cookie(cookies):
    """提取pt_key和pt_pin
    """
    for item in cookies.split('; '):
        if 'pt_pin' in item:
            pt_pin = item
        if 'pt_key' in item:
            pt_key = item
    jd_cookie = pt_pin+';'+pt_key+';'
    pyperclip.copy(jd_cookie)   # 拷贝JDcookie到剪切板
    print("Cookie:", jd_cookie)
    print("已拷贝Cookie到剪切板、直接黏贴即可。")
    # return jd_cookie
    os.system('pause')  # 按任意键继续



async def main():
    """使用pyppeteer库来登录京东、并获取cookie
    """
    print('请在弹出的网页中登录账号、推荐使用账户短信验证码的形式登录。')
    browser = await launch(headless=False, dumpio=True, autoClose=False,
                           args=['--no-sandbox', '--window-size=800,700', '--disable-infobars'])   # 进入有头模式
    context = await browser.createIncognitoBrowserContext() # 隐身模式
    page = await context.newPage()           # 打开新的标签页
    await page.setViewport({'width': 800, 'height': 700})      # 页面大小一致
    await page.goto('https://home.m.jd.com') # 访问主页

    await page.waitFor(1000)
    elm = await page.waitForXPath('//*[@id="myHeader"]',timeout=0)  # 通过判断用户头像是否存在来确定登录状态
    if elm:
        cookie = await page.cookies()
        # print(cookie)
        # 格式化cookie
        cookies_temp = []
        for i in cookie:
            cookies_temp.append('{}={}'.format(i["name"], i["value"]))
        cookies = '; '.join(cookies_temp)
        find_cookie(cookies)
        # print("cookies:{}".format(await page.cookies())) 


if __name__== "__main__":
    asyncio.get_event_loop().run_until_complete(main()) #调用