412 lines
14 KiB
Python
412 lines
14 KiB
Python
"""
|
||
AdsPower + Playwright CDP 集成测试
|
||
演示如何通过 CDP 连接到 AdsPower 指纹浏览器
|
||
"""
|
||
|
||
from loguru import logger
|
||
from adspower_client import AdsPowerClient
|
||
from config import Config
|
||
import sys
|
||
|
||
# 配置日志
|
||
logger.remove()
|
||
logger.add(
|
||
sys.stdout,
|
||
format="<green>{time:HH:mm:ss}</green> | <level>{level: <8}</level> | <level>{message}</level>",
|
||
level="INFO"
|
||
)
|
||
|
||
|
||
def test_adspower_connection(use_proxy: bool = False, proxy_info: dict = None, use_api_v1: bool = False):
|
||
"""测试 AdsPower + Playwright CDP 连接
|
||
|
||
Args:
|
||
use_proxy: 是否使用大麦IP代理
|
||
proxy_info: 已获取的代理信息
|
||
use_api_v1: 是否使用API v1方式更新代理
|
||
"""
|
||
|
||
# ==================== 配置区 ====================
|
||
# 访问的网页地址，在这里修改
|
||
TEST_URL = "https://health.baidu.com/m/detail/ar_2366617956693492811" # IP检测网站，可查看代理是否生效
|
||
# 其他可选项：
|
||
# TEST_URL = "https://www.baidu.com" # 百度
|
||
# TEST_URL = "https://www.google.com" # Google
|
||
# TEST_URL = "https://你的MIP页面地址" # 你的目标网页
|
||
# =====================================================
|
||
|
||
client = AdsPowerClient()
|
||
|
||
try:
|
||
# 0. 先查询 Profile 列表
|
||
logger.info("=" * 60)
|
||
logger.info("步骤 0: 查询 Profile 列表")
|
||
logger.info("=" * 60)
|
||
|
||
result = client.list_profiles()
|
||
if not result:
|
||
logger.error("查询 Profile 失败")
|
||
return False
|
||
|
||
profiles = result.get('data', {}).get('list', [])
|
||
if not profiles:
|
||
logger.error("没有可用的 Profile，请先在 AdsPower 中创建 Profile")
|
||
return False
|
||
|
||
# 使用第一个 Profile
|
||
first_profile = profiles[0]
|
||
profile_id = first_profile.get('profile_id')
|
||
profile_name = first_profile.get('name', 'N/A')
|
||
|
||
logger.info(f"将使用 Profile: {profile_name} (ID: {profile_id})")
|
||
logger.info("")
|
||
|
||
# 1. 启动 AdsPower 浏览器（可选使用代理）
|
||
logger.info("=" * 60)
|
||
logger.info(f"步骤 1: {'[使用代理] ' if use_proxy else ''}启动 AdsPower 浏览器")
|
||
if use_proxy:
|
||
logger.info(f"代理更新方式: {'API v1 (直接传入proxy_config)' if use_api_v1 else 'API v2 (使用proxy_id引用)'}")
|
||
logger.info("=" * 60)
|
||
|
||
if use_proxy and proxy_info:
|
||
if use_api_v1:
|
||
# 使用 API v1 方式：直接传入 proxy_config
|
||
logger.info("使用 API v1 方式更新代理...")
|
||
proxy_config_v1 = {
|
||
"proxy_type": "http",
|
||
"proxy_host": proxy_info["host"],
|
||
"proxy_port": proxy_info["port"],
|
||
"proxy_user": client.DAMAI_USER,
|
||
"proxy_password": client.DAMAI_PASSWORD,
|
||
"proxy_soft": "other"
|
||
}
|
||
|
||
# 直接更新 Profile
|
||
success = client.update_profile_proxy_v1(profile_id, proxy_config_v1)
|
||
if not success:
|
||
logger.warning("更新代理失败 (API v1)，将不使用代理启动")
|
||
else:
|
||
# 使用 API v2 方式：先创建代理，再引用
|
||
logger.info("使用 API v2 方式更新代理...")
|
||
proxy_config = {
|
||
"type": "http",
|
||
"host": proxy_info["host"],
|
||
"port": proxy_info["port"],
|
||
"user": client.DAMAI_USER,
|
||
"password": client.DAMAI_PASSWORD,
|
||
"ipchecker": "ip2location",
|
||
"remark": "Damai Auto Proxy"
|
||
}
|
||
|
||
# 创建代理
|
||
proxy_id = client.create_proxy(proxy_config)
|
||
if proxy_id:
|
||
# 更新 Profile
|
||
client.update_profile_proxy(profile_id, proxy_id)
|
||
else:
|
||
logger.warning("创建代理失败，将不使用代理启动")
|
||
|
||
browser_info = client.start_browser(user_id=profile_id)
|
||
else:
|
||
browser_info = client.start_browser(user_id=profile_id)
|
||
if not browser_info:
|
||
logger.error("启动 AdsPower 浏览器失败")
|
||
return False
|
||
|
||
logger.info(f"浏览器信息: {browser_info}")
|
||
|
||
# 2. 通过 CDP 连接到浏览器
|
||
logger.info("")
|
||
logger.info("=" * 60)
|
||
logger.info("步骤 2: 通过 CDP 连接到浏览器")
|
||
logger.info("=" * 60)
|
||
|
||
browser = client.connect_browser(browser_info)
|
||
if not browser:
|
||
logger.error("CDP 连接失败")
|
||
return False
|
||
|
||
logger.info(f"浏览器版本: {browser.version}")
|
||
logger.info(f"上下文数量: {len(browser.contexts)}")
|
||
|
||
# 3. 获取页面
|
||
logger.info("")
|
||
logger.info("=" * 60)
|
||
logger.info("步骤 3: 获取浏览器页面")
|
||
logger.info("=" * 60)
|
||
|
||
page = client.get_page(browser)
|
||
if not page:
|
||
logger.error("获取页面失败")
|
||
return False
|
||
|
||
logger.info(f"页面 URL: {page.url}")
|
||
|
||
# 3.5. 关闭其他标签页，只保留AdsPower启动页
|
||
logger.info("")
|
||
logger.info("=" * 60)
|
||
logger.info("步骤 3.5: 清理多余标签页")
|
||
logger.info("=" * 60)
|
||
|
||
context = browser.contexts[0]
|
||
all_pages = context.pages
|
||
logger.info(f"当前打开的标签页数: {len(all_pages)}")
|
||
|
||
# 遍历所有页面，关闭非 AdsPower 启动页
|
||
closed_count = 0
|
||
for p in all_pages:
|
||
try:
|
||
page_url = p.url
|
||
# 保留 AdsPower 启动页
|
||
if 'start.adspower.net' in page_url:
|
||
logger.info(f"保留启动页: {page_url}")
|
||
else:
|
||
logger.info(f"关闭标签页: {page_url}")
|
||
p.close()
|
||
closed_count += 1
|
||
except Exception as e:
|
||
logger.warning(f"关闭页面失败: {str(e)}")
|
||
|
||
logger.info(f"已关闭 {closed_count} 个标签页")
|
||
|
||
# 重新获取当前页面列表
|
||
remaining_pages = context.pages
|
||
logger.info(f"剩余标签页数: {len(remaining_pages)}")
|
||
|
||
# 如果所有页面都被关闭了，创建一个新页面
|
||
if len(remaining_pages) == 0:
|
||
logger.info("所有页面已关闭，创建新标签页")
|
||
page = context.new_page()
|
||
else:
|
||
# 使用第一个剩余页面
|
||
page = remaining_pages[0]
|
||
logger.info(f"使用剩余页面: {page.url}")
|
||
|
||
# 4. 测试页面操作
|
||
logger.info("")
|
||
logger.info("=" * 60)
|
||
logger.info("步骤 4: 测试页面操作")
|
||
logger.info("=" * 60)
|
||
|
||
# 访问配置的网页
|
||
logger.info(f"访问测试页面: {TEST_URL}")
|
||
page.goto(TEST_URL, wait_until='domcontentloaded')
|
||
|
||
# 等待广告接口响应完成
|
||
logger.info("等待广告接口响应...")
|
||
try:
|
||
# 等待广告接口请求完成
|
||
response = page.wait_for_response(
|
||
lambda r: 'getRefreshAdAssets' in r.url and r.status == 200,
|
||
timeout=10000 # 10秒超时
|
||
)
|
||
logger.info(f"广告接口已响应: {response.url}")
|
||
|
||
# 等待DOM更新
|
||
import time
|
||
time.sleep(2)
|
||
except Exception as e:
|
||
logger.warning(f"等待广告接口超时或失败: {str(e)}")
|
||
logger.info("继续执行...")
|
||
import time
|
||
time.sleep(2)
|
||
|
||
# 获取页面标题
|
||
title = page.title()
|
||
logger.info(f"页面标题: {title}")
|
||
|
||
# 获取页面 URL
|
||
current_url = page.url
|
||
logger.info(f"当前 URL: {current_url}")
|
||
|
||
# 截图测试（点击前）
|
||
screenshot_path = "./test_screenshot_before.png"
|
||
page.screenshot(path=screenshot_path)
|
||
logger.info(f"截图已保存: {screenshot_path}")
|
||
|
||
# 查找并点击广告
|
||
logger.info("")
|
||
logger.info("-" * 60)
|
||
logger.info("开始查找广告元素...")
|
||
|
||
try:
|
||
# 查找所有广告元素
|
||
ad_selector = 'span.ec-tuiguang.ecfc-tuiguang.xz81bbe'
|
||
ad_elements = page.locator(ad_selector)
|
||
ad_count = ad_elements.count()
|
||
|
||
logger.info(f"找到 {ad_count} 个广告元素")
|
||
|
||
if ad_count > 0:
|
||
# 点击第一个广告
|
||
logger.info("准备点击第一个广告...")
|
||
|
||
# 滚动到元素可见
|
||
first_ad = ad_elements.first
|
||
first_ad.scroll_into_view_if_needed()
|
||
time.sleep(1)
|
||
|
||
# 点击广告
|
||
first_ad.click()
|
||
logger.info("✅ 已点击第一个广告")
|
||
|
||
# 等待页面跳转
|
||
time.sleep(3)
|
||
|
||
# 获取点击后的页面信息
|
||
new_url = page.url
|
||
new_title = page.title()
|
||
logger.info(f"点击后 URL: {new_url}")
|
||
logger.info(f"点击后标题: {new_title}")
|
||
|
||
# 截图（点击后）
|
||
screenshot_path_after = "./test_screenshot_after.png"
|
||
page.screenshot(path=screenshot_path_after)
|
||
logger.info(f"点击后截图已保存: {screenshot_path_after}")
|
||
else:
|
||
logger.warning("⚠ 未找到广告元素")
|
||
|
||
except Exception as e:
|
||
logger.error(f"查找/点击广告失败: {str(e)}")
|
||
# 保存错误时的截图
|
||
page.screenshot(path="./test_screenshot_error.png")
|
||
logger.info("错误截图已保存: ./test_screenshot_error.png")
|
||
|
||
# 5. 清理资源
|
||
logger.info("")
|
||
logger.info("=" * 60)
|
||
logger.info("步骤 5: 测试完成")
|
||
logger.info("=" * 60)
|
||
|
||
# 注意：不停止浏览器，保持运行状态供手动操作
|
||
logger.info("浏览器保持运行状态，可继续手动操作")
|
||
logger.info("如需停止浏览器，请在AdsPower中手动关闭")
|
||
|
||
logger.info("")
|
||
logger.info("=" * 60)
|
||
logger.info("测试完成！浏览器未关闭")
|
||
logger.info("=" * 60)
|
||
|
||
return True
|
||
|
||
except Exception as e:
|
||
logger.error(f"测试异常: {str(e)}")
|
||
return False
|
||
|
||
finally:
|
||
# 注意：不自动清理资源，保持浏览器运行
|
||
pass
|
||
|
||
|
||
def test_multiple_pages():
|
||
"""测试多页面操作"""
|
||
|
||
client = AdsPowerClient()
|
||
profile_id = None
|
||
|
||
try:
|
||
logger.info("=" * 60)
|
||
logger.info("测试多页面操作")
|
||
logger.info("=" * 60)
|
||
|
||
# 查询 Profile
|
||
result = client.list_profiles()
|
||
if not result:
|
||
return False
|
||
|
||
profiles = result.get('data', {}).get('list', [])
|
||
if not profiles:
|
||
logger.error("没有可用的 Profile")
|
||
return False
|
||
|
||
profile_id = profiles[0].get('profile_id')
|
||
|
||
# 启动并连接
|
||
browser_info = client.start_browser(user_id=profile_id)
|
||
if not browser_info:
|
||
return False
|
||
|
||
browser = client.connect_browser(browser_info)
|
||
if not browser:
|
||
return False
|
||
|
||
# 获取第一个页面
|
||
page1 = client.get_page(browser)
|
||
logger.info("访问百度...")
|
||
page1.goto("https://www.baidu.com")
|
||
logger.info(f"页面1标题: {page1.title()}")
|
||
|
||
# 创建新页面
|
||
context = browser.contexts[0]
|
||
page2 = context.new_page()
|
||
logger.info("访问必应...")
|
||
page2.goto("https://www.bing.com")
|
||
logger.info(f"页面2标题: {page2.title()}")
|
||
|
||
logger.info(f"当前打开的页面数: {len(context.pages)}")
|
||
|
||
# 关闭页面
|
||
page2.close()
|
||
logger.info("已关闭页面2")
|
||
|
||
return True
|
||
|
||
except Exception as e:
|
||
logger.error(f"测试异常: {str(e)}")
|
||
return False
|
||
|
||
finally:
|
||
try:
|
||
if profile_id:
|
||
client.stop_browser(user_id=profile_id)
|
||
else:
|
||
client.stop_browser()
|
||
except:
|
||
pass
|
||
|
||
|
||
if __name__ == "__main__":
|
||
logger.info("开始测试 AdsPower + Playwright CDP 集成")
|
||
logger.info("")
|
||
|
||
logger.info(f"当前环境: {Config.ENV}")
|
||
logger.info(f"AdsPower API: {Config.ADSPOWER_API_URL}")
|
||
logger.info("")
|
||
|
||
# 创建客户端
|
||
client = AdsPowerClient()
|
||
|
||
# ==================== 配置区 ====================
|
||
# 默认使用代理，如不需要改为 False
|
||
use_proxy = True
|
||
# 默认使用 API v2 Direct 方式（0=v2 proxy_id, 1=v1, 2=v2 direct）
|
||
use_api_v1 = True # True=API v1, False=API v2
|
||
# =====================================================
|
||
|
||
proxy_info = None
|
||
|
||
# 如果使用代理，提前获取
|
||
if use_proxy:
|
||
logger.info("")
|
||
logger.info(f"使用代理模式: {'API v1 (直接传入proxy_config)' if use_api_v1 else 'API v2 (使用proxy_id引用)'}")
|
||
logger.info("步骤 0: 提前获取大麦IP代理")
|
||
proxy_info = client.get_damai_proxy()
|
||
if not proxy_info:
|
||
logger.error("获取代理失败，终止测试")
|
||
sys.exit(1)
|
||
logger.info(f"代理地址: {proxy_info['host']}:{proxy_info['port']}")
|
||
logger.info("")
|
||
|
||
# 测试基本连接
|
||
if test_adspower_connection(use_proxy=use_proxy, proxy_info=proxy_info, use_api_v1=use_api_v1):
|
||
logger.info("\n基本连接测试通过\n")
|
||
else:
|
||
logger.error("\n基本连接测试失败\n")
|
||
sys.exit(1)
|
||
|
||
# 测试多页面操作
|
||
# if test_multiple_pages():
|
||
# logger.info("\n多页面操作测试通过\n")
|
||
# else:
|
||
# logger.error("\n多页面操作测试失败\n")
|