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

refactor: Add error handling to get_all_trading_plans function

Browse Source
This commit is contained in:
Bobby (aider) 2025-02-11 18:02:38 -08:00
parent 2a790a164a
commit 41de719250
1 changed files with 14 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/trading/trading_plan.py
View File

@ -255,6 +255,7 @@ def get_trading_plan(plan_id: int) -> Optional[TradingPlan]:
def get_all_trading_plans(status: Optional[PlanStatus] = None) -> List[TradingPlan]: def get_all_trading_plans(status: Optional[PlanStatus] = None) -> List[TradingPlan]:
"""Get all trading plans, optionally filtered by status""" """Get all trading plans, optionally filtered by status"""
with create_client() as client: with create_client() as client:
try:
query = "SELECT * FROM trading_plans" query = "SELECT * FROM trading_plans"
params = {} params = {}
@ -267,6 +268,10 @@ def get_all_trading_plans(status: Optional[PlanStatus] = None) -> List[TradingPl
results = client.execute(query, params) results = client.execute(query, params)
return [get_trading_plan(result[0]) for result in results] return [get_trading_plan(result[0]) for result in results]
except Exception as e:
print(f"Error retrieving trading plans: {e}")
return []
def update_trading_plan(plan: TradingPlan) -> bool: def update_trading_plan(plan: TradingPlan) -> bool:
"""Update an existing trading plan""" """Update an existing trading plan"""
if not plan.id: if not plan.id: