refactor: Update scanner initialization to use passed interval parameter

This commit is contained in:
Bobby (aider) 2025-02-12 19:37:48 -08:00
parent 7d842f661b
commit 3b2cd794ec
2 changed files with 10 additions and 4 deletions

View File

@ -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:

View File

@ -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)