58 lines
2.1 KiB
Python
58 lines
2.1 KiB
Python
|
import asyncio
|
|||
|
|
import json
|
|||
|
|
import os
|
|||
|
|
from playwright.async_api import async_playwright
|
|||
|
|
from utils.stealth import stealth_async
|
|||
|
|
|
|||
|
|
class Authenticator:
|
|||
|
|
def __init__(self, auth_file="auth_state.json"):
|
|||
|
|
self.auth_file = auth_file
|
|||
|
|
|
|||
|
|
async def login(self, initial_url="https://weidian.com/"):
|
|||
|
|
"""
|
|||
|
|
打开浏览器由用户手动扫码登录。
|
|||
|
|
登录成功后保存 session state。
|
|||
|
|
"""
|
|||
|
|
async with async_playwright() as p:
|
|||
|
|
browser = await p.chromium.launch(headless=False, args=['--disable-gpu'])
|
|||
|
|
# 模拟 iPhone 13 配置以确保显示手机版界面
|
|||
|
|
device = p.devices['iPhone 13']
|
|||
|
|
context = await browser.new_context(**device)
|
|||
|
|
page = await context.new_page()
|
|||
|
|
await stealth_async(page)
|
|||
|
|
|
|||
|
|
await page.goto(initial_url)
|
|||
|
|
print("请在浏览器中完成登录（扫码等操作）...")
|
|||
|
|
|
|||
|
|
# 等待用户手动登录，直到 cookie 中出现登录凭证或 URL 变化
|
|||
|
|
# 这里简单处理，等待用户按回车确认已登录
|
|||
|
|
await asyncio.get_event_loop().run_in_executor(None, input, "登录完成后请在这里按回车控制台继续...")
|
|||
|
|
|
|||
|
|
# 保存状态
|
|||
|
|
storage = await context.storage_state(path=self.auth_file)
|
|||
|
|
print(f"登录状态已保存至 {self.auth_file}")
|
|||
|
|
|
|||
|
|
await browser.close()
|
|||
|
|
|
|||
|
|
def has_auth(self):
|
|||
|
|
return os.path.exists(self.auth_file)
|
|||
|
|
|
|||
|
|
async def get_context(self, playwright_instance, headless=False):
|
|||
|
|
"""
|
|||
|
|
创建一个带有已保存状态的 context，并模拟手机环境
|
|||
|
|
"""
|
|||
|
|
browser = await playwright_instance.chromium.launch(headless=headless, args=['--disable-gpu'])
|
|||
|
|
|
|||
|
|
# 模拟 iPhone 13 配置
|
|||
|
|
device = playwright_instance.devices['iPhone 13']
|
|||
|
|
|
|||
|
|
if self.has_auth():
|
|||
|
|
context = await browser.new_context(
|
|||
|
|
**device,
|
|||
|
|
storage_state=self.auth_file
|
|||
|
|
)
|
|||
|
|
else:
|
|||
|
|
context = await browser.new_context(**device)
|
|||
|
|
|
|||
|
|
return browser, context
|