#!/usr/bin/env python3
"""
A股板块盘后分析系统
获取东方财富板块数据，生成分析报告，发送邮件通知
"""

import urllib.request
import json
import os
import sys
import subprocess
from datetime import datetime
from typing import List, Dict, Optional
from pathlib import Path

# 清除代理环境变量（解决代理问题）
for proxy_var in ['http_proxy', 'https_proxy', 'HTTP_PROXY', 'HTTPS_PROXY']:
    os.environ.pop(proxy_var, None)

# 配置
SCRIPT_DIR = Path(__file__).parent
DATA_DIR = SCRIPT_DIR / "data"
DATA_DIR.mkdir(exist_ok=True)

# 东方财富API配置
EASTMONEY_BASE_URL = "http://push2.eastmoney.com/api/qt/clist/get"

# 板块类型
BOARD_TYPES = {
    "industry": "m:90+t:2",      # 行业板块
    "concept": "m:90+t:3",       # 概念板块
}

# 数据字段
FIELDS = "f12,f14,f2,f3,f62,f66,f84,f104,f125,f126,f127,f128"


def get_board_data(board_type: str, sort_by: str = "f3", limit: int = 100) -> Optional[List[Dict]]:
    """
    获取板块数据
    
    参数:
        board_type: 板块类型 (industry/concept)
        sort_by: 排序字段 (f3=涨跌幅, f62=主力资金)
        limit: 返回数量
    
    返回:
        List[Dict]: 板块数据列表
    """
    fs = BOARD_TYPES.get(board_type)
    if not fs:
        print(f"❌ 未知的板块类型: {board_type}")
        return None
    
    url = f"{EASTMONEY_BASE_URL}?fid={sort_by}&po=1&pz={limit}&pn=1&np=1&fltt=2&invt=2&fs={fs}&fields={FIELDS}"
    
    try:
        req = urllib.request.Request(url, headers={
            'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36',
            'Referer': 'http://quote.eastmoney.com/'
        })
        
        with urllib.request.urlopen(req, timeout=15) as resp:
            data = json.loads(resp.read().decode())
            
            if data.get('data') and data['data'].get('diff'):
                items = data['data']['diff']
                boards = []
                
                for item in items:
                    board = {
                        'code': item.get('f12', ''),
                        'name': item.get('f14', ''),
                        'price': item.get('f2', 0) / 100 if item.get('f2') else 0,
                        'pct_change': item.get('f3', 0) / 100 if item.get('f3') else 0,
                        'main_flow': item.get('f62', 0) / 1e8 if item.get('f62') else 0,  # 亿元
                        'leader_code': item.get('f84', ''),
                        'leader_name': item.get('f104', ''),
                    }
                    boards.append(board)
                
                return boards
            else:
                print(f"⚠️ API返回数据为空")
                return None
                
    except urllib.error.URLError as e:
        print(f"❌ 网络请求失败: {e}")
        return None
    except Exception as e:
        print(f"❌ 获取数据异常: {e}")
        return None


def generate_daily_report(boards_data: Dict, to_email: str = "wlq@tphai.com") -> bool:
    """
    生成盘后分析报告并发送邮件
    
    参数:
        boards_data: 板块数据字典 {'industry': [], 'concept': []}
        to_email: 收件人邮箱
    
    返回:
        bool: 是否发送成功
    """
    all_industry = boards_data.get('industry', [])
    all_concept = boards_data.get('concept', [])
    
    if not all_industry and not all_concept:
        print("❌ 无数据，无法生成报告")
        return False
    
    # 分析总结
    timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
    
    # 计算市场趋势
    avg_pct = 0
    if all_industry:
        avg_pct = sum(b['pct_change'] for b in all_industry) / len(all_industry)
    
    market_trend = '平稳'
    if avg_pct > 0.5:
        market_trend = '上涨'
    elif avg_pct < -0.5:
        market_trend = '下跌'
    
    # 排序数据
    industry_by_pct = sorted(all_industry, key=lambda x: x['pct_change'], reverse=True)
    industry_by_flow = sorted(all_industry, key=lambda x: x['main_flow'], reverse=True)
    concept_by_pct = sorted(all_concept, key=lambda x: x['pct_change'], reverse=True)
    concept_by_flow = sorted(all_concept, key=lambda x: x['main_flow'], reverse=True)
    
    # 生成HTML正文（分析总结）
    html_lines = [
        "<h2>📊 A股板块盘后分析报告</h2>",
        f"<p>报告时间: {timestamp}</p>",
        f"<p>市场整体: <strong>{market_trend}</strong> (行业平均涨跌 {avg_pct:+.2f}%)</p>",
        "",
        "<hr>",
        "",
        "<h3>🔥 今日热门概念板块 TOP5</h3>",
        "<table border='1' cellpadding='6' cellspacing='0' style='border-collapse: collapse;'>",
        "<tr style='background:#f0f0f0'><th>板块</th><th>涨跌幅</th><th>主力资金</th></tr>",
    ]
    
    for board in concept_by_pct[:5]:
        pct_str = f"+{board['pct_change']:.2f}%" if board['pct_change'] > 0 else f"{board['pct_change']:.2f}%"
        flow_str = f"+{board['main_flow']:.2f}亿" if board['main_flow'] > 0 else f"{board['main_flow']:.2f}亿"
        html_lines.append(f"<tr><td>{board['name']}</td><td>{pct_str}</td><td>{flow_str}</td></tr>")
    
    html_lines.append("</table>")
    html_lines.append("")
    
    html_lines.append("<h3>📈 行业板块涨幅 TOP5</h3>")
    html_lines.append("<table border='1' cellpadding='6' cellspacing='0' style='border-collapse: collapse;'>")
    html_lines.append("<tr style='background:#f0f0f0'><th>板块</th><th>涨跌幅</th><th>主力资金</th><th>领涨股</th></tr>")
    
    for board in industry_by_pct[:5]:
        pct_str = f"+{board['pct_change']:.2f}%" if board['pct_change'] > 0 else f"{board['pct_change']:.2f}%"
        flow_str = f"+{board['main_flow']:.2f}亿" if board['main_flow'] > 0 else f"{board['main_flow']:.2f}亿"
        html_lines.append(f"<tr><td>{board['name']}</td><td>{pct_str}</td><td>{flow_str}</td><td>{board['leader_name'] or '-'}</td></tr>")
    
    html_lines.append("</table>")
    html_lines.append("")
    
    html_lines.append("<h3>📉 行业板块跌幅 TOP5</h3>")
    html_lines.append("<table border='1' cellpadding='6' cellspacing='0' style='border-collapse: collapse;'>")
    html_lines.append("<tr style='background:#f0f0f0'><th>板块</th><th>涨跌幅</th><th>主力资金</th></tr>")
    
    for board in industry_by_pct[-5:]:
        pct_str = f"+{board['pct_change']:.2f}%" if board['pct_change'] > 0 else f"{board['pct_change']:.2f}%"
        flow_str = f"+{board['main_flow']:.2f}亿" if board['main_flow'] > 0 else f"{board['main_flow']:.2f}亿"
        html_lines.append(f"<tr><td>{board['name']}</td><td>{pct_str}</td><td>{flow_str}</td></tr>")
    
    html_lines.append("</table>")
    html_lines.append("")
    
    html_lines.append("<h3>💰 主力资金大幅流入 TOP10</h3>")
    html_lines.append("<table border='1' cellpadding='6' cellspacing='0' style='border-collapse: collapse;'>")
    html_lines.append("<tr style='background:#f0f0f0'><th>板块</th><th>资金流入(亿)</th><th>涨跌幅</th></tr>")
    
    inflow_boards = [b for b in industry_by_flow if b['main_flow'] > 10][:10]
    for board in inflow_boards:
        pct_str = f"+{board['pct_change']:.2f}%" if board['pct_change'] > 0 else f"{board['pct_change']:.2f}%"
        html_lines.append(f"<tr><td>{board['name']}</td><td>+{board['main_flow']:.2f}</td><td>{pct_str}</td></tr>")
    
    html_lines.append("</table>")
    html_lines.append("")
    
    html_lines.append("<h3>💸 主力资金大幅流出 TOP10</h3>")
    html_lines.append("<table border='1' cellpadding='6' cellspacing='0' style='border-collapse: collapse;'>")
    html_lines.append("<tr style='background:#f0f0f0'><th>板块</th><th>资金流出(亿)</th><th>涨跌幅</th></tr>")
    
    outflow_boards = [b for b in industry_by_flow if b['main_flow'] < -10][:10]
    for board in outflow_boards:
        pct_str = f"+{board['pct_change']:.2f}%" if board['pct_change'] > 0 else f"{board['pct_change']:.2f}%"
        html_lines.append(f"<tr><td>{board['name']}</td><td>{board['main_flow']:.2f}</td><td>{pct_str}</td></tr>")
    
    html_lines.append("</table>")
    html_lines.append("")
    html_lines.append("<hr>")
    html_lines.append("<p><em>📊 详细数据请查看附件 CSV 文件</em></p>")
    
    html_body = "\n".join(html_lines)
    
    # 生成附件文件（CSV格式）
    attachment_file = DATA_DIR / f"board_detail_{datetime.now().strftime('%Y%m%d')}.csv"
    
    csv_lines = [
        "# A股板块详细数据",
        f"# 生成时间: {timestamp}",
        "",
        "=== 行业板块涨跌幅排行 ===",
        "板块名称,涨跌幅(%),主力资金(亿),领涨股",
    ]
    
    for board in industry_by_pct:
        csv_lines.append(f"{board['name']},{board['pct_change']:.2f},{board['main_flow']:.2f},{board['leader_name'] or ''}")
    
    csv_lines.append("")
    csv_lines.append("=== 行业板块资金流向排行 ===")
    csv_lines.append("板块名称,主力资金(亿),涨跌幅(%),领涨股")
    
    for board in industry_by_flow:
        csv_lines.append(f"{board['name']},{board['main_flow']:.2f},{board['pct_change']:.2f},{board['leader_name'] or ''}")
    
    csv_lines.append("")
    csv_lines.append("=== 概念板块涨跌幅排行 ===")
    csv_lines.append("板块名称,涨跌幅(%),主力资金(亿),领涨股")
    
    for board in concept_by_pct:
        csv_lines.append(f"{board['name']},{board['pct_change']:.2f},{board['main_flow']:.2f},{board['leader_name'] or ''}")
    
    csv_lines.append("")
    csv_lines.append("=== 概念板块资金流向排行 ===")
    csv_lines.append("板块名称,主力资金(亿),涨跌幅(%),领涨股")
    
    for board in concept_by_flow:
        csv_lines.append(f"{board['name']},{board['main_flow']:.2f},{board['pct_change']:.2f},{board['leader_name'] or ''}")
    
    attachment_file.write_text("\n".join(csv_lines), encoding='utf-8')
    
    # 发送邮件
    email_script = SCRIPT_DIR.parent.parent / "skills/email/scripts/send_email.py"
    
    subject = f"【A股板块盘后分析】{datetime.now().strftime('%Y-%m-%d')}"
    
    cmd = [
        "python3", str(email_script),
        "send",
        "--to", to_email,
        "--subject", subject,
        "--body", html_body,
        "--html",
        "--attach", str(attachment_file)
    ]
    
    try:
        result = subprocess.run(cmd, capture_output=True, text=True, timeout=30)
        if result.returncode == 0:
            print(f"✅ 报告发送成功: {to_email}")
            print(f"   附件: {attachment_file}")
            return True
        else:
            print(f"❌ 发送失败: {result.stderr}")
            return False
    except Exception as e:
        print(f"❌ 发送异常: {e}")
        return False


def run_daily_report(to_email: str = "wlq@tphai.com", verbose: bool = False) -> bool:
    """
    执行盘后报告生成和发送
    
    参数:
        to_email: 收件人邮箱
        verbose: 是否显示详细日志
    
    返回:
        bool: 是否成功
    """
    if verbose:
        print(f"\n📊 A股板块盘后分析")
        print("=" * 50)
        print(f"收件人: {to_email}")
        print(f"时间: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}")
    
    # 获取所有板块数据
    boards_data = {}
    
    for board_type in ["industry", "concept"]:
        if verbose:
            print(f"\n📡 获取 {board_type} 板块数据...")
        
        boards = get_board_data(board_type, limit=100)
        
        if boards:
            boards_data[board_type] = boards
            if verbose:
                print(f"✅ 成功获取 {len(boards)} 条数据")
        else:
            print(f"❌ 获取 {board_type} 数据失败")
            boards_data[board_type] = []
    
    # 生成并发送报告
    if boards_data.get('industry') or boards_data.get('concept'):
        return generate_daily_report(boards_data, to_email)
    else:
        print("❌ 所有数据获取失败，无法生成报告")
        return False


def main():
    """命令行入口"""
    import argparse
    
    parser = argparse.ArgumentParser(description="A股板块盘后分析系统")
    subparsers = parser.add_subparsers(dest="command", help="可用命令")
    
    # 测试命令
    subparsers.add_parser("test", help="测试API连接")
    
    # 获取数据命令
    get_parser = subparsers.add_parser("get", help="获取板块数据")
    get_parser.add_argument("type", choices=["industry", "concept"], help="板块类型")
    get_parser.add_argument("--limit", type=int, default=20, help="返回数量")
    
    # 发送报告命令
    report_parser = subparsers.add_parser("report", help="生成并发送盘后报告")
    report_parser.add_argument("--to", default="wlq@tphai.com", help="收件人邮箱")
    report_parser.add_argument("-v", "--verbose", action="store_true", help="显示详细日志")
    
    args = parser.parse_args()
    
    if args.command == "test":
        print("\n🧪 测试东方财富API连接...")
        for board_type in ["industry", "concept"]:
            print(f"\n测试 {board_type} 板块...")
            boards = get_board_data(board_type, limit=5)
            if boards:
                print(f"✅ 成功获取 {len(boards)} 条数据")
                for board in boards[:3]:
                    pct_str = f"+{board['pct_change']:.2f}%" if board['pct_change'] > 0 else f"{board['pct_change']:.2f}%"
                    print(f"  - {board['name']}: {pct_str}")
            else:
                print(f"❌ {board_type} 测试失败")
    
    elif args.command == "get":
        boards = get_board_data(args.type, limit=args.limit)
        if boards:
            print(f"\n📊 {args.type} 板块数据 ({len(boards)} 条)")
            print("=" * 50)
            for board in boards:
                pct_str = f"+{board['pct_change']:.2f}%" if board['pct_change'] > 0 else f"{board['pct_change']:.2f}%"
                flow_str = f"+{board['main_flow']:.2f}亿" if board['main_flow'] > 0 else f"{board['main_flow']:.2f}亿"
                print(f"{board['name']}: {pct_str}, 主力{flow_str}")
        else:
            print("❌ 获取数据失败")
    
    elif args.command == "report":
        run_daily_report(to_email=args.to, verbose=args.verbose)
    
    else:
        parser.print_help()


if __name__ == "__main__":
    main()