"""
测试根据环境查询分组和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()