From 3b2cd794ec32bdb2c65db683dbef54edb986c051 Mon Sep 17 00:00:00 2001 From: "Bobby (aider)" Date: Wed, 12 Feb 2025 19:37:48 -0800 Subject: [PATCH] refactor: Update scanner initialization to use passed interval parameter --- src/screener/t_atr_ema.py | 4 ++-- src/utils/scanner_utils.py | 10 ++++++++-- 2 files changed, 10 insertions(+), 4 deletions(-) diff --git a/src/screener/t_atr_ema.py b/src/screener/t_atr_ema.py index 97fee56..737208d 100644 --- a/src/screener/t_atr_ema.py +++ b/src/screener/t_atr_ema.py @@ -56,11 +56,11 @@ def check_entry_signal(df: pd.DataFrame) -> list: return signals -def run_atr_ema_scanner(min_price: float, max_price: float, min_volume: int, portfolio_size: float = None) -> None: +def run_atr_ema_scanner(min_price: float, max_price: float, min_volume: int, portfolio_size: float = None, interval: str = "1d") -> None: try: # Initialize scanner components interval, start_date, end_date, qualified_stocks, calculator = initialize_scanner( - min_price, max_price, min_volume, portfolio_size + min_price, max_price, min_volume, portfolio_size, interval ) if not qualified_stocks: diff --git a/src/utils/scanner_utils.py b/src/utils/scanner_utils.py index 475e833..0687e28 100644 --- a/src/utils/scanner_utils.py +++ b/src/utils/scanner_utils.py @@ -3,13 +3,19 @@ from utils.data_utils import get_user_input, get_stock_data, get_qualified_stock from screener.user_input import get_interval_choice, get_date_range from trading.position_calculator import PositionCalculator -def initialize_scanner(min_price: float, max_price: float, min_volume: int, portfolio_size: float = None) -> tuple: +def initialize_scanner(min_price: float, max_price: float, min_volume: int, portfolio_size: float = None, interval: str = "1d") -> tuple: """ Initialize common scanner components + + Args: + min_price (float): Minimum stock price + max_price (float): Maximum stock price + min_volume (int): Minimum volume threshold + portfolio_size (float, optional): Portfolio size for position calculations + interval (str, optional): Time interval for data (default: "1d") """ print(f"\nScanning for stocks ${min_price:.2f}-${max_price:.2f} with min volume {min_volume:,}") - interval = get_interval_choice() start_date, end_date = get_date_range() qualified_stocks = get_qualified_stocks(start_date, end_date, min_price, max_price, min_volume)