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

fix: Update run_candlestick_scanner signature to accept selected_patterns

Browse Source
This commit is contained in:
Bobby (aider) 2025-02-19 22:13:37 -08:00
parent e3ac3bebc1
commit fa30002ef9
1 changed files with 13 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
src/screener/t_candlestick.py
View File

@ -83,9 +83,20 @@ def check_entry_signal(df: pd.DataFrame, selected_patterns: list = None) -> list
def run_candlestick_scanner(min_price: float, max_price: float, min_volume: int, def run_candlestick_scanner(min_price: float, max_price: float, min_volume: int,
portfolio_size: float = None, interval: str = "1d", portfolio_size: float = None, interval: str = "1d",
start_date: datetime = None, end_date: datetime = None) -> None: start_date: datetime = None, end_date: datetime = None,
selected_patterns: list = None) -> None:
""" """
Run candlestick pattern scanner to find bullish patterns Run candlestick pattern scanner to find bullish patterns
Args:
min_price (float): Minimum stock price
max_price (float): Maximum stock price
min_volume (int): Minimum volume
portfolio_size (float, optional): Portfolio size for position sizing
interval (str, optional): Time interval for data. Defaults to "1d"
start_date (datetime, optional): Start date for scanning
end_date (datetime, optional): End date for scanning
selected_patterns (list, optional): List of patterns to scan for
""" """
try: try:
# Initialize scanner components # Initialize scanner components
@ -111,7 +122,7 @@ def run_candlestick_scanner(min_price: float, max_price: float, min_volume: int,
if df.empty or len(df) < 5: # Need at least 5 bars if df.empty or len(df) < 5: # Need at least 5 bars
continue continue
signals = check_entry_signal(df) signals = check_entry_signal(df, selected_patterns) # Pass selected_patterns here
for signal, signal_date, signal_data in signals: for signal, signal_date, signal_data in signals:
signal_data['date'] = signal_date signal_data['date'] = signal_date
entry_data = process_signal_data( entry_data = process_signal_data(