68 lines
2.1 KiB
Python
68 lines
2.1 KiB
Python
|
#!/usr/bin/env python3
|
|||
|
|
# -*- coding: utf-8 -*-
|
|||
|
|
"""
|
|||
|
|
测试 playwright_stealth 修复
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
import asyncio
|
|||
|
|
from loguru import logger
|
|||
|
|
from playwright.async_api import async_playwright
|
|||
|
|
from playwright_stealth.stealth import Stealth
|
|||
|
|
|
|||
|
|
async def test_stealth():
|
|||
|
|
"""测试 stealth 功能"""
|
|||
|
|
logger.info("开始测试 playwright_stealth 功能")
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
# 测试 Stealth 类导入和实例化
|
|||
|
|
stealth = Stealth()
|
|||
|
|
logger.success("Stealth 类实例化成功")
|
|||
|
|
|
|||
|
|
# 检查 Stealth 类的方法
|
|||
|
|
methods = [m for m in dir(stealth) if 'apply' in m.lower()]
|
|||
|
|
logger.info(f"Stealth 类包含的方法: {methods}")
|
|||
|
|
|
|||
|
|
# 测试 Playwright 浏览器启动
|
|||
|
|
logger.info("测试 Playwright 浏览器启动...")
|
|||
|
|
playwright = await async_playwright().start()
|
|||
|
|
|
|||
|
|
# 启动浏览器
|
|||
|
|
browser = await playwright.chromium.launch(headless=True)
|
|||
|
|
logger.success("Playwright 浏览器启动成功")
|
|||
|
|
|
|||
|
|
# 创建页面上下文
|
|||
|
|
context = await browser.new_context()
|
|||
|
|
page = await context.new_page()
|
|||
|
|
logger.success("页面创建成功")
|
|||
|
|
|
|||
|
|
# 测试 stealth 应用
|
|||
|
|
logger.info("测试应用 stealth 设置...")
|
|||
|
|
await stealth.apply_stealth_async(page)
|
|||
|
|
logger.success("stealth 设置应用成功")
|
|||
|
|
|
|||
|
|
# 关闭浏览器
|
|||
|
|
await browser.close()
|
|||
|
|
await playwright.stop()
|
|||
|
|
logger.success("浏览器关闭成功")
|
|||
|
|
|
|||
|
|
logger.success("所有测试通过！playwright_stealth 修复成功")
|
|||
|
|
return True
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.error(f"测试失败: {e}")
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
async def main():
|
|||
|
|
"""主函数"""
|
|||
|
|
logger.info("=== playwright_stealth 修复测试 ===")
|
|||
|
|
|
|||
|
|
success = await test_stealth()
|
|||
|
|
|
|||
|
|
if success:
|
|||
|
|
logger.info("✅ 修复成功！您现在可以正常运行 new_data_stealth.py")
|
|||
|
|
logger.info("运行命令: python product/new_data_stealth.py")
|
|||
|
|
else:
|
|||
|
|
logger.error("❌ 修复失败，请检查错误信息")
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
asyncio.run(main())
|