71 lines
2.4 KiB
Python
71 lines
2.4 KiB
Python
"""
|
|
测试根据环境查询分组和Profile的功能
|
|
"""
|
|
from adspower_client import AdsPowerClient
|
|
from config import Config
|
|
from loguru import logger
|
|
|
|
def main():
|
|
logger.info("="*70)
|
|
logger.info("测试根据环境查询分组和Profile")
|
|
logger.info("="*70)
|
|
logger.info(f"当前环境: {Config.ENV}")
|
|
logger.info("")
|
|
|
|
# 创建客户端
|
|
client = AdsPowerClient()
|
|
|
|
# 步骤1: 查询所有分组
|
|
logger.info("步骤1: 查询所有分组")
|
|
logger.info("-"*70)
|
|
all_groups = client.list_groups()
|
|
|
|
if all_groups and all_groups.get('code') == 0:
|
|
groups = all_groups.get('data', {}).get('list', [])
|
|
logger.info(f"✅ 找到 {len(groups)} 个分组\n")
|
|
else:
|
|
logger.error("❌ 查询所有分组失败\n")
|
|
return
|
|
|
|
# 步骤2: 根据环境查询分组
|
|
logger.info("步骤2: 根据环境自动获取分组ID")
|
|
logger.info("-"*70)
|
|
group_id = client.get_group_by_env()
|
|
|
|
if group_id:
|
|
logger.info(f"✅ 获取到分组ID: {group_id}\n")
|
|
else:
|
|
logger.error("❌ 未找到对应环境的分组\n")
|
|
logger.info("提示: 请在AdsPower中创建名为 'dev' 的分组（开发环境）或 'prod' 的分组（生产环境）")
|
|
return
|
|
|
|
# 步骤3: 使用分组ID查询Profile列表
|
|
logger.info("步骤3: 使用分组ID查询Profile列表")
|
|
logger.info("-"*70)
|
|
profiles = client.list_profiles(group_id=group_id)
|
|
|
|
if profiles and profiles.get('code') == 0:
|
|
profile_list = profiles.get('data', {}).get('list', [])
|
|
logger.info(f"✅ 在分组 {group_id} 中找到 {len(profile_list)} 个Profile\n")
|
|
|
|
if profile_list:
|
|
logger.info("前5个Profile详情:")
|
|
for idx, profile in enumerate(profile_list[:5], 1):
|
|
profile_id = profile.get('profile_id', 'N/A')
|
|
profile_name = profile.get('name', 'N/A')
|
|
group_name = profile.get('group_name', 'N/A')
|
|
logger.info(f" {idx}. ID: {profile_id}")
|
|
logger.info(f" 名称: {profile_name}")
|
|
logger.info(f" 分组: {group_name}")
|
|
logger.info("")
|
|
else:
|
|
logger.error("❌ 查询Profile列表失败\n")
|
|
return
|
|
|
|
logger.info("="*70)
|
|
logger.info("测试完成！")
|
|
logger.info("="*70)
|
|
|
|
if __name__ == "__main__":
|
|
main()
|