From 516533800f1139c9bb4fbb08865522899280a3b5 Mon Sep 17 00:00:00 2001 From: "Bobby (aider)" Date: Sat, 8 Feb 2025 13:20:17 -0800 Subject: [PATCH] refactor: Use consistent date range input in sunny band scanner --- src/screener/t_sunnyband.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/src/screener/t_sunnyband.py b/src/screener/t_sunnyband.py index fee14a1..21cfb9f 100644 --- a/src/screener/t_sunnyband.py +++ b/src/screener/t_sunnyband.py @@ -5,7 +5,7 @@ import pandas as pd from db.db_connection import create_client from indicators.sunny_bands import SunnyBands from trading.position_calculator import PositionCalculator -from screener.user_input import get_interval_choice +from screener.user_input import get_interval_choice, get_date_range from utils.data_utils import get_stock_data @@ -166,8 +166,9 @@ def run_sunny_scanner(min_price: float, max_price: float, min_volume: int, portf print(f"\nScanning for stocks ${min_price:.2f}-${max_price:.2f} with min volume {min_volume:,}") interval = get_interval_choice() - end_date = datetime.now() - start_date = end_date - timedelta(days=1) # Get last trading day + + # Get date range from user input + start_date, end_date = get_date_range() # First get qualified stocks from database client = create_client()