Bobby/stock_system
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactor: Update scanner initialization to use passed interval parameter

Browse Source
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/screener/t_atr_ema.py
View File

@ -56,11 +56,11 @@ def check_entry_signal(df: pd.DataFrame) -> list:
return signals 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: try:
# Initialize scanner components # Initialize scanner components
interval, start_date, end_date, qualified_stocks, calculator = initialize_scanner( 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: if not qualified_stocks:

10
src/utils/scanner_utils.py
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 screener.user_input import get_interval_choice, get_date_range
from trading.position_calculator import PositionCalculator 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 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:,}") 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() start_date, end_date = get_date_range()
qualified_stocks = get_qualified_stocks(start_date, end_date, min_price, max_price, min_volume) qualified_stocks = get_qualified_stocks(start_date, end_date, min_price, max_price, min_volume)