Bobby/stock_system
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: Correctly handle query results in trading plan retrieval functions

Browse Source
This commit is contained in:
Bobby (aider) 2025-02-11 18:25:02 -08:00
parent 1d423d3291
commit 5acfe4bfc4
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/trading/trading_plan.py
View File

@ -209,10 +209,12 @@ def get_trading_plan(plan_id: int) -> Optional[TradingPlan]:
SELECT * FROM trading_plans WHERE id = %(id)s SELECT * FROM trading_plans WHERE id = %(id)s
""", {'id': plan_id}) """, {'id': plan_id})
if not result: rows = result.result_rows
if not rows:
return None return None
plan = result[0] plan = rows[0]
return TradingPlan( return TradingPlan(
id=plan[0], id=plan[0],
plan_name=plan[1], plan_name=plan[1],
@ -267,7 +269,7 @@ def get_all_trading_plans(status: Optional[PlanStatus] = None) -> List[TradingPl
query += " ORDER BY updated_at DESC" query += " ORDER BY updated_at DESC"
results = client.query(query, params) results = client.query(query, params)
rows = results.result_rows # Ensure this is the correct method to access rows rows = results.result_rows
return [TradingPlan( return [TradingPlan(
id=row[0], id=row[0],
plan_name=row[1], plan_name=row[1],