103 lines
3.4 KiB
Python
103 lines
3.4 KiB
Python
"""
|
||
测试AdsPower查询Cookie功能
|
||
"""
|
||
import asyncio
|
||
import sys
|
||
from pathlib import Path
|
||
|
||
# 添加项目路径
|
||
sys.path.insert(0, str(Path(__file__).parent))
|
||
|
||
from fingerprint_browser import get_fingerprint_manager
|
||
from loguru import logger
|
||
|
||
async def test_get_cookies():
|
||
"""测试获取Cookie"""
|
||
logger.info("="*50)
|
||
logger.info("开始测试AdsPower Cookie查询功能")
|
||
logger.info("="*50)
|
||
|
||
manager = get_fingerprint_manager()
|
||
|
||
# 检查AdsPower状态
|
||
if not await manager.check_adspower_status():
|
||
logger.error("❌ AdsPower未运行，请先启动AdsPower")
|
||
return
|
||
|
||
logger.success("✅ AdsPower运行正常")
|
||
|
||
# 获取所有配置
|
||
logger.info("\n查询所有浏览器配置...")
|
||
profiles = await manager.get_browser_profiles()
|
||
|
||
if not profiles:
|
||
logger.error("❌ 没有找到任何浏览器配置")
|
||
return
|
||
|
||
logger.success(f"✅ 找到 {len(profiles)} 个配置")
|
||
|
||
# 显示配置列表
|
||
print("\n可用的配置列表:")
|
||
for i, profile in enumerate(profiles[:5], 1): # 只显示前5个
|
||
profile_id = profile.get('user_id') or profile.get('id')
|
||
profile_name = profile.get('name', 'unknown')
|
||
print(f"{i}. ID: {profile_id}, 名称: {profile_name}")
|
||
|
||
# 选择看起来像是小红书账号的配置
|
||
test_profile = None
|
||
for profile in profiles:
|
||
name = profile.get('name', '')
|
||
if 'XHS' in name or '小红书' in name or '1570' in name:
|
||
test_profile = profile
|
||
break
|
||
|
||
if not test_profile:
|
||
# 如果没有找到，就用第一个
|
||
test_profile = profiles[0]
|
||
|
||
profile_id = test_profile.get('user_id') or test_profile.get('id')
|
||
profile_name = test_profile.get('name', 'unknown')
|
||
|
||
logger.info("\n" + "="*50)
|
||
logger.info(f"测试配置: {profile_name} (ID: {profile_id})")
|
||
logger.info("="*50)
|
||
|
||
# 查询Cookie
|
||
logger.info("\n开始查询Cookie...")
|
||
cookies = await manager.get_profile_cookies(profile_id)
|
||
|
||
if cookies:
|
||
logger.success(f"\n✅ 成功获取Cookie!")
|
||
logger.info(f"Cookie数量: {len(cookies)}")
|
||
|
||
# 显示前3个Cookie的详细信息
|
||
print("\n前3个Cookie详情:")
|
||
for i, cookie in enumerate(cookies[:3], 1):
|
||
print(f"\n{i}. {cookie.get('name', 'unknown')}")
|
||
print(f" Domain: {cookie.get('domain', 'N/A')}")
|
||
print(f" Path: {cookie.get('path', 'N/A')}")
|
||
print(f" Value: {cookie.get('value', '')[:20]}..." if len(cookie.get('value', '')) > 20 else f" Value: {cookie.get('value', '')}")
|
||
print(f" HttpOnly: {cookie.get('httpOnly', False)}")
|
||
print(f" Secure: {cookie.get('secure', False)}")
|
||
print(f" SameSite: {cookie.get('sameSite', 'N/A')}")
|
||
|
||
# 统计Cookie域名分布
|
||
domains = {}
|
||
for cookie in cookies:
|
||
domain = cookie.get('domain', 'unknown')
|
||
domains[domain] = domains.get(domain, 0) + 1
|
||
|
||
print("\nCookie域名分布:")
|
||
for domain, count in sorted(domains.items(), key=lambda x: x[1], reverse=True)[:5]:
|
||
print(f" {domain}: {count}个")
|
||
|
||
else:
|
||
logger.error("❌ 获取Cookie失败")
|
||
|
||
logger.info("\n" + "="*50)
|
||
logger.info("测试完成")
|
||
logger.info("="*50)
|
||
|
||
if __name__ == "__main__":
|
||
asyncio.run(test_get_cookies())
|