"""
龙回头选股器测试脚本
运行: cd backend && python ../tests/test_pullback_selector.py
"""
import asyncio
import sys
import os
from pathlib import Path

# 添加 backend 目录到 Python 路径
backend_dir = Path(__file__).parent.parent / "backend"
sys.path.insert(0, str(backend_dir))

# 切换到 backend 目录（确保 .env 能被找到）
os.chdir(backend_dir)

from app.config import get_settings
from app.astock_agent.pullback_selector import get_pullback_selector
import logging

# 启用详细日志
logging.getLogger('stock_agent').setLevel(logging.DEBUG)


async def test_pullback_selector():
    """测试龙回头选股器"""
    print("=" * 60)
    print("龙回头选股器测试")
    print("=" * 60)

    settings = get_settings()
    print(f"\n配置:")
    print(f"  Tushare Token: {'已配置' if settings.tushare_token else '未配置'}")
    print(f"  选股启用: {settings.pullback_selector_enabled}")
    print(f"  检查板块数: {settings.pullback_sectors_to_check}")

    if not settings.tushare_token:
        print("\n❌ 错误: 请先在 .env 中配置 TUSHARE_TOKEN")
        return

    # 创建选股器
    selector = get_pullback_selector()

    print(f"\n开始选股...")
    print(f"检查板块数: {settings.pullback_sectors_to_check}")
    print(f"每个板块最多选: 2 只\n")

    # 执行选股
    results = selector.select_from_hot_sectors(top_n=settings.pullback_sectors_to_check)

    # 显示结果
    if results:
        print("\n" + "=" * 60)
        print("选股结果")
        print("=" * 60)

        for sector_name, stocks in results.items():
            print(f"\n【{sector_name}】")
            for stock in stocks:
                print(f"  📈 {stock['name']}({stock['ts_code']})")
                print(f"     现价: ¥{stock['close']:.2f} | 回踩: {stock['pullback_pct']:.2f}% | 涨幅: {stock['rise_pct']:.2f}%")
                print(f"     MA5: ¥{stock['ma5']:.2f} | MA10: ¥{stock['ma10']:.2f} | MA30: ¥{stock['ma30']:.2f} | MA60: ¥{stock['ma60']:.2f}")
                print(f"     缩量比: {stock['volume_shrink_ratio']:.0%} | 再放量: {stock['recent_volume_ratio']:.2f}x")

        total = sum(len(stocks) for stocks in results.values())
        print(f"\n共选出 {total} 只股票")

        # 格式化完整消息
        print("\n" + "=" * 60)
        print("通知消息预览")
        print("=" * 60)
        print(selector.format_result(results))

    else:
        print("\n未选出符合条件的股票")

    print("\n" + "=" * 60)
    print("测试完成")
    print("=" * 60)


if __name__ == "__main__":
    asyncio.run(test_pullback_selector())