supply_gap_strategy.py 588 B

123456789101112131415161718
  1. from typing import Any
  2. from app.strategies.impl._supply_gap_base import SupplyGapStrategyBase
  3. class SupplyGapStrategy(SupplyGapStrategyBase):
  4. """当下供需gap:demand_name = merge_leve2 + 空格 + name。"""
  5. strategy_id = "supply_gap"
  6. name = "当下供需gap"
  7. version = "1.0.0"
  8. def build_demand_name(self, row: dict[str, Any]) -> str:
  9. merge_leve2 = str(row.get("merge_leve2") or "").strip()
  10. name = str(row.get("name") or "").strip()
  11. if merge_leve2 and name:
  12. return f"{merge_leve2} {name}"
  13. return merge_leve2 or name