118 lines
5.2 KiB
Python
118 lines
5.2 KiB
Python
import datetime
|
||
from screener.data_fetcher import validate_date_range, fetch_financial_data, get_stocks_in_time_range
|
||
from screener.t_sunnyband import run_sunny_scanner
|
||
from screener.t_atr_ema import run_atr_ema_scanner
|
||
from screener.c_canslim import check_quarterly_earnings, check_return_on_equity, check_sales_growth
|
||
from screener.a_canslim import check_annual_eps_growth
|
||
from screener.l_canslim import check_industry_leadership
|
||
from screener.i_canslim import check_institutional_sponsorship
|
||
from screener.csv_appender import append_scores_to_csv
|
||
from screener.screeners import SCREENERS
|
||
from screener.user_input import get_user_screener_selection, get_interval_choice
|
||
from indicators.three_atr_ema import ThreeATREMAIndicator
|
||
|
||
def get_float_input(prompt: str) -> float:
|
||
while True:
|
||
try:
|
||
return float(input(prompt))
|
||
except ValueError:
|
||
print("Please enter a valid number")
|
||
|
||
def get_scanner_parameters():
|
||
"""Get user input for scanner parameters"""
|
||
min_price = get_float_input("Enter minimum stock price ($): ")
|
||
max_price = get_float_input("Enter maximum stock price ($): ")
|
||
min_volume = int(input("Enter minimum volume: "))
|
||
portfolio_size = get_float_input("Enter portfolio size ($) or 0 to skip position sizing: ")
|
||
return min_price, max_price, min_volume, portfolio_size
|
||
|
||
def main():
|
||
print("\nStock Analysis System")
|
||
print("1. Run CANSLIM Screener")
|
||
print("2. Run Technical Scanners (SunnyBands/ATR-EMA)")
|
||
print("3. Launch Trading System")
|
||
print("4. Exit")
|
||
|
||
choice = input("\nSelect an option (1-4): ")
|
||
|
||
if choice == "1":
|
||
# 1️⃣ Ask user for start and end date
|
||
user_start_date = input("Enter start date (YYYY-MM-DD): ")
|
||
user_end_date = input("Enter end date (YYYY-MM-DD): ")
|
||
|
||
# 2️⃣ Validate and adjust date range if needed
|
||
start_date, end_date = validate_date_range(user_start_date, user_end_date, required_quarters=4)
|
||
selected_screeners = get_user_screener_selection()
|
||
symbol_list = get_stocks_in_time_range(start_date, end_date)
|
||
|
||
if not symbol_list:
|
||
print("No stocks found within the given date range.")
|
||
return
|
||
|
||
print(f"Processing {len(symbol_list)} stocks within the given date range...\n")
|
||
|
||
for symbol in symbol_list:
|
||
data = fetch_financial_data(symbol, start_date, end_date)
|
||
|
||
if not data:
|
||
print(f"⚠️ Warning: No data returned for {symbol}. Assigning default score.\n")
|
||
scores = {screener: 0.25 for category in selected_screeners for screener in selected_screeners[category]}
|
||
else:
|
||
scores = {}
|
||
|
||
for category, screeners in selected_screeners.items():
|
||
for screener, threshold in screeners.items():
|
||
if screener == "EPS_Score":
|
||
scores[screener] = check_quarterly_earnings(data.get("quarterly_eps", []))
|
||
elif screener == "Annual_EPS_Score":
|
||
scores[screener] = check_annual_eps_growth(data.get("annual_eps", []))
|
||
elif screener == "Sales_Score":
|
||
scores[screener] = check_sales_growth(data.get("sales_growth", []))
|
||
elif screener == "ROE_Score":
|
||
scores[screener] = check_return_on_equity(data.get("roe", []))
|
||
elif screener == "L_Score":
|
||
scores[screener] = check_industry_leadership(symbol)
|
||
print(f"🟢 {symbol} - L_Score: {scores[screener]}")
|
||
elif screener == "I_Score":
|
||
scores[screener] = check_institutional_sponsorship(symbol)
|
||
print(f"🏢 {symbol} - I_Score: {scores[screener]}")
|
||
|
||
if isinstance(threshold, (int, float)):
|
||
scores[screener] = scores[screener] >= threshold
|
||
|
||
scores["Total_Score"] = sum(scores.values())
|
||
append_scores_to_csv(symbol, scores)
|
||
|
||
print("✅ Scores saved in data/metrics/stock_scores.csv\n")
|
||
|
||
elif choice == "2":
|
||
print("\nTechnical Scanner Options:")
|
||
print("1. SunnyBands Scanner")
|
||
print("2. ATR-EMA Scanner")
|
||
|
||
scanner_choice = input("\nEnter your choice (1-2): ")
|
||
|
||
if scanner_choice == "1":
|
||
from screener.t_sunnyband import run_sunny_scanner
|
||
min_price, max_price, min_volume, portfolio_size = get_scanner_parameters()
|
||
run_sunny_scanner(min_price, max_price, min_volume, portfolio_size)
|
||
elif scanner_choice == "2":
|
||
from screener.t_atr_ema import run_atr_ema_scanner
|
||
min_price, max_price, min_volume, portfolio_size = get_scanner_parameters()
|
||
run_atr_ema_scanner(min_price, max_price, min_volume, portfolio_size)
|
||
else:
|
||
print("Invalid choice. Please try again.")
|
||
|
||
elif choice == "3":
|
||
from trading.main import main as trading_main
|
||
trading_main()
|
||
|
||
elif choice == "4":
|
||
print("Exiting...")
|
||
return
|
||
else:
|
||
print("Invalid choice. Please try again.")
|
||
|
||
if __name__ == "__main__":
|
||
main()
|