feat: refactor portfolio strategy references

This commit is contained in:
2026-04-03 08:16:07 +00:00
parent ebd4559939
commit 908068d9aa
14 changed files with 294 additions and 139 deletions
+11 -12
View File
@@ -39,8 +39,8 @@ class FakeRepo:
*,
broker: str = "kis",
status: str = "enabled",
strategy_name: str = "",
strategy_params: dict | None = None,
active_strategy_id: str | None = None,
avabile_strategies: list[str] | None = None,
initial_cash: float = 0.0,
) -> Portfolio:
if self._portfolio is None:
@@ -49,9 +49,9 @@ class FakeRepo:
broker=broker,
status=status,
symbol=symbol,
strategy_name=strategy_name,
strategy_params=strategy_params or {},
cash_balance=initial_cash,
active_strategy_id=active_strategy_id,
avabile_strategies=avabile_strategies,
)
return self._portfolio
@@ -62,24 +62,24 @@ class FakeRepo:
broker: str,
status: str,
symbol: str,
strategy_name: str,
strategy_params: dict,
cash_balance: float,
asset_quantity: float,
asset_price: float,
profit_buffer: float,
active_strategy_id: str | None,
avabile_strategies: list[str] | None,
) -> Portfolio:
self._portfolio = Portfolio(
portfolio_id=portfolio_id,
broker=broker,
status=status,
symbol=symbol,
strategy_name=strategy_name,
strategy_params=strategy_params,
cash_balance=cash_balance,
asset_quantity=asset_quantity,
price=asset_price,
profit_buffer=profit_buffer,
active_strategy_id=active_strategy_id,
avabile_strategies=avabile_strategies,
)
return self._portfolio
@@ -149,7 +149,7 @@ class FakeRepo:
portfolio.asset_quantity = 0.0
portfolio.price = 0.0
portfolio.profit_buffer = 0.0
portfolio.current_strategy_set_id = None
portfolio.active_strategy_id = None
self.entries = []
return portfolio
@@ -194,13 +194,12 @@ def _portfolio() -> Portfolio:
broker="kis",
status="enabled",
symbol="005930",
strategy_name="target_value",
strategy_params={"target_asset_ratio": 0.7},
cash_balance=1_000_000.0,
asset_quantity=1.0,
price=70_000.0,
profit_buffer=50_000.0,
current_strategy_set_id="set-1",
active_strategy_id="set-1",
avabile_strategies=["set-1"],
)