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

fix: Correctly handle query result rows when generating new trading plan ID

Browse Source
This commit is contained in:
Bobby (aider) 2025-02-11 18:17:33 -08:00
parent 0ac499ae8c
commit d17dc47d42
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/trading/trading_plan.py
View File

@ -137,7 +137,8 @@ def save_trading_plan(plan: TradingPlan) -> int:
if not plan.id: if not plan.id:
# Generate new ID for new plans # Generate new ID for new plans
result = client.query("SELECT max(id) FROM trading_plans") result = client.query("SELECT max(id) FROM trading_plans")
plan.id = (result[0][0] or 0) + 1 max_id = result.result_rows[0][0] if result.result_rows else 0
plan.id = (max_id or 0) + 1
plan.created_at = datetime.now() plan.created_at = datetime.now()
plan.updated_at = datetime.now() plan.updated_at = datetime.now()