feat: 初始提交 - 科创企业特有风险的识别与管理 (数智风控系统)

XH-202626_原型系统源码与部署手册/01_原型系统源码/collectors/financial_collector.py

@@ -0,0 +1,105 @@
+# -*- coding: utf-8 -*-
+"""
+财务数据采集模块
+双轨策略：优先尝试 AKShare 在线采集，失败则回退到预置数据
+"""
+import json
+import logging
+from pathlib import Path
+from typing import Optional
+
+logger = logging.getLogger(__name__)
+
+DATA_DIR = Path(__file__).parent.parent / "data"
+
+
+def _load_preset_data() -> list:
+    """加载预置的科创板企业数据"""
+    filepath = DATA_DIR / "sample_companies.json"
+    with open(filepath, "r", encoding="utf-8") as f:
+        return json.load(f)
+
+
+def collect_financial_data(stock_code: str) -> Optional[dict]:
+    """
+    采集指定股票代码的财务数据
+    双轨策略：在线采集 → 离线预置
+    """
+    # 尝试在线采集
+    try:
+        return _collect_online(stock_code)
+    except Exception as e:
+        logger.warning(f"在线采集 {stock_code} 失败: {e}，回退到预置数据")
+
+    # 回退到预置数据
+    return _collect_from_preset(stock_code)
+
+
+def _collect_online(stock_code: str) -> Optional[dict]:
+    """通过 AKShare 在线采集财务数据"""
+    try:
+        import akshare as ak
+
+        # 科创板企业利润表
+        profit_df = ak.stock_profit_sheet_by_report_em(symbol=stock_code)
+        # 科创板企业资产负债表
+        balance_df = ak.stock_balance_sheet_by_report_em(symbol=stock_code)
+
+        if profit_df is not None and not profit_df.empty:
+            latest = profit_df.iloc[0]
+            return {
+                "stock_code": stock_code,
+                "revenue": float(latest.get("营业收入", 0)),
+                "net_profit": float(latest.get("净利润", 0)),
+                "rd_expense": float(latest.get("研发费用", 0)),
+                "source": "akshare_online",
+            }
+    except ImportError:
+        logger.warning("AKShare 未安装，跳过在线采集")
+    except Exception as e:
+        logger.warning(f"AKShare 采集异常: {e}")
+
+    raise RuntimeError("在线采集失败")
+
+
+def _collect_from_preset(stock_code: str) -> Optional[dict]:
+    """从预置数据中查找企业"""
+    companies = _load_preset_data()
+    for company in companies:
+        if company["stock_code"] == stock_code:
+            return {
+                "stock_code": stock_code,
+                "company_name": company["company_name"],
+                "industry": company["industry"],
+                "sector": company["sector"],
+                "financials": company["financials"],
+                "core_tech_personnel": company["core_tech_personnel"],
+                "tech_route": company["tech_route"],
+                "compliance": company["compliance"],
+                "supply_chain": company["supply_chain"],
+                "source": "preset_data",
+            }
+    return None
+
+
+def get_all_companies() -> list:
+    """获取所有预置企业列表"""
+    return _load_preset_data()
+
+
+def get_company_by_code(stock_code: str) -> Optional[dict]:
+    """通过股票代码查找企业完整数据"""
+    companies = _load_preset_data()
+    for company in companies:
+        if company["stock_code"] == stock_code:
+            return company
+    return None
+
+
+def get_company_by_name(name: str) -> Optional[dict]:
+    """通过企业名称查找（支持简称）"""
+    companies = _load_preset_data()
+    for company in companies:
+        if name in company["company_name"] or name in company["short_name"]:
+            return company
+    return None