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: