1#!/usr/bin/env python3
2# -*- coding: utf-8 -*-
3 
4import argparse
5from typing import Any, Dict
6 
7from adapters import AkshareAdapter
8from formatter import render_output
9from router import (
10    DERIVATIVES,
11    FUND_BOND,
12    FUNDAMENTAL,
13    HK_US_MARKET,
14    INDEX_REALTIME,
15    INTRADAY_ANALYSIS,
16    KLINE_ANALYSIS,
17    KLINE_CHART,
18    LIMIT_STATS,
19    MARGIN_LHB,
20    MONEY_FLOW,
21    NEWS,
22    RESEARCH_REPORT,
23    SECTOR_ANALYSIS,
24    STOCK_OVERVIEW,
25    STOCK_PICK,
26    VOLUME_ANALYSIS,
27    HELP,
28    PORTFOLIO,
29    parse_query,
30)
31 
32 
33def dispatch(intent_obj, adapter: AkshareAdapter) -> Dict[str, Any]:
34    if intent_obj.intent == INDEX_REALTIME:
35        return adapter.index_spot(top_n=300)
36 
37    if intent_obj.intent == KLINE_ANALYSIS:
38        top_n = intent_obj.top_n or 10
39        symbol = intent_obj.symbol or "000001"
40        period = intent_obj.period or "daily"
41        return adapter.stock_kline(symbol=symbol, period=period, top_n=top_n)
42 
43    if intent_obj.intent == KLINE_CHART:
44        symbol = intent_obj.symbol or "000001"
45        period = intent_obj.period or "daily"
46        days = intent_obj.top_n or 30
47        return adapter.stock_chart(symbol=symbol, period=period, days=days)
48 
49    if intent_obj.intent == INTRADAY_ANALYSIS:
50        top_n = intent_obj.top_n or 30
51        symbol = intent_obj.symbol or "000001"
52        period = intent_obj.period if intent_obj.period in {"1", "5", "15", "30", "60"} else "1"
53        return adapter.stock_intraday(symbol=symbol, period=period, top_n=top_n)
54 
55    if intent_obj.intent == VOLUME_ANALYSIS:
56        # 调用 a-stock-analysis 脚本进行量能分析
57        symbol = intent_obj.symbol or "000001"
58        import subprocess
59        import os
60        script_path = os.path.join(os.path.dirname(os.path.dirname(__file__)), "a-stock-analysis", "scripts", "analyze.py")
61        result = subprocess.run(
62            ["python3", script_path, symbol, "--minute"],
63            capture_output=True, text=True, timeout=30
64        )
65        if result.returncode == 0:
66            return {"ok": True, "text": result.stdout}
67        else:
68            return {"ok": False, "error": result.stderr}
69 
70    if intent_obj.intent == LIMIT_STATS:
71        top_n = intent_obj.top_n or 20
72        return adapter.limit_pool(date=intent_obj.date, top_n=top_n)
73 
74    if intent_obj.intent == STOCK_OVERVIEW:
75        symbol = intent_obj.symbol
76        if not symbol:
77            return {
78                "ok": False,
79                "error": "请输入股票代码或名称，如：茅台怎么样、宁德时代分析",
80                "intent": "STOCK_OVERVIEW",
81            }
82        return adapter.stock_overview(symbol=symbol)
83 
84    if intent_obj.intent == MONEY_FLOW:
85        top_n = intent_obj.top_n or 10
86        query = intent_obj.query or ""
87        if any(k in query for k in ["北向", "南向", "东向", "市场资金", "大盘资金"]):
88            return adapter.market_money_flow(top_n=top_n, date=intent_obj.date)
89        if any(k in query for k in ["行业资金", "板块资金", "行业流入", "板块流入"]):
90            return adapter.sector_money_flow(top_n=top_n)
91        symbol = intent_obj.symbol
92        if not symbol:
93            return {
94                "ok": False,
95                "error": "请输入股票代码或名称，如：茅台资金流向、600519资金流",
96                "intent": "MONEY_FLOW",
97            }
98        return adapter.money_flow(symbol=symbol, top_n=top_n)
99 
100    if intent_obj.intent == FUNDAMENTAL:
101        top_n = intent_obj.top_n or 20
102        symbol = intent_obj.symbol
103        if not symbol:
104            return {
105                "ok": False,
106                "error": "请输入股票代码或名称，如：茅台财务指标、600519基本面",
107                "intent": "FUNDAMENTAL",
108            }
109        return adapter.fundamental(symbol=symbol, top_n=top_n)
110 
111    if intent_obj.intent == MARGIN_LHB:
112        top_n = intent_obj.top_n or 10
113        return adapter.margin_lhb(symbol=intent_obj.symbol, date=intent_obj.date, top_n=top_n)
114 
115    if intent_obj.intent == NEWS:
116        top_n = min(intent_obj.top_n or 10, 10)
117        return adapter.news(top_n=top_n)
118 
119    if intent_obj.intent == RESEARCH_REPORT:
120        top_n = min(intent_obj.top_n or 10, 10)
121        symbol = intent_obj.symbol
122        if not symbol:
123            return {
124                "ok": False,
125                "error": "请输入股票代码或名称，如：宁德时代研报、300750机构评级",
126                "intent": "RESEARCH_REPORT",
127            }
128        return adapter.research_report(symbol=symbol, top_n=top_n)
129 
130    if intent_obj.intent == STOCK_PICK:
131        query = intent_obj.query or ""
132        # 提取板块关键词
133        sector = None
134        sector_keywords = [
135            "半导体", "电子", "汽车", "医药生物", "医药",
136            "银行", "保险", "证券", "金融",
137            "房地产", "地产", "电力", "传媒",
138            "锂电池", "电池", "光伏", "光伏设备",
139            "软件", "军工", "食品", "饮料", "白酒", "家电", "纺织"
140        ]
141        for kw in sector_keywords:
142            if kw in query:
143                sector = kw
144                break
145        return adapter.stock_pick(top_n=5, sector=sector)
146 
147    if intent_obj.intent == SECTOR_ANALYSIS:
148        top_n = intent_obj.top_n or 10
149        query = intent_obj.query or ""
150        if any(k in query for k in ["概念", "题材"]):
151            return adapter.sector_analysis(sector_type="concept", top_n=top_n)
152        return adapter.sector_analysis(sector_type="industry", top_n=top_n)
153 
154    if intent_obj.intent == FUND_BOND:
155        top_n = intent_obj.top_n or 10
156        query = (intent_obj.query or "").lower()
157        scope = "bond" if any(k in query for k in ["可转债", "转债", "债"]) else "fund"
158        return adapter.fund_bond(scope=scope, symbol=intent_obj.symbol, top_n=top_n)
159 
160    if intent_obj.intent == HK_US_MARKET:
161        top_n = intent_obj.top_n or 10
162        query = (intent_obj.query or "").lower()
163        us_tokens = ["美股", "nasdaq", "dow", "道琼斯", "标普", "sp500", "s&p", "纳指", "us"]
164        market = "us" if any(token in query for token in us_tokens) else "hk"
165        return adapter.hk_us_market(market=market, top_n=top_n, symbol=intent_obj.symbol)
166 
167    if intent_obj.intent == DERIVATIVES:
168        top_n = intent_obj.top_n or 10
169        query = intent_obj.query or ""
170        scope = "options" if any(k in query for k in ["期权", "option", "Option", "OPTIONS"]) else "futures"
171        return adapter.derivatives(scope=scope, symbol=intent_obj.symbol, top_n=top_n)
172 
173    if intent_obj.intent == HELP:
174        return {
175            "ok": True,
176            "source": "help",
177            "text": """📈 A股分析 Skill 使用指南
178 
179| 类型 | 示例 |
180|------|------|
181| 大盘 | A股大盘、上证指数 |
182| 分时量能 | 茅台量能分析、600519放量分析 |
183| K线 | 茅台近30日K线、600519周线 |
184| K线图 | 茅台走势图、宁德时代K线图 |
185| 涨跌停 | 今日涨停、跌停统计 |
186| 资金流 | 茅台资金流向、市场资金流向 |
187| 基本面 | 茅台财务指标、ROE |
188| 个股综合 | 茅台怎么样、宁德时代分析 |
189| 板块 | 行业板块涨跌、概念板块涨跌 |
190| 股票推荐 | 推荐股票、半导体股票推荐 |
191| 基金/可转债 | 基金净值、可转债行情 |
192| 港股 | 港股行情 |
193| 新闻 | 财经新闻、宁德时代研报 |
194| 持仓管理 | 我的持仓、添加持仓 600519 --cost 10.5 --qty 1000、持仓分析 |
195 
196直接发给我就能查~"""
197        }
198 
199    if intent_obj.intent == PORTFOLIO:
200        import subprocess
201        import os
202        portfolio_script = os.path.join(os.path.dirname(__file__), "..", "a-stock-analysis", "scripts", "portfolio.py")
203        
204        query = intent_obj.query or ""
205        
206        # 解析持仓命令
207        if "添加" in query or "add" in query.lower():
208            # 提取代码、成本、数量
209            import re
210            code_match = re.search(r"\b(\d{6})\b", query)
211            cost_match = re.search(r"--?cost\s*(\d+\.?\d*)", query)
212            qty_match = re.search(r"--?qty\s*(\d+)", query) or re.search(r"数量\s*(\d+)", query)
213            
214            if code_match and cost_match and qty_match:
215                code = code_match.group(1)
216                cost = cost_match.group(1)
217                qty = qty_match.group(1)
218                result = subprocess.run(
219                    ["python3", portfolio_script, "add", code, "--cost", cost, "--qty", qty],
220                    capture_output=True, text=True, timeout=10
221                )
222                return {"ok": True, "source": "portfolio", "text": result.stdout or "已添加持仓"}
223            else:
224                return {"ok": False, "error": "请输入：添加持仓 代码 --cost 成本价 --qty 数量\n例如：添加持仓 600519 --cost 10.5 --qty 1000"}
225        
226        elif "分析" in query:
227            result = subprocess.run(
228                ["python3", portfolio_script, "analyze"],
229                capture_output=True, text=True, timeout=60
230            )
231            return {"ok": True, "source": "portfolio", "text": result.stdout or "暂无持仓"}
232        
233        elif "删除" in query or "移除" in query:
234            import re
235            code_match = re.search(r"\b(\d{6})\b", query)
236            if code_match:
237                code = code_match.group(1)
238                result = subprocess.run(
239                    ["python3", portfolio_script, "remove", code],
240                    capture_output=True, text=True, timeout=10
241                )
242                return {"ok": True, "source": "portfolio", "text": result.stdout or "已删除"}
243            else:
244                return {"ok": False, "error": "请输入要删除的股票代码"}
245        
246        else:
247            # 显示持仓
248            result = subprocess.run(
249                ["python3", portfolio_script, "show"],
250                capture_output=True, text=True, timeout=10
251            )
252            return {"ok": True, "source": "portfolio", "text": result.stdout or "暂无持仓"}
253 
254    return {
255        "ok": True,
256        "source": "framework",
257        "message": "该意图已识别，当前阶段先返回基础占位结果",
258        "intent": intent_obj.intent,
259        "parsed": {
260            "symbol": intent_obj.symbol,
261            "date": intent_obj.date,
262            "period": intent_obj.period,
263            "top_n": intent_obj.top_n,
264        },
265    }
266 
267 
268def main() -> None:
269    parser = argparse.ArgumentParser(description="A股分析 Skill 基础框架")
270    parser.add_argument("--query", required=True, help="自然语言请求，例如：分析 600519 最近 30 天 K线")
271    parser.add_argument("--platform", default="qq", choices=["qq", "telegram"], help="输出平台")
272    args = parser.parse_args()
273 
274    intent_obj = parse_query(args.query)
275    adapter = AkshareAdapter()
276    result = dispatch(intent_obj, adapter)
277    output = render_output(intent_obj, result, platform=args.platform)
278    print(output)
279 
280 
281if __name__ == "__main__":
282    main()
283