92 lines
3.6 KiB
Python
92 lines
3.6 KiB
Python
|
import asyncio
|
|||
|
|
import yaml
|
|||
|
|
import time
|
|||
|
|
from playwright.async_api import async_playwright
|
|||
|
|
from utils.stealth import stealth_async
|
|||
|
|
from utils.auth import Authenticator
|
|||
|
|
from utils.timer import PrecisionTimer
|
|||
|
|
|
|||
|
|
async def snatch(config):
|
|||
|
|
auth = Authenticator(config.get("auth_file", "auth_state.json"))
|
|||
|
|
timer = PrecisionTimer()
|
|||
|
|
timer.sync_time()
|
|||
|
|
|
|||
|
|
async with async_playwright() as p:
|
|||
|
|
browser, context = await auth.get_context(p, headless=config.get("headless", False))
|
|||
|
|
page = await context.new_page()
|
|||
|
|
await stealth_async(page)
|
|||
|
|
|
|||
|
|
target_url = config.get("target_url")
|
|||
|
|
if not target_url:
|
|||
|
|
print("错误: 未配置 target_url")
|
|||
|
|
return
|
|||
|
|
|
|||
|
|
# 1. 预热：先打开页面
|
|||
|
|
print(f"正在打开商品页面: {target_url}")
|
|||
|
|
await page.goto(target_url)
|
|||
|
|
|
|||
|
|
# 如果未登录，可能需要在这里处理扫码
|
|||
|
|
if not auth.has_auth():
|
|||
|
|
print("未发现登录状态，请手动操作并将状态保存...")
|
|||
|
|
# 注意：login 会启动一个新的浏览器窗口
|
|||
|
|
await auth.login(target_url)
|
|||
|
|
|
|||
|
|
# 登录完成后，我们需要关闭当前空的 context 并重新加载带有 cookie 的 context
|
|||
|
|
await context.close()
|
|||
|
|
await browser.close()
|
|||
|
|
browser, context = await auth.get_context(p, headless=config.get("headless", False))
|
|||
|
|
page = await context.new_page()
|
|||
|
|
await stealth_async(page)
|
|||
|
|
|
|||
|
|
print("已重新加载登录状态，正在打开商品页面...")
|
|||
|
|
await page.goto(target_url)
|
|||
|
|
|
|||
|
|
# 2. 等待抢购时间
|
|||
|
|
snatch_time = config.get("snatch_time")
|
|||
|
|
if snatch_time:
|
|||
|
|
await timer.wait_until(snatch_time)
|
|||
|
|
|
|||
|
|
# 3. 抢购核心逻辑 (H5 流程)
|
|||
|
|
# 注意：这里的选择器需要根据微店 H5 实际页面结构进行微调
|
|||
|
|
# 这里演示一般的微店抢购流程：点击购买 -> 选择规格 -> 确定 -> 提交订单
|
|||
|
|
try:
|
|||
|
|
# 刷新页面以获取最新状态（可选，视具体页面倒计时逻辑而定）
|
|||
|
|
# await page.reload()
|
|||
|
|
|
|||
|
|
# 点击“立即购买”按钮
|
|||
|
|
# 微店 H5 常见的购买按钮类名类似于 .buy-btn, .footer-buy
|
|||
|
|
# 我们尝试使用 text 匹配
|
|||
|
|
buy_button = page.get_by_text("立即购买")
|
|||
|
|
await buy_button.click()
|
|||
|
|
print("点击立即购买")
|
|||
|
|
|
|||
|
|
# 处理 SKU 选择（如果弹出 SKU 选择框）
|
|||
|
|
# 这里简单起见，如果弹出了规格选择，点击第一个选项并确定
|
|||
|
|
# 实际需要根据 config 中的 sku_id 进行精准点击
|
|||
|
|
confirm_btn = page.get_by_text("确定")
|
|||
|
|
if await confirm_btn.is_visible():
|
|||
|
|
await confirm_btn.click()
|
|||
|
|
print("点击确定（SKU）")
|
|||
|
|
|
|||
|
|
# 进入确认订单页面后，点击“提交订单”
|
|||
|
|
# 提交订单按钮通常在底部，文字为“提交订单”
|
|||
|
|
submit_btn = page.get_by_text("提交订单")
|
|||
|
|
await submit_btn.wait_for(state="visible", timeout=5000)
|
|||
|
|
await submit_btn.click()
|
|||
|
|
print("点击提交订单！抢购请求已发送！")
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"抢购过程中发生错误: {e}")
|
|||
|
|
|
|||
|
|
# 保持浏览器打开一段时间查看结果
|
|||
|
|
await asyncio.sleep(10)
|
|||
|
|
await browser.close()
|
|||
|
|
|
|||
|
|
def load_config():
|
|||
|
|
with open("config.yaml", "r", encoding="utf-8") as f:
|
|||
|
|
return yaml.safe_load(f)
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
config = load_config()
|
|||
|
|
asyncio.run(snatch(config))
|