ai_mip/test_consultation_message.py

"""
测试咨询消息发送功能
"""
from loguru import logger
from adspower_client import AdsPowerClient
from ad_automation import MIPAdAutomation
from config import Config
from db_manager import SiteManager, ClickManager, InteractionManager
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 main():
    """测试主流程"""
    
    # 测试URL（请替换为实际的MIP页面）
    TEST_URL = "https://health.baidu.com/m/detail/ar_2366617956693492811"
    
    logger.info("="*70)
    logger.info("测试咨询消息发送功能")
    logger.info("="*70)
    logger.info(f"当前环境: {Config.ENV}")
    logger.info(f"测试URL: {TEST_URL}")
    logger.info("")
    
    client = AdsPowerClient()
    
    try:
        # 1. 初始化数据库，添加测试站点
        logger.info("步骤1: 初始化测试站点")
        logger.info("-"*70)
        
        site_mgr = SiteManager()
        site = site_mgr.get_site_by_url(TEST_URL)
        
        if not site:
            site_id = site_mgr.add_site(
                site_url=TEST_URL,
                site_name="测试站点-咨询消息",
                site_dimension="医疗健康"
            )
            logger.info(f"✅ 创建测试站点: site_id={site_id}")
        else:
            site_id = site['id']
            logger.info(f"✅ 使用已存在站点: site_id={site_id}")
        
        logger.info("")
        
        # 2. 获取当前环境的Profile
        logger.info("步骤2: 获取浏览器环境")
        logger.info("-"*70)
        
        group_id = client.get_group_by_env()
        if not group_id:
            logger.error("未找到对应环境的分组")
            return
        
        result = client.list_profiles(group_id=group_id)
        if not result or not result.get('data', {}).get('list'):
            logger.error("未找到可用的浏览器环境")
            return
        
        profile = result['data']['list'][0]
        profile_id = profile['profile_id']
        logger.info(f"✅ 使用Profile: {profile['name']} (ID: {profile_id})")
        logger.info("")
        
        # 3. 启动浏览器
        logger.info("步骤3: 启动浏览器")
        logger.info("-"*70)
        
        browser_info = client.start_browser(user_id=profile_id)
        if not browser_info:
            logger.error("启动浏览器失败")
            return
        
        logger.info("✅ 浏览器启动成功")
        logger.info("")
        
        # 4. 连接浏览器
        logger.info("步骤4: 连接浏览器")
        logger.info("-"*70)
        
        browser = client.connect_browser(browser_info)
        if not browser:
            logger.error("连接浏览器失败")
            return
        
        page = client.get_page(browser)
        if not page:
            logger.error("获取页面失败")
            return
        
        logger.info("✅ 浏览器连接成功")
        logger.info("")
        
        # 5. 执行广告点击和消息发送
        logger.info("步骤5: 执行广告点击和消息发送")
        logger.info("-"*70)
        
        automation = MIPAdAutomation(page)
        click_success, has_reply = automation.check_and_click_ad(TEST_URL, site_id=site_id)
        
        if click_success:
            logger.info("✅ 广告点击成功")
            if has_reply:
                logger.info("✅ 收到了医生回复")
            else:
                logger.info("⚠️  未收到医生回复（可能需要等待更长时间）")
        else:
            logger.warning("❌ 广告点击失败")
        
        logger.info("")
        
        # 6. 查询数据库记录
        logger.info("步骤6: 查询数据库记录")
        logger.info("-"*70)
        
        # 查询点击记录
        click_mgr = ClickManager()
        clicks = click_mgr.get_clicks_by_site(site_id, limit=5)
        logger.info(f"点击记录数: {len(clicks)}")
        if clicks:
            last_click = clicks[0]
            logger.info(f"最新点击: ID={last_click['id']}, 时间={last_click['click_time']}")
        
        # 查询互动记录
        interaction_mgr = InteractionManager()
        interactions = interaction_mgr.get_interactions_by_site(site_id, limit=5)
        logger.info(f"互动记录数: {len(interactions)}")
        if interactions:
            last_interaction = interactions[0]
            logger.info(f"最新互动: ID={last_interaction['id']}")
            logger.info(f"  发送内容: {last_interaction['reply_content']}")
            logger.info(f"  收到回复: {'是' if last_interaction['response_received'] else '否'}")
        
        logger.info("")
        
        # 7. 保持浏览器运行，等待手动检查
        logger.info("="*70)
        logger.info("测试完成！浏览器保持运行，可手动检查结果")
        logger.info("按 Ctrl+C 停止测试并关闭浏览器")
        logger.info("="*70)
        
        input("\n按回车键停止测试...")
        
    except KeyboardInterrupt:
        logger.info("\n用户中断测试")
    except Exception as e:
        logger.error(f"测试异常: {str(e)}")
        import traceback
        traceback.print_exc()
    finally:
        # 清理资源
        try:
            if 'profile_id' in locals():
                client.stop_browser(user_id=profile_id)
                logger.info("✅ 浏览器已关闭")
        except Exception as e:
            logger.warning(f"关闭浏览器失败: {str(e)}")


if __name__ == "__main__":
    main()