Create bbb.py

bbb.py

@@ -0,0 +1,428 @@
+# main.py - Dynamic Stock Input + Test Alert Dashboard
+import yfinance as yf
+import pandas as pd
+import ta
+import requests
+import time
+import re
+from datetime import datetime
+from flask import Flask, request, render_template_string, flash, get_flashed_messages
+from threading import Thread
+import logging
+
+# === CONFIGURE LOGGING ===
+logging.basicConfig(level=logging.INFO,
+                    format='%(asctime)s [%(levelname)s] %(message)s',
+                    datefmt='%Y-%m-%d %H:%M:%S')
+
+# === CONFIGURATION ===
+DISCORD_WEBHOOK_URL = "https://discord.com/api/webhooks/1402083180785176637/hm5Efv4z3-OwCJpTJWQB78bvUixMIItPKiPdsUCjmmakr4MXOk_thRk5aBdn3uswIGVV"  # ← REPLACE
+SECRET_KEY = "bigly#Redis#NodJSforWin@GOLangSucks"  # ← CHANGE THIS! Use a strong key
+ATR_PERIOD = 7
+RSI_PERIOD = 7
+SUPER_TREND_FACTOR = 3
+INTERVAL = "5m"
+
+# 📝 Runtime Data
+monitored_stocks = ["SPY", "QQQ"]  # Default list
+alert_history = []
+
+# 🔐 Thread Locks
+import threading
+
+stocks_lock = threading.Lock()
+history_lock = threading.Lock()
+
+
+# === Validate Stock Symbol ===
+def is_valid_symbol(symbol):
+    """Allow only 1–5 uppercase letters (e.g., AAPL, MSFT)"""
+    return bool(re.fullmatch(r"^[A-Z]{1,5}$", symbol.strip()))
+
+
+# === Is Market Open? (US: 9:30 AM - 4:00 PM EST) ===
+def is_market_open():
+    from pytz import timezone
+    from datetime import time as dt_time
+    est = timezone('US/Eastern')
+    now = datetime.now(est)
+
+    if now.weekday() >= 5:
+        return False
+
+    market_open_time = dt_time(9, 30)
+    market_close_time = dt_time(16, 0)
+    return market_open_time <= now.time() <= market_close_time
+
+
+# === SuperTrend Calculation ===
+def calculate_supertrend(df, factor=SUPER_TREND_FACTOR, period=ATR_PERIOD):
+    data = df.copy()
+    hl2 = (data['high'] + data['low']) / 2
+    atr = ta.volatility.average_true_range(data['high'], data['low'], data['close'], window=period)
+    data['upper_band'] = hl2 + (factor * atr)
+    data['lower_band'] = hl2 - (factor * atr)
+    data['supertrend_trend'] = 1
+    for i in range(1, len(data)):
+        prev_trend = data['supertrend_trend'].iloc[i - 1]
+        curr_close = data['close'].iloc[i]
+        if prev_trend == 1:
+            if curr_close < data['lower_band'].iloc[i - 1]:
+                data.loc[data.index[i], 'supertrend_trend'] = -1
+            else:
+                data.loc[data.index[i], 'lower_band'] = max(data['lower_band'].iloc[i], data['lower_band'].iloc[i - 1])
+        else:
+            if curr_close > data['upper_band'].iloc[i - 1]:
+                data.loc[data.index[i], 'supertrend_trend'] = 1
+            else:
+                data.loc[data.index[i], 'upper_band'] = min(data['upper_band'].iloc[i], data['upper_band'].iloc[i - 1])
+
+    data.ffill(inplace=True)
+    return data
+
+
+# === Send to Discord ===
+def send_discord_alert(symbol, signal, price, rsi):
+    if not DISCORD_WEBHOOK_URL:
+        logging.warning("Discord webhook URL not set. Skipping alert.")
+        return
+    color = 0x00ff00 if signal == "BUY" else 0xff0000
+    embed = {
+        "title": f"{signal} Signal",
+        "description": f"**{symbol}**",
+        "color": color,
+        "fields": [
+            {"name": "Price", "value": f"${price:.2f}", "inline": True},
+            {"name": f"RSI ({RSI_PERIOD})", "value": f"{rsi:.2f}", "inline": True},
+            {"name": "Interval", "value": INTERVAL, "inline": True},
+        ],
+        "timestamp": datetime.utcnow().isoformat(),
+        "footer": {"text": "SuperTrend ATR+RSI Bot"}
+    }
+    try:
+        response = requests.post(DISCORD_WEBHOOK_URL, json={"embeds": [embed]})
+        if response.status_code == 204:
+            logging.info(f"✅ {signal} alert sent for {symbol} at ${price:.2f}")
+        else:
+            logging.error(f"❌ Discord API error: {response.status_code} - {response.text}")
+    except Exception as e:
+        logging.error(f"❌ Failed to send alert: {e}")
+
+
+# === Send Test Message ===
+def send_test_alert():
+    if not DISCORD_WEBHOOK_URL:
+        logging.error("❌ Test alert failed: Discord Webhook URL is not set.")
+        return False
+    embed = {
+        "title": "✅ Test Alert",
+        "description": "**Your bot is working perfectly!**",
+        "color": 0x3498db,
+        "fields": [
+            {"name": "Status", "value": "All systems operational", "inline": False},
+            {"name": "Time", "value": datetime.now().strftime("%Y-%m-%d %H:%M:%S"), "inline": False},
+        ],
+        "footer": {"text": "SuperTrend Signal Bot"}
+    }
+    try:
+        response = requests.post(DISCORD_WEBHOOK_URL, json={"embeds": [embed]})
+        return response.status_code == 204
+    except Exception as e:
+        logging.error(f"❌ Test alert failed: {e}")
+        return False
+
+
+# === Check One Stock ===
+def check_stock(symbol):
+    try:
+        if not is_market_open():
+            return
+
+        ticker = yf.Ticker(symbol)
+        data = ticker.history(period="2d", interval=INTERVAL)
+        data.columns = data.columns.str.lower()
+
+        if data.empty or len(data) < RSI_PERIOD + 10:
+            logging.warning(f"Insufficient data for {symbol} on {INTERVAL} interval. Skipping.")
+            return
+
+        rsi_indicator = ta.momentum.RSIIndicator(data['close'], window=RSI_PERIOD)
+        data['rsi'] = rsi_indicator.rsi()
+        data = calculate_supertrend(data, factor=SUPER_TREND_FACTOR, period=ATR_PERIOD)
+
+        data.dropna(inplace=True)
+        if len(data) < 2:
+            logging.warning(f"Not enough data for {symbol} after indicator calculation.")
+            return
+
+        latest = data.iloc[-1]
+        previous = data.iloc[-2]
+        current_trend = latest['supertrend_trend']
+        prev_trend = previous['supertrend_trend']
+        price = latest['close']
+        rsi_val = latest['rsi']
+
+        # ADDED THIS LINE TO PROVIDE A LIVE HEARTBEAT FOR EACH STOCK
+        logging.info(f"{symbol}: Price=${price:.2f}, RSI={rsi_val:.2f}, Trend={int(current_trend)}")
+
+        signal = None
+        if current_trend == 1 and prev_trend == -1:
+            signal = "BUY"
+        elif current_trend == -1 and prev_trend == 1:
+            signal = "SELL"
+
+        if signal:
+            # --- TREND FILTER LOGIC ---
+            logging.info(f"Potential {signal} for {symbol}. Checking 4-hour trend filter...")
+            long_term_data = ticker.history(period="1mo", interval="4h")
+            long_term_data.columns = long_term_data.columns.str.lower()
+
+            if len(long_term_data) < 50:
+                logging.warning(f"Not enough 4h data for {symbol} to apply trend filter. Allowing signal.")
+            else:
+                long_term_data['ma50'] = long_term_data['close'].rolling(window=50).mean()
+                last_long_term_price = long_term_data['close'].iloc[-1]
+                last_ma_value = long_term_data['ma50'].iloc[-1]
+
+                if signal == "BUY" and last_long_term_price < last_ma_value:
+                    logging.info(f"BUY signal for {symbol} blocked by 4-hour trend filter (price is below MA).")
+                    return
+
+                if signal == "SELL" and last_long_term_price > last_ma_value:
+                    logging.info(f"SELL signal for {symbol} blocked by 4-hour trend filter (price is above MA).")
+                    return
+
+            logging.info(f"✅ Signal for {symbol} passed trend filter. Sending alert.")
+            # --- END OF TREND FILTER LOGIC ---
+
+            alert = {
+                "time": datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
+                "symbol": symbol,
+                "signal": signal,
+                "price": f"${price:.2f}",
+                "rsi": f"{rsi_val:.2f}",
+                "interval": INTERVAL
+            }
+            with history_lock:
+                alert_history.insert(0, alert)
+                if len(alert_history) > 50:
+                    alert_history.pop()
+            send_discord_alert(symbol, signal, price, rsi_val)
+
+    except Exception as e:
+        logging.error(f"💥 Error processing {symbol}: {e}")
+
+
+# === Scanner Loop ===
+def run_scanner():
+    logging.info("🚀 Scanner started...")
+    while True:
+        with stocks_lock:
+            current_stocks = monitored_stocks.copy()
+
+        if not current_stocks:
+            logging.info("No stocks to monitor. Waiting...")
+        elif not is_market_open():
+            logging.info("Market is currently closed. Scanner is sleeping.")
+        else:
+            logging.info(f"Market is open. Checking {len(current_stocks)} stocks...")
+            for symbol in current_stocks:
+                check_stock(symbol)
+            logging.info("Finished checking all stocks for this interval.")
+
+        time.sleep(300)
+
+
+# === FLASK WEB APP (HTML is unchanged) ===
+app = Flask(__name__)
+app.secret_key = SECRET_KEY
+
+HTML_TEMPLATE = """
+<!DOCTYPE html>
+<html lang="en">
+<head>
+  <meta charset="UTF-8" />
+  <meta name="viewport" content="width=device-width, initial-scale=1.0"/>
+  <title>Stock Signal Dashboard</title>
+  <style>
+    body { font-family: Arial, sans-serif; margin: 20px; background: #f4f6f9; color: #333; }
+    h1, h2 { color: #2c3e50; }
+    .container { max-width: 1000px; margin: 0 auto; }
+    .section { background: #fff; padding: 20px; border-radius: 8px; box-shadow: 0 2px 5px rgba(0,0,0,0.1); margin-bottom: 20px; }
+    input[type="text"], input[type="password"] { width: calc(100% - 22px); padding: 10px; margin: 5px 0 10px 0; border: 1px solid #ddd; border-radius: 4px; }
+    button { padding: 10px 15px; border: none; border-radius: 4px; cursor: pointer; color: white; font-weight: bold; }
+    .btn-add { background: #27ae60; } .btn-add:hover { background: #2ecc71; }
+    .btn-remove { background: #e74c3c; padding: 5px 10px; font-size: 0.9em; } .btn-remove:hover { background: #c0392b; }
+    .btn-test { background: #9b59b6; } .btn-test:hover { background: #8e44ad; }
+    table { width: 100%; border-collapse: collapse; margin-top: 15px; }
+    th, td { padding: 12px; text-align: left; border-bottom: 1px solid #ddd; }
+    th { background-color: #34495e; color: white; }
+    tr:nth-child(even) { background-color: #f9f9f9; }
+    tr:hover { background-color: #f1f1f1; }
+    .buy { color: #27ae60; font-weight: bold; }
+    .sell { color: #c0392b; font-weight: bold; }
+    .empty { color: #7f8c8d; font-style: italic; text-align: center; padding: 20px; }
+    .flash { padding: 15px; margin-bottom: 20px; border-radius: 4px; border: 1px solid transparent; }
+    .flash.success { background: #d4edda; color: #155724; border-color: #c3e6cb; }
+    .flash.error { background: #f8d7da; color: #721c24; border-color: #f5c6cb; }
+    .flash.info { background: #d1ecf1; color: #0c5460; border-color: #bee5eb; }
+    .footer { margin-top: 30px; color: #7f8c8d; font-size: 0.9em; text-align: center; }
+    .form-inline { display: flex; align-items: center; gap: 10px; }
+    .form-inline input[type="password"] { width: 120px; margin: 0; }
+  </style>
+</head>
+<body>
+  <div class="container">
+    <h1>📈 SuperTrend ATR+RSI Dashboard</h1>
+    {% with messages = get_flashed_messages(with_categories=true) %}
+      {% if messages %}
+        {% for category, msg in messages %}
+          <div class="flash {{ category }}">{{ msg }}</div>
+        {% endfor %}
+      {% endif %}
+    {% endwith %}
+    <div class="section">
+      <h2>➕ Add Stock to Watchlist</h2>
+      <form method="POST" action="/add_stock">
+        <input type="text" name="symbol" placeholder="Enter symbol (e.g., AAPL)" required autocomplete="off" />
+        <input type="password" name="key" placeholder="Secret Key" required />
+        <button type="submit" class="btn-add">Add Stock</button>
+      </form>
+    </div>
+    <div class="section">
+      <h2>📋 Monitored Stocks</h2>
+      {% if stocks %}
+        <table>
+          <thead><tr><th>Symbol</th><th>Action</th></tr></thead>
+          <tbody>
+            {% for symbol in stocks %}
+            <tr>
+              <td><strong>{{ symbol }}</strong></td>
+              <td>
+                <form method="POST" action="/remove_stock" class="form-inline">
+                  <input type="hidden" name="symbol" value="{{ symbol }}" />
+                  <input type="password" name="key" placeholder="Key" required />
+                  <button type="submit" class="btn-remove">Remove</button>
+                </form>
+              </td>
+            </tr>
+            {% endfor %}
+          </tbody>
+        </table>
+      {% else %}
+        <p class="empty">🚫 No stocks are being monitored. Add one above to get started!</p>
+      {% endif %}
+    </div>
+    <div class="section">
+      <h2>🧪 Test Discord Webhook</h2>
+      <form method="POST" action="/test_discord">
+        <input type="password" name="key" placeholder="Secret Key" required />
+        <button type="submit" class="btn-test">📤 Send Test Message</button>
+      </form>
+    </div>
+    <div class="section">
+      <h2>🔔 Recent Alert History</h2>
+      {% if alerts %}
+        <table>
+          <thead><tr><th>Time</th><th>Symbol</th><th>Signal</th><th>Price</th><th>RSI</th><th>Interval</th></tr></thead>
+          <tbody>
+            {% for alert in alerts %}
+            <tr>
+              <td>{{ alert.time }}</td>
+              <td><strong>{{ alert.symbol }}</strong></td>
+              <td class="{{ alert.signal.lower() }}">{{ alert.signal }}</td>
+              <td>{{ alert.price }}</td>
+              <td>{{ alert.rsi }}</td>
+              <td>{{ alert.interval }}</td>
+            </tr>
+            {% endfor %}
+          </tbody>
+        </table>
+      {% else %}
+        <p class="empty">No alerts generated yet. Waiting for a signal...</p>
+      {% endif %}
+    </div>
+    <div class="footer">
+      Bot status as of: {{ now }} | Scan Interval: {{ interval }}
+    </div>
+  </div>
+</body>
+</html>
+"""
+
+
+@app.route('/', methods=['GET'])
+def home():
+    with stocks_lock:
+        stocks = sorted(monitored_stocks.copy())
+    with history_lock:
+        alerts = alert_history.copy()
+    now = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+    return render_template_string(HTML_TEMPLATE,
+                                  stocks=stocks,
+                                  alerts=alerts,
+                                  now=now,
+                                  interval=INTERVAL)
+
+
+@app.route('/add_stock', methods=['POST'])
+def add_stock():
+    symbol = request.form.get('symbol', '').strip().upper()
+    key = request.form.get('key', '').strip()
+    if key != SECRET_KEY:
+        flash("🔒 Access denied: Invalid secret key.", "error")
+    elif not symbol:
+        flash("⚠️ Please enter a stock symbol.", "error")
+    elif not is_valid_symbol(symbol):
+        flash(f"❌ Invalid symbol: '{symbol}'. Use 1–5 uppercase letters.", "error")
+    else:
+        with stocks_lock:
+            if symbol in monitored_stocks:
+                flash(f"🟡 '{symbol}' is already being monitored.", "info")
+            else:
+                monitored_stocks.append(symbol)
+                flash(f"✅ Successfully added '{symbol}' to the watchlist!", "success")
+                logging.info(f"➕ User added stock: {symbol}")
+    return home()
+
+
+@app.route('/remove_stock', methods=['POST'])
+def remove_stock():
+    symbol = request.form.get('symbol', '').strip().upper()
+    key = request.form.get('key', '').strip()
+    if key != SECRET_KEY:
+        flash("🔒 Access denied: Invalid secret key.", "error")
+    else:
+        with stocks_lock:
+            if symbol not in monitored_stocks:
+                flash(f"🔍 '{symbol}' is not on the list.", "info")
+            else:
+                monitored_stocks.remove(symbol)
+                flash(f"🗑️ Removed '{symbol}' from monitoring.", "success")
+                logging.info(f"➖ User removed stock: {symbol}")
+    return home()
+
+
+@app.route('/test_discord', methods=['POST'])
+def test_discord():
+    key = request.form.get('key', '').strip()
+    if key != SECRET_KEY:
+        flash("🔒 Access denied: Invalid secret key.", "error")
+    else:
+        success = send_test_alert()
+        if success:
+            flash("✅ Test message sent to Discord! Check your server.", "success")
+        else:
+            flash("❌ Failed to send test message. Check your Discord Webhook URL.", "error")
+    return home()
+
+
+def start():
+    scanner_thread = Thread(target=run_scanner, daemon=True)
+    scanner_thread.start()
+    logging.info("🌐 Starting Flask web server...")
+    app.run(host='0.0.0.0', port=8080)
+
+
+if __name__ == '__main__':
+    start()