aggroot 1.0.0

package/share/scripts/stock_compare.py

@@ -0,0 +1,244 @@
+"""
+股票对比分析
+
+用法: python stock_compare.py <stock_codes> [--metrics <list>]
+  stock_codes: 逗号分隔的股票代码，如 600519,000858(A股) 或 03888,00700(港股)
+  --metrics:   对比指标，逗号分隔（默认全部）
+
+AKShare接口:
+  A股主: ak.stock_zh_a_spot() — 新浪全市场行情（稳定可用）
+  A股备: ak.stock_zh_a_spot_em() — 东方财富全市场行情
+  港股:   ak.stock_hk_financial_indicator_em(symbol) — 逐只获取财务指标对比
+
+注意: 新浪接口代码列含市场前缀(sh/sz/bj)，频繁调用会被封IP
+      港股对比为逐只获取，速度较慢但数据丰富
+"""
+
+import sys
+import argparse
+
+sys.path.insert(0, '.')
+from _utils import normalize_code, get_market_prefix, output_json, output_error, check_akshare, safe_float, is_hk_code
+
+
+# A股可对比的指标（东方财富列名）
+METRIC_MAP_EM = {
+    "pe_ratio": "市盈率-动态",
+    "pb_ratio": "市净率",
+    "total_market_cap": "总市值",
+    "circulating_market_cap": "流通市值",
+    "turnover_rate": "换手率",
+    "change_pct": "涨跌幅",
+    "volume_ratio": "量比",
+    "amplitude": "振幅",
+}
+
+# A股新浪接口可用的指标
+METRIC_MAP_SINA = {
+    "change_pct": "涨跌幅",
+    "amount": "成交额",
+    "high": "最高",
+    "low": "最低",
+    "open": "今开",
+    "prev_close": "昨收",
+    "volume": "成交量",
+}
+
+# 港股可对比的指标
+METRIC_MAP_HK = {
+    "pe_ratio": "市盈率",
+    "pb_ratio": "市净率",
+    "market_cap": "总市值(港元)",
+    "roe": "股东权益回报率(%)",
+    "net_margin": "销售净利率(%)",
+    "net_profit_growth": "净利润滚动环比增长(%)",
+    "revenue_growth": "营业总收入滚动环比增长(%)",
+    "dividend_yield": "股息率TTM(%)",
+}
+
+ALL_A_METRICS = list(METRIC_MAP_EM.keys())
+ALL_HK_METRICS = list(METRIC_MAP_HK.keys())
+
+
+def _fetch_em_batch(codes, metrics):
+    """东方财富A股批量获取"""
+    import akshare as ak
+
+    df = ak.stock_zh_a_spot_em()
+    stocks = []
+    for code in codes:
+        row = df[df['代码'] == code]
+        if row.empty:
+            stocks.append({"code": code, "name": "未找到", "error": f"未找到股票代码 {code}"})
+            continue
+
+        r = row.iloc[0]
+        stock_data = {
+            "code": code,
+            "name": str(r.get('名称', '')),
+            "price": safe_float(r.get('最新价')),
+            "change_pct": safe_float(r.get('涨跌幅')),
+        }
+        for metric_key in metrics:
+            col_name = METRIC_MAP_EM.get(metric_key)
+            if col_name:
+                stock_data[metric_key] = safe_float(r.get(col_name))
+        stocks.append(stock_data)
+    return stocks, metrics, [METRIC_MAP_EM[m] for m in metrics if m in METRIC_MAP_EM]
+
+
+def _fetch_sina_batch(codes, metrics):
+    """新浪A股批量获取（主接口）"""
+    import akshare as ak
+
+    df = ak.stock_zh_a_spot()
+    stocks = []
+    sina_metrics = [m for m in metrics if m in METRIC_MAP_SINA]
+    for code in codes:
+        prefix = get_market_prefix(code)
+        sina_code = f"{prefix}{code}"
+        row = df[df['代码'] == sina_code]
+        if row.empty:
+            stocks.append({"code": code, "name": "未找到", "error": f"未找到股票代码 {code}"})
+            continue
+
+        r = row.iloc[0]
+        stock_data = {
+            "code": code,
+            "name": str(r.get('名称', '')),
+            "price": safe_float(r.get('最新价')),
+            "change_pct": safe_float(r.get('涨跌幅')),
+            "data_source": "sina",
+        }
+        for metric_key in sina_metrics:
+            col_name = METRIC_MAP_SINA.get(metric_key)
+            if col_name:
+                stock_data[metric_key] = safe_float(r.get(col_name))
+        stocks.append(stock_data)
+    return stocks, sina_metrics, [METRIC_MAP_SINA[m] for m in sina_metrics if m in METRIC_MAP_SINA]
+
+
+def _fetch_hk_batch(codes, metrics):
+    """港股逐只获取财务指标对比"""
+    import akshare as ak
+
+    stocks = []
+    hk_metrics = [m for m in metrics if m in METRIC_MAP_HK]
+    for code in codes:
+        try:
+            fi = ak.stock_hk_financial_indicator_em(symbol=code)
+            if fi.empty:
+                stocks.append({"code": code, "name": "未找到", "error": f"未找到港股 {code}"})
+                continue
+
+            r = fi.iloc[0]
+            stock_data = {
+                "code": code,
+                "pe_ratio": safe_float(r.get('市盈率')),
+                "pb_ratio": safe_float(r.get('市净率')),
+                "market_cap": safe_float(r.get('总市值(港元)')),
+                "data_source": "eastmoney_hk",
+            }
+            for metric_key in hk_metrics:
+                col_name = METRIC_MAP_HK.get(metric_key)
+                if col_name:
+                    stock_data[metric_key] = safe_float(r.get(col_name))
+            stocks.append(stock_data)
+        except Exception as e:
+            stocks.append({"code": code, "error": str(e)})
+
+    return stocks, hk_metrics, [METRIC_MAP_HK[m] for m in hk_metrics if m in METRIC_MAP_HK]
+
+
+def main():
+    parser = argparse.ArgumentParser(description='股票对比分析')
+    parser.add_argument('stock_codes', help='股票代码，逗号分隔，如 600519,000858 或 03888,00700')
+    parser.add_argument('--metrics', default=None,
+                        help='对比指标，逗号分隔（默认全部）')
+    args = parser.parse_args()
+
+    err = check_akshare()
+    if err:
+        output_error(err)
+        return
+
+    codes = [normalize_code(c.strip()) for c in args.stock_codes.split(',') if c.strip()]
+
+    if not codes:
+        output_error("请提供至少一个股票代码")
+        return
+
+    if len(codes) > 10:
+        output_error("最多对比10只股票")
+        return
+
+    # 自动判断市场类型
+    hk_codes = [c for c in codes if is_hk_code(c)]
+    a_codes = [c for c in codes if not is_hk_code(c)]
+
+    try:
+        import akshare as ak
+
+        # 纯港股对比
+        if hk_codes and not a_codes:
+            metrics = ALL_HK_METRICS
+            if args.metrics:
+                metrics = [m.strip() for m in args.metrics.split(',') if m.strip() in METRIC_MAP_HK]
+
+            stocks, used_metrics, metrics_cn = _fetch_hk_batch(hk_codes, metrics)
+
+            data = {
+                "market": "HK",
+                "stocks": stocks,
+                "metrics_compared": used_metrics,
+                "metrics_compared_cn": metrics_cn,
+            }
+            output_json(data)
+            return
+
+        # A股对比（含混合情况，只比A股部分）
+        if a_codes:
+            metrics = ALL_A_METRICS
+            if args.metrics:
+                metrics = [m.strip() for m in args.metrics.split(',') if m.strip() in METRIC_MAP_EM]
+
+            stocks = None
+            used_metrics = metrics
+            metrics_cn = []
+
+            try:
+                stocks, used_metrics, metrics_cn = _fetch_sina_batch(a_codes, metrics)
+            except Exception:
+                pass
+
+            if stocks is None:
+                try:
+                    stocks, used_metrics, metrics_cn = _fetch_em_batch(a_codes, metrics)
+                except Exception as e2:
+                    output_error(f"股票对比失败: {str(e2)}")
+                    return
+
+            # 如果有港股混合，追加港股数据
+            if hk_codes:
+                hk_metrics = ALL_HK_METRICS
+                hk_stocks, hk_used, hk_cn = _fetch_hk_batch(hk_codes, hk_metrics)
+                stocks.extend(hk_stocks)
+                used_metrics = list(set(used_metrics + hk_used))
+                metrics_cn = list(set(metrics_cn + hk_cn))
+
+            data = {
+                "stocks": stocks,
+                "metrics_compared": used_metrics,
+                "metrics_compared_cn": metrics_cn,
+            }
+            output_json(data)
+            return
+
+        output_error("未提供有效的股票代码")
+
+    except Exception as e:
+        output_error(f"股票对比失败: {str(e)}")
+
+
+if __name__ == '__main__':
+    main()

package/share/scripts/stock_financial.py

@@ -0,0 +1,211 @@
+"""
+股票财务数据查询
+
+用法: python stock_financial.py <stock_code> [--report <type>] [--count <n>]
+  stock_code: 6位A股代码(如 600519) 或 5位港股代码(如 03888)
+  --report:   报表类型 income/balance/cashflow，默认income
+  --count:    报告期数，默认4
+
+AKShare接口:
+  A股:   ak.stock_financial_report_sina(stock, symbol) — 新浪完整多期报表
+  港股补: ak.stock_hk_financial_indicator_em(symbol) — 东方财富关键财务指标
+  港股补: ak.stock_hk_dividend_payout_em(symbol) — 东方财富分红派息历史
+  港股补: ak.stock_hk_valuation_comparison_em(symbol) — 东方财富估值行业对比
+  港股补: ak.stock_hk_growth_comparison_em(symbol) — 东方财富成长性行业对比
+"""
+
+import sys
+import argparse
+
+sys.path.insert(0, '.')
+from _utils import normalize_code, output_json, output_error, check_akshare, safe_float, is_hk_code
+
+
+REPORT_MAP = {
+    "income": "利润表",
+    "balance": "资产负债表",
+    "cashflow": "现金流量表",
+}
+
+
+def _fetch_hk_financial(code):
+    """港股财务数据：指标+分红+估值对比+成长对比"""
+    import akshare as ak
+
+    result = {
+        "code": code,
+        "report_type": "hk_comprehensive",
+        "report_type_cn": "港股综合财务",
+        "data_source": "eastmoney_hk",
+    }
+
+    # 1. 关键财务指标
+    try:
+        fi = ak.stock_hk_financial_indicator_em(symbol=code)
+        if not fi.empty:
+            r = fi.iloc[0]
+            result["indicators"] = {
+                "eps": safe_float(r.get('基本每股收益(元)')),
+                "bps": safe_float(r.get('每股净资产(元)')),
+                "dividend_per_share_ttm": safe_float(r.get('每股股息TTM(港元)')),
+                "payout_ratio": safe_float(r.get('派息比率(%)')),
+                "dividend_yield_ttm": safe_float(r.get('股息率TTM(%)')),
+                "revenue": safe_float(r.get('营业总收入')),
+                "revenue_growth": safe_float(r.get('营业总收入滚动环比增长(%)')),
+                "net_profit": safe_float(r.get('净利润')),
+                "net_profit_growth": safe_float(r.get('净利润滚动环比增长(%)')),
+                "net_margin": safe_float(r.get('销售净利率(%)')),
+                "roe": safe_float(r.get('股东权益回报率(%)')),
+                "roa": safe_float(r.get('总资产回报率(%)')),
+                "pe": safe_float(r.get('市盈率')),
+                "pb": safe_float(r.get('市净率')),
+                "market_cap": safe_float(r.get('总市值(港元)')),
+                "shares_outstanding": safe_float(r.get('已发行股本(股)')),
+            }
+        else:
+            result["indicators"] = None
+            result["indicators_error"] = "东方财富港股指标接口返回空数据"
+    except Exception as e:
+        result["indicators"] = None
+        result["indicators_error"] = f"获取关键指标失败: {str(e)}"
+
+    # 2. 分红派息历史
+    try:
+        div_df = ak.stock_hk_dividend_payout_em(symbol=code)
+        if not div_df.empty:
+            dividends = []
+            for _, row in div_df.head(5).iterrows():
+                dividends.append({
+                    "announce_date": str(row.get('最新公告日期', '')),
+                    "fiscal_year": str(row.get('财政年度', '')),
+                    "plan": str(row.get('分红方案', '')),
+                    "type": str(row.get('分配类型', '')),
+                    "ex_date": str(row.get('除净日', '')),
+                    "pay_date": str(row.get('发放日', '')),
+                })
+            result["dividends"] = dividends
+            result["dividend_count"] = len(dividends)
+        else:
+            result["dividends"] = []
+            result["dividends_error"] = "分红派息接口返回空数据"
+    except Exception as e:
+        result["dividends"] = []
+        result["dividends_error"] = f"获取分红数据失败: {str(e)}"
+
+    # 3. 估值行业对比
+    try:
+        val_df = ak.stock_hk_valuation_comparison_em(symbol=code)
+        if not val_df.empty:
+            r = val_df.iloc[0]
+            result["valuation_comparison"] = {
+                "pe_ttm": safe_float(r.get('市盈率-TTM')),
+                "pe_ttm_rank": safe_float(r.get('市盈率-TTM排名')),
+                "pe_lyr": safe_float(r.get('市盈率-LYR')),
+                "pb_mrq": safe_float(r.get('市净率-MRQ')),
+                "pb_mrq_rank": safe_float(r.get('市净率-MRQ排名')),
+                "ps_ttm": safe_float(r.get('市销率-TTM')),
+                "ps_ttm_rank": safe_float(r.get('市销率-TTM排名')),
+            }
+        else:
+            result["valuation_comparison"] = None
+            result["valuation_comparison_error"] = "估值对比接口返回空数据"
+    except Exception as e:
+        result["valuation_comparison"] = None
+        result["valuation_comparison_error"] = f"获取估值对比失败: {str(e)}"
+
+    # 4. 成长性行业对比
+    try:
+        grow_df = ak.stock_hk_growth_comparison_em(symbol=code)
+        if not grow_df.empty:
+            r = grow_df.iloc[0]
+            result["growth_comparison"] = {
+                "eps_growth": safe_float(r.get('基本每股收益同比增长率')),
+                "eps_growth_rank": safe_float(r.get('基本每股收益同比增长率排名')),
+                "revenue_growth": safe_float(r.get('营业收入同比增长率')),
+                "revenue_growth_rank": safe_float(r.get('营业收入同比增长率排名')),
+                "op_profit_growth": safe_float(r.get('营业利润率同比增长率')),
+                "op_profit_growth_rank": safe_float(r.get('营业利润率同比增长率排名')),
+            }
+        else:
+            result["growth_comparison"] = None
+            result["growth_comparison_error"] = "成长性对比接口返回空数据"
+    except Exception as e:
+        result["growth_comparison"] = None
+        result["growth_comparison_error"] = f"获取成长性对比失败: {str(e)}"
+
+    return result
+
+
+def main():
+    parser = argparse.ArgumentParser(description='股票财务数据查询')
+    parser.add_argument('stock_code', help='股票代码')
+    parser.add_argument('--report', default='income', choices=REPORT_MAP.keys(),
+                        help='报表类型: income/balance/cashflow')
+    parser.add_argument('--count', type=int, default=4, help='报告期数(默认4)')
+    args = parser.parse_args()
+
+    err = check_akshare()
+    if err:
+        output_error(err)
+        return
+
+    code = normalize_code(args.stock_code)
+    report_cn = REPORT_MAP[args.report]
+
+    try:
+        import akshare as ak
+
+        # 港股：综合财务数据
+        if is_hk_code(code):
+            data = _fetch_hk_financial(code)
+            if data.get("indicators") is None and not data.get("dividends"):
+                output_error(f"未找到港股 {code} 的财务数据")
+                return
+            output_json(data)
+            return
+
+        # A股：新浪财务报表
+        df = None
+        try:
+            df = ak.stock_financial_report_sina(stock=code, symbol=report_cn)
+        except (TypeError, KeyError, AttributeError):
+            pass
+
+        if df is None or df.empty:
+            output_error(f"未找到 {code} 的{report_cn}数据，该接口仅支持A股")
+            return
+
+        # 取前N期
+        df = df.head(args.count)
+
+        # 转为字典列表，处理NaN
+        periods = []
+        for _, row in df.iterrows():
+            period_data = {}
+            for col in df.columns:
+                val = row[col]
+                if str(val) == 'nan':
+                    period_data[col] = None
+                else:
+                    period_data[col] = safe_float(val, str(val))
+            periods.append(period_data)
+
+        # 获取股票名称
+        name = periods[0].get('股票代码', code) if periods else code
+
+        data = {
+            "code": code,
+            "name": name,
+            "report_type": args.report,
+            "report_type_cn": report_cn,
+            "periods": periods,
+        }
+
+        output_json(data)
+
+    except Exception as e:
+        output_error(f"获取财务数据失败: {str(e)}")
+
+
+if __name__ == '__main__':
+    main()

package/share/scripts/stock_news.py

@@ -0,0 +1,181 @@
+"""
+股票新闻/公司资讯查询
+
+用法: python stock_news.py <stock_code> [--count <n>]
+  stock_code: 6位A股代码(如 600519) 或 5位港股代码(如 03888)
+  --count:    新闻条数，默认10
+
+AKShare接口:
+  A股:   ak.stock_news_em(symbol) — 东方财富个股新闻
+  港股:  ak.stock_hk_company_profile_em(symbol) — 东方财富公司资料(港股无新闻接口)
+  港股:  ak.stock_hk_security_profile_em(symbol) — 东方财富证券资料
+"""
+
+import sys
+import argparse
+
+sys.path.insert(0, '.')
+from _utils import normalize_code, output_json, output_error, check_akshare, safe_float, is_hk_code
+
+
+def _fetch_hk_profile(code):
+    """港股公司资料（替代新闻）"""
+    import akshare as ak
+
+    data = {"code": code, "type": "hk_company_profile"}
+
+    # 公司资料
+    try:
+        profile = ak.stock_hk_company_profile_em(symbol=code)
+        if not profile.empty:
+            p = profile.iloc[0]
+            data["company"] = {
+                "name": str(p.get('公司名称', '')),
+                "name_en": str(p.get('英文名称', '')),
+                "industry": str(p.get('所属行业', '')),
+                "chairman": str(p.get('董事长', '')),
+                "employees": safe_float(p.get('员工人数')),
+                "website": str(p.get('公司网址', '')),
+                "founded": str(p.get('公司成立日期', '')),
+                "description": str(p.get('公司介绍', ''))[:500],
+            }
+    except Exception:
+        data["company"] = None
+
+    # 证券资料
+    try:
+        sec = ak.stock_hk_security_profile_em(symbol=code)
+        if not sec.empty:
+            s = sec.iloc[0]
+            data["security"] = {
+                "name": str(s.get('证券简称', '')),
+                "list_date": str(s.get('上市日期', '')),
+                "type": str(s.get('证券类型', '')),
+                "board": str(s.get('板块', '')),
+                "ipo_price": safe_float(s.get('发行价')),
+                "lot_size": safe_float(s.get('每手股数')),
+                "exchange": str(s.get('交易所', '')),
+                "is_sh_hkconnect": str(s.get('是否沪港通标的', '')),
+                "is_sz_hkconnect": str(s.get('是否深港通标的', '')),
+            }
+    except Exception:
+        data["security"] = None
+
+    return data
+
+
+def main():
+    parser = argparse.ArgumentParser(description='股票新闻/公司资讯查询')
+    parser.add_argument('stock_code', help='股票代码')
+    parser.add_argument('--count', type=int, default=10, help='新闻条数(默认10)')
+    args = parser.parse_args()
+
+    err = check_akshare()
+    if err:
+        output_error(err)
+        return
+
+    code = normalize_code(args.stock_code)
+
+    try:
+        import akshare as ak
+
+        # 港股：先尝试东方财富新闻接口，失败则返回公司资料
+        if is_hk_code(code):
+            # 1. 尝试东方财富新闻（已支持港股代码）
+            try:
+                df = ak.stock_news_em(symbol=code)
+                if df is not None and not df.empty:
+                    df = df.head(args.count * 3)
+                    seen = set()
+                    news_items = []
+                    for _, row in df.iterrows():
+                        title = str(row.get('新闻标题', ''))
+                        url = str(row.get('新闻链接', ''))
+                        dedup_key = (title.strip(), url.strip())
+                        if dedup_key in seen:
+                            continue
+                        seen.add(dedup_key)
+                        item = {
+                            "title": title,
+                            "content": str(row.get('新闻内容', ''))[:200],
+                            "source": str(row.get('文章来源', '')),
+                            "publish_time": str(row.get('发布时间', '')),
+                            "url": url,
+                        }
+                        news_items.append(item)
+                        if len(news_items) >= args.count:
+                            break
+
+                    data = {
+                        "code": code,
+                        "type": "hk_news",
+                        "news_count": len(news_items),
+                        "news": news_items,
+                    }
+                    # 同时获取公司资料作为补充
+                    profile_data = _fetch_hk_profile(code)
+                    if profile_data.get("company"):
+                        data["company"] = profile_data["company"]
+                    if profile_data.get("security"):
+                        data["security"] = profile_data["security"]
+                    output_json(data)
+                    return
+            except Exception:
+                pass
+
+            # 2. 新闻接口失败，回退到公司资料
+            data = _fetch_hk_profile(code)
+            data["type"] = "hk_company_profile"
+            data["news_note"] = "东方财富新闻接口暂不可用，仅返回公司资料"
+            if data.get("company") is None and data.get("security") is None:
+                output_error(f"未找到港股 {code} 的公司资料")
+                return
+            output_json(data)
+            return
+
+        # A股：东方财富新闻
+        df = ak.stock_news_em(symbol=code)
+
+        if df.empty:
+            output_error(f"未找到 {code} 的新闻数据")
+            return
+
+        df = df.head(args.count * 3)  # 取更多以应对去重后不足
+
+        seen = set()
+        news_items = []
+        for _, row in df.iterrows():
+            title = str(row.get('新闻标题', ''))
+            url = str(row.get('新闻链接', ''))
+            # 去重：标题+链接组合唯一
+            dedup_key = (title.strip(), url.strip())
+            if dedup_key in seen:
+                continue
+            seen.add(dedup_key)
+
+            item = {
+                "title": title,
+                "content": str(row.get('新闻内容', ''))[:200],
+                "source": str(row.get('文章来源', '')),
+                "publish_time": str(row.get('发布时间', '')),
+                "url": url,
+            }
+            news_items.append(item)
+            if len(news_items) >= args.count:
+                break
+
+        data = {
+            "code": code,
+            "news_count": len(news_items),
+            "news": news_items,
+        }
+
+        output_json(data)
+
+    except Exception as e:
+        output_error(f"获取新闻失败: {str(e)}")
+
+
+if __name__ == '__main__':
+    main()