From 26238a6afb51b4d7bea0efb5bdb16fb2ff7e01c6 Mon Sep 17 00:00:00 2001 From: "Bobby (aider)" Date: Tue, 11 Feb 2025 18:21:33 -0800 Subject: [PATCH] refactor: Update query result handling and deprecated Streamlit function --- src/streamlit_app.py | 4 ++-- src/trading/trading_plan.py | 3 ++- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/src/streamlit_app.py b/src/streamlit_app.py index 759cb5f..e135d00 100644 --- a/src/streamlit_app.py +++ b/src/streamlit_app.py @@ -244,7 +244,7 @@ def trading_journal_page(): add_trade(trade) st.success("Trade added successfully!") - st.experimental_set_query_params(rerun=True) + st.query_params(rerun=True) except Exception as e: st.error(f"Error adding trade: {str(e)}") @@ -310,7 +310,7 @@ def trading_journal_page(): update_trade(trade_id, updates) st.success("Trade updated successfully!") - st.experimental_set_query_params(rerun=True) + st.query_params(rerun=True) except Exception as e: st.error(f"Error updating trade: {str(e)}") else: diff --git a/src/trading/trading_plan.py b/src/trading/trading_plan.py index 33a5b35..a2f4a8a 100644 --- a/src/trading/trading_plan.py +++ b/src/trading/trading_plan.py @@ -267,7 +267,8 @@ def get_all_trading_plans(status: Optional[PlanStatus] = None) -> List[TradingPl query += " ORDER BY updated_at DESC" results = client.query(query, params) - return [get_trading_plan(result[0]) for result in results] + rows = results.result_rows + return [get_trading_plan(row[0]) for row in rows] except Exception as e: print(f"Error retrieving trading plans: {e}")