Update Get-JDCookie.py

判断是否登录成功

Get-JDCookie.py

@@ -1,4 +1,5 @@
 import asyncio
+from socket import timeout
 import pyperclip
 import os
 from pyppeteer import launch
@@ -15,30 +16,36 @@ def find_cookie(cookies):
     jd_cookie = pt_pin+';'+pt_key+';'
     pyperclip.copy(jd_cookie)   # 拷贝JDcookie到剪切板
     print("Cookie:", jd_cookie)
-    print("已拷贝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'])   # 进入有头模式
-    page = await browser.newPage()           # 打开新的标签页
+    context = await browser.createIncognitoBrowserContext() # 隐身模式
+    page = await context.newPage()           # 打开新的标签页
     await page.setViewport({'width': 800, 'height': 700})      # 页面大小一致
     await page.goto('https://home.m.jd.com') # 访问主页
 
-    await asyncio.sleep(60)
-    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()))
+    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()) #调用
+    asyncio.get_event_loop().run_until_complete(main()) #调用