#!/usr/bin/env python3
"""
A股短期题材选股 - 调试版本
用于诊断为什么没有选出股票
"""
import asyncio
import sys
import os
from datetime import datetime, timedelta

# 添加项目根目录到Python路径
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))

from app.utils.logger import logger
from app.config import get_settings
from app.astock_agent.tushare_client import get_tushare_client


async def debug_tushare_connection():
    """测试Tushare连接"""
    print("\n" + "=" * 60)
    print("📊 测试1: Tushare连接测试")
    print("=" * 60)

    try:
        settings = get_settings()
        print(f"Token配置: {'已配置' if settings.tushare_token else '未配置'}")

        if not settings.tushare_token:
            print("❌ Tushare Token未配置，请在.env文件中设置TUSHARE_TOKEN")
            return False

        ts_client = get_tushare_client(settings.tushare_token)
        print(f"✅ Tushare客户端初始化成功")

        # 测试基本API调用
        print("\n测试API调用...")

        # 测试获取概念板块
        sectors_df = ts_client.get_concept_sectors()
        print(f"概念板块数量: {len(sectors_df)}")
        if not sectors_df.empty:
            print(f"示例板块: {sectors_df.head(3)['name'].tolist()}")
        else:
            print("❌ 无法获取概念板块列表")
            return False

        return True

    except Exception as e:
        print(f"❌ Tushare连接失败: {e}")
        import traceback
        print(traceback.format_exc())
        return False


async def debug_hot_sectors():
    """测试异动板块获取"""
    print("\n" + "=" * 60)
    print("📊 测试2: 异动板块检测")
    print("=" * 60)

    try:
        settings = get_settings()
        ts_client = get_tushare_client(settings.tushare_token)

        sectors_df = ts_client.get_concept_sectors()
        print(f"总概念板块数: {len(sectors_df)}")

        # 检查异动板块
        today = datetime.now().strftime('%Y%m%d')
        print(f"当前日期: {today}")

        # 检查是否是交易日
        weekday = datetime.now().weekday()
        if weekday >= 5:
            print(f"⚠️  当前是周末（周{weekday}），可能没有最新数据")
        else:
            print(f"✅ 当前是工作日（周{weekday}）")

        # 手动检查几个热门板块
        print("\n检查前10个板块的行情...")
        check_count = min(10, len(sectors_df))

        hot_count = 0
        for idx, row in sectors_df.head(check_count).iterrows():
            ts_code = row['ts_code']
            name = row['name']

            try:
                yesterday = (datetime.now() - timedelta(days=10)).strftime('%Y%m%d')
                daily_df = ts_client.pro.ths_daily(
                    ts_code=ts_code,
                    start_date=yesterday,
                    end_date=today
                )

                if not daily_df.empty:
                    latest = daily_df.sort_values('trade_date').iloc[-1]
                    change_pct = float(latest.get('pct_change', 0))
                    trade_date = str(latest.get('trade_date', ''))

                    status = "🔥" if change_pct >= 2.0 else "📊"
                    print(f"  {status} {name}: {change_pct:+.2f}% (日期: {trade_date})")

                    if change_pct >= 2.0:
                        hot_count += 1
                else:
                    print(f"  ⚠️  {name}: 无数据")

            except Exception as e:
                print(f"  ❌ {name}: 查询失败 ({e})")

        print(f"\n找到 {hot_count} 个涨幅≥2%的板块")

        if hot_count == 0:
            print("\n⚠️  没有找到符合条件的异动板块，可能原因：")
            print("  1. 当前不是交易日（周末或节假日）")
            print("  2. 盘中时段数据未更新")
            print("  3. 市场整体表现平淡")

        return hot_count > 0

    except Exception as e:
        print(f"❌ 异动板块检测失败: {e}")
        import traceback
        print(traceback.format_exc())
        return False


async def debug_stock_screening():
    """测试个股筛选"""
    print("\n" + "=" * 60)
    print("📊 测试3: 个股筛选条件分析")
    print("=" * 60)

    print("\n筛选条件回顾：")
    print("  1. 市值: 50-500亿")
    print("  2. 换手率: 3%-15%")
    print("  3. 涨跌幅: -5% 到 +8%")
    print("  4. MA多头排列: MA5 > MA10 > MA20")
    print("  5. 量能配合: 量比 > 1.2")
    print("  6. 20日动量 > 0")
    print("  7. 距离高点回撤 < 15%")

    print("\n⚠️  如果没有选出股票，可能是因为：")
    print("  1. 市场整体不符合技术形态（没有MA多头排列的股票）")
    print("  2. 筛选条件较严格（可以尝试放宽参数）")
    print("  3. 数据时间窗口问题（需要30天以上历史数据）")

    # 建议放宽的条件
    print("\n建议放宽的参数（在当前市场环境下）：")
    print("  - 换手率: 1%-15% (降低下限)")
    print("  - 涨跌幅: -7% 到 +10% (扩大范围)")
    print("  - 市值: 30-500亿 (降低下限)")


async def debug_selector_run():
    """尝试运行选股器并显示详细信息"""
    print("\n" + "=" * 60)
    print("📊 测试4: 运行选股器（详细日志）")
    print("=" * 60)

    try:
        from app.astock_agent.short_term_thematic_selector import get_thematic_selector

        settings = get_settings()
        ts_client = get_tushare_client(settings.tushare_token)
        selector = get_thematic_selector(ts_client)

        # 运行选股，启用详细日志
        import logging
        logging.basicConfig(level=logging.DEBUG)

        result = selector.select_stocks(max_stocks=10)

        print(f"\n选股结果: {result['total_stocks']} 只")

        if result['total_stocks'] == 0:
            print("\n❌ 未选出股票")
            print("\n请查看上方详细日志，分析哪个环节过滤掉了股票")
        else:
            print("\n✅ 选股成功！")
            print(selector.format_output_text(result))

    except Exception as e:
        print(f"❌ 选股器运行失败: {e}")
        import traceback
        print(traceback.format_exc())


async def main():
    """主函数"""
    print("\n" + "=" * 60)
    print("🔍 A股选股器诊断工具")
    print("=" * 60)

    # 运行所有测试
    step1_ok = await debug_tushare_connection()
    if not step1_ok:
        print("\n❌ Tushare连接失败，请检查配置")
        return 1

    step2_ok = await debug_hot_sectors()
    if not step2_ok:
        print("\n⚠️  没有找到异动板块，这是正常的（取决于市场情况）")

    await debug_stock_screening()

    print("\n" + "=" * 60)
    print("📋 诊断完成")
    print("=" * 60)
    print("\n如果所有测试通过但未选出股票，说明当前市场条件不符合策略要求。")
    print("这是正常的，策略不会在市场条件不符合时强行选股。")
    print("\n建议：")
    print("  1. 在交易日15:00后运行（确保有完整数据）")
    print("  2. 或者放宽筛选条件以适应当前市场环境")
    print("")

    return 0


if __name__ == "__main__":
    exit_code = asyncio.run(main())
    sys.exit(exit_code)