From b9704646f91e5a4a37a3025300cd891a87ea9016 Mon Sep 17 00:00:00 2001 From: "Bobby (aider)" Date: Wed, 12 Feb 2025 19:42:49 -0800 Subject: [PATCH] refactor: Update run_sunny_scanner with consistent parameter handling --- src/screener/t_sunnyband.py | 16 ++++++++++++---- 1 file changed, 12 insertions(+), 4 deletions(-) diff --git a/src/screener/t_sunnyband.py b/src/screener/t_sunnyband.py index 29fcab1..7a393df 100644 --- a/src/screener/t_sunnyband.py +++ b/src/screener/t_sunnyband.py @@ -1,5 +1,5 @@ import pandas as pd -from datetime import datetime, timedelta +from datetime import datetime from db.db_connection import create_client from utils.data_utils import ( get_stock_data, validate_signal_date, print_signal, @@ -202,11 +202,19 @@ def view_stock_details(ticker: str, interval: str, start_date: datetime, end_dat except Exception as e: print(f"Error analyzing {ticker}: {str(e)}") -def run_sunny_scanner(min_price: float, max_price: float, min_volume: int, portfolio_size: float = None, interval: str = "1d") -> None: +def run_sunny_scanner(min_price: float, max_price: float, min_volume: int, + portfolio_size: float = None, interval: str = "1d", + start_date: datetime = None, end_date: datetime = None) -> None: try: - # Initialize scanner components + # Initialize scanner components with all parameters interval, start_date, end_date, qualified_stocks, calculator = initialize_scanner( - min_price, max_price, min_volume, portfolio_size + min_price=min_price, + max_price=max_price, + min_volume=min_volume, + portfolio_size=portfolio_size, + interval=interval, + start_date=start_date, + end_date=end_date ) if not qualified_stocks: