Spaces:

Alvin3y1
/

ping

Paused

App Files Files Community

Alvin3y1 commited on Mar 8

Commit

706ce69

verified ·

1 Parent(s): 14e0dfb

Update app.py

Browse files

Files changed (1) hide show

app.py +69 -236

app.py CHANGED Viewed

@@ -3,15 +3,15 @@ import time
 import requests
 import os
 import datetime
-from flask import Flask, request, jsonify
 app = Flask(__name__)
 # --- CONFIGURATION ---
 URL_FILE = 'urls.txt'
 SELF_PING_URL = "https://alvin3y1-ping.hf.space/"
-INTERVAL_HOURS = 4
-INTERVAL_SECONDS = INTERVAL_HOURS * 3600
 # --- GLOBAL STATE ---
 monitoring_state = {}
@@ -19,304 +19,137 @@ scan_state = {
     'is_scanning': False,
     'total': 0,
     'completed': 0,
-    'last_scan_time': 'Waiting for initial scan...'
 }
 state_lock = threading.Lock()
-# --- HTML / FRONTEND ---
-# Pure static HTML using Tailwind CSS for a premium, minimal UI.
 HTML_TEMPLATE = """
 <!DOCTYPE html>
 <html lang="en" class="dark">
 <head>
     <meta charset="UTF-8">
     <meta name="viewport" content="width=device-width, initial-scale=1.0">
-    <title>Status | Alvin</title>
     <script src="https://cdn.tailwindcss.com"></script>
-    <script>
-        tailwind.config = {
-            darkMode: 'class',
-            theme: {
-                extend: {
-                    colors: { bgMain: '#0a0a0a', card: '#111111', borderSubtle: '#222222' },
-                    fontFamily: { sans: ['Inter', 'system-ui', 'sans-serif'], mono: ['ui-monospace', 'monospace'] }
-                }
-            }
-        }
-    </script>
-    <style>
-        body { background-color: #0a0a0a; color: #ededed; }
-        .fade-in { animation: fadeIn 0.4s ease-in-out; }
-        @keyframes fadeIn { from { opacity: 0; transform: translateY(5px); } to { opacity: 1; transform: translateY(0); } }
-        /* Smooth progress bar transition */
-        #progress-fill { transition: width 0.3s ease-in-out; }
-    </style>
 </head>
-<body class="font-sans antialiased min-h-screen flex flex-col items-center py-12 px-4 sm:px-6">
-    <div class="w-full max-w-3xl">
-        <div class="flex flex-col sm:flex-row justify-between items-start sm:items-center mb-8 border-b border-borderSubtle pb-6">
-            <div>
-                <h1 class="text-2xl font-semibold tracking-tight">System Status</h1>
-                <p id="last-updated" class="text-sm text-gray-500 mt-1">Fetching latest data...</p>
-            </div>
-            <button onclick="triggerScan()" id="scan-btn" class="mt-4 sm:mt-0 px-4 py-2 bg-white text-black text-sm font-medium rounded-md hover:bg-gray-200 transition-colors focus:outline-none focus:ring-2 focus:ring-offset-2 focus:ring-white focus:ring-offset-bgMain">
-                Run Diagnostics
-            </button>
         </div>
-        <div id="progress-container" class="hidden mb-8 fade-in">
-            <div class="flex justify-between text-xs text-gray-400 mb-2">
-                <span id="progress-text">Scanning...</span>
-                <span id="progress-percent">0%</span>
-            </div>
-            <div class="w-full h-1 bg-borderSubtle rounded-full overflow-hidden">
-                <div id="progress-fill" class="h-full bg-blue-500 w-0"></div>
             </div>
         </div>
-        <div id="status-container" class="space-y-4">
-            </div>
     </div>
     <script>
-        let isScanning = false;
         async function fetchStatus() {
-            try {
-                const res = await fetch('/api/status');
-                const data = await res.json();
-                updateUI(data);
-            } catch (err) {
-                console.error("Failed to fetch status", err);
-            }
-        }
-        async function triggerScan() {
-            if (isScanning) return;
-            document.getElementById('scan-btn').disabled = true;
-            document.getElementById('scan-btn').innerText = 'Starting...';
-            await fetch('/api/scan', { method: 'POST' });
-            fetchStatus(); // immediate update
-        }
-        async function retryUrl(url) {
-            await fetch('/api/retry', {
-                method: 'POST',
-                headers: { 'Content-Type': 'application/json' },
-                body: JSON.stringify({ url: url })
-            });
-            fetchStatus(); // immediate update
-        }
-        function updateUI(data) {
-            // 1. Update Scan State
-            const scanBtn = document.getElementById('scan-btn');
-            const progressContainer = document.getElementById('progress-container');
-            const progressFill = document.getElementById('progress-fill');
-            const progressText = document.getElementById('progress-text');
-            const progressPercent = document.getElementById('progress-percent');
-            isScanning = data.scan.is_scanning;
-            if (isScanning) {
-                scanBtn.disabled = true;
-                scanBtn.innerText = 'Scanning...';
-                progressContainer.classList.remove('hidden');
-                let percent = 0;
-                if (data.scan.total > 0) {
-                    percent = Math.round((data.scan.completed / data.scan.total) * 100);
-                }
-                progressFill.style.width = percent + '%';
-                progressText.innerText = `Checking ${data.scan.completed} of ${data.scan.total} services...`;
-                progressPercent.innerText = percent + '%';
             } else {
-                scanBtn.disabled = false;
-                scanBtn.innerText = 'Run Diagnostics';
-                progressContainer.classList.add('hidden');
             }
-            // 2. Update Header Time
-            document.getElementById('last-updated').innerText = `Last cycle: ${data.scan.last_scan_time}`;
-            // 3. Render Status List (Showing ONLY DOWN urls)
-            const container = document.getElementById('status-container');
-            const downUrls = data.down_urls;
-            if (downUrls.length === 0) {
-                // All Operational State
-                container.innerHTML = `
-                    <div class="flex items-center justify-center py-16 bg-card border border-borderSubtle rounded-xl fade-in">
-                        <div class="flex flex-col items-center">
-                            <div class="h-3 w-3 bg-green-500 rounded-full shadow-[0_0_15px_rgba(34,197,94,0.5)] mb-4 animate-pulse"></div>
-                            <h2 class="text-lg font-medium text-gray-200">All Systems Operational</h2>
-                            <p class="text-sm text-gray-500 mt-1">Monitoring ${data.total_monitored} endpoints</p>
-                        </div>
-                    </div>
-                `;
-            } else {
-                // Down URLs State
-                let html = `
-                    <div class="mb-4 flex items-center gap-2">
-                        <div class="h-2 w-2 bg-red-500 rounded-full shadow-[0_0_10px_rgba(239,68,68,0.8)]"></div>
-                        <h2 class="text-sm font-medium text-red-400 uppercase tracking-widest">${downUrls.length} Service(s) Down</h2>
-                    </div>
-                `;
-                downUrls.forEach(item => {
-                    html += `
-                        <div class="bg-card border border-borderSubtle rounded-xl p-5 flex flex-col sm:flex-row justify-between items-start sm:items-center gap-4 fade-in">
-                            <div class="overflow-hidden">
-                                <a href="${item.url}" target="_blank" class="text-white font-mono text-sm hover:underline truncate block w-full max-w-md">
-                                    ${item.url}
-                                </a>
-                                <div class="flex items-center gap-3 mt-2">
-                                    <span class="inline-flex items-center rounded-md bg-red-400/10 px-2 py-1 text-xs font-medium text-red-400 ring-1 ring-inset ring-red-400/20">
-                                        ERR ${item.code}
-                                    </span>
-                                    <span class="text-xs text-gray-500">${item.msg}</span>
-                                </div>
-                            </div>
-                            <button onclick="retryUrl('${item.url}')" class="px-3 py-1.5 bg-borderSubtle hover:bg-gray-700 text-xs font-medium rounded-md transition-colors whitespace-nowrap">
-                                Retry
-                            </button>
                         </div>
-                    `;
-                });
-                container.innerHTML = html;
-            }
         }
-        // Poll API every 2 seconds for real-time updates
         setInterval(fetchStatus, 2000);
-        // Initial fetch on load
         fetchStatus();
     </script>
 </body>
 </html>
 """
 def get_all_urls():
-    """Reads URLs from file and appends the hardwired self-ping URL."""
-    urls = []
     if os.path.exists(URL_FILE):
         with open(URL_FILE, 'r') as f:
-            urls = [line.strip() for line in f if line.strip()]
-    if SELF_PING_URL not in urls:
-        urls.append(SELF_PING_URL)
-    with state_lock:
-        for url in urls:
-            if url not in monitoring_state:
-                monitoring_state[url] = {
-                    'status': 'PENDING',
-                    'code': '-',
-                    'time': '-',
-                    'msg': 'Waiting for check...'
-                }
-    return urls
 def check_single_url(url):
-    """Pings a single URL and updates the global state."""
-    timestamp = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
-    result = {}
     try:
-        response = requests.get(url, timeout=10)
-        if response.status_code >= 400:
-            result = {'status': 'DOWN', 'code': response.status_code, 'time': timestamp, 'msg': "HTTP Error"}
-        else:
-            result = {'status': 'UP', 'code': response.status_code, 'time': timestamp, 'msg': "OK"}
-    except requests.exceptions.Timeout:
-        result = {'status': 'DOWN', 'code': 'TIMEOUT', 'time': timestamp, 'msg': "Request Timed Out"}
-    except requests.exceptions.ConnectionError:
-        result = {'status': 'DOWN', 'code': 'ERR', 'time': timestamp, 'msg': "Connection Failed"}
     except Exception as e:
-        result = {'status': 'DOWN', 'code': 'ERR', 'time': timestamp, 'msg': "Failure"} # Hiding raw exception for cleaner UI
     with state_lock:
-        monitoring_state[url] = result
 def perform_full_scan():
-    """Iterates through all URLs and updates progress."""
     urls = get_all_urls()
     with state_lock:
-        if scan_state['is_scanning']:
-            return # Prevent concurrent full scans
-        scan_state['is_scanning'] = True
-        scan_state['total'] = len(urls)
-        scan_state['completed'] = 0
     for url in urls:
         check_single_url(url)
         with state_lock:
             scan_state['completed'] += 1
     with state_lock:
-        scan_state['is_scanning'] = False
-        scan_state['last_scan_time'] = datetime.datetime.now().strftime("%b %d, %H:%M:%S")
 def background_worker():
-    """Runs the full scan periodically."""
     while True:
         perform_full_scan()
         time.sleep(INTERVAL_SECONDS)
-# --- FLASK ROUTES (API-FIRST) ---
 @app.route('/')
-def home():
-    """Serves the static SPA template."""
-    return HTML_TEMPLATE
-@app.route('/api/status', methods=['GET'])
 def get_status():
-    """Returns real-time data to the frontend."""
     with state_lock:
-        state_copy = monitoring_state.copy()
-        scan_copy = scan_state.copy()
-    # Filter out only the DOWN urls to send back to the client
-    down_urls = []
-    for url, info in state_copy.items():
-        if info['status'] != 'UP':
-            down_urls.append({'url': url, **info})
-    return jsonify({
-        'scan': scan_copy,
-        'down_urls': down_urls,
-        'total_monitored': len(state_copy)
-    })
 @app.route('/api/scan', methods=['POST'])
-def scan_all():
-    """Triggers a background full scan."""
     threading.Thread(target=perform_full_scan, daemon=True).start()
     return jsonify({"status": "started"})
 @app.route('/api/retry', methods=['POST'])
-def retry_url():
-    """Retries a single URL."""
-    data = request.json
-    url_to_retry = data.get('url')
-    if url_to_retry:
-        threading.Thread(target=check_single_url, args=(url_to_retry,), daemon=True).start()
     return jsonify({"status": "retrying"})
 if __name__ == '__main__':
-    # 1. Initialize URLs
-    get_all_urls()
-    # 2. Create the background thread
-    pinger_thread = threading.Thread(target=background_worker, daemon=True)
-    pinger_thread.start()
-    # 3. Start the Flask server
-    # Note: For strict production environments, replace app.run with Gunicorn/Waitress.
-    # Because you are deploying to a specific host (like HuggingFace), keeping this as-is works perfectly.
-    print("Server starting on port 7860...")
-    app.run(host='0.0.0.0', port=7860)

 import requests
 import os
 import datetime
+from flask import Flask, jsonify, request
+from wsgiref.simple_server import make_server
 app = Flask(__name__)
 # --- CONFIGURATION ---
 URL_FILE = 'urls.txt'
 SELF_PING_URL = "https://alvin3y1-ping.hf.space/"
+INTERVAL_SECONDS = 4 * 3600  # 4 hours
 # --- GLOBAL STATE ---
 monitoring_state = {}
     'is_scanning': False,
     'total': 0,
     'completed': 0,
+    'last_scan_time': 'Never'
 }
 state_lock = threading.Lock()
+# --- FRONTEND TEMPLATE ---
+# Served as a static string to keep the project to one file.
 HTML_TEMPLATE = """
 <!DOCTYPE html>
 <html lang="en" class="dark">
 <head>
     <meta charset="UTF-8">
     <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>Service Status</title>
     <script src="https://cdn.tailwindcss.com"></script>
 </head>
+<body class="bg-black text-gray-200 min-h-screen p-6 font-sans">
+    <div class="max-w-2xl mx-auto">
+        <div class="flex justify-between items-center mb-8 border-b border-gray-800 pb-4">
+            <h1 class="text-xl font-bold">System Status</h1>
+            <button onclick="triggerScan()" id="scan-btn" class="bg-gray-800 hover:bg-gray-700 px-3 py-1 text-sm rounded transition">Run Scan</button>
         </div>
+        <div id="progress" class="hidden mb-6">
+            <div class="w-full bg-gray-800 h-1 rounded-full overflow-hidden">
+                <div id="bar" class="bg-blue-500 h-full w-0 transition-all duration-300"></div>
             </div>
         </div>
+        <div id="alerts" class="space-y-3"></div>
     </div>
     <script>
         async function fetchStatus() {
+            const res = await fetch('/api/status');
+            const data = await res.json();
+            // Handle Progress
+            const prog = document.getElementById('progress');
+            if (data.scan.is_scanning) {
+                prog.classList.remove('hidden');
+                const percent = (data.scan.completed / data.scan.total) * 100;
+                document.getElementById('bar').style.width = percent + '%';
             } else {
+                prog.classList.add('hidden');
             }
+            // Handle Down URLs
+            const container = document.getElementById('alerts');
+            container.innerHTML = data.down_urls.length ? '' : '<p class="text-green-500 text-center py-10">All Systems Operational</p>';
+            data.down_urls.forEach(u => {
+                container.innerHTML += `
+                    <div class="bg-red-950/20 border border-red-900/50 p-4 rounded flex justify-between items-center">
+                        <div>
+                            <div class="text-sm font-mono text-red-200">${u.url}</div>
+                            <div class="text-xs text-red-500 mt-1">${u.msg} (${u.code})</div>
                         </div>
+                        <button onclick="retry('${u.url}')" class="text-xs bg-red-900 px-2 py-1 rounded">Retry</button>
+                    </div>`;
+            });
+        }
+        async function triggerScan() {
+            await fetch('/api/scan', {method: 'POST'});
+            fetchStatus();
+        }
+        async function retry(url) {
+            await fetch('/api/retry', {method: 'POST', headers:{'Content-Type':'application/json'}, body:JSON.stringify({url})});
+            fetchStatus();
         }
         setInterval(fetchStatus, 2000);
         fetchStatus();
     </script>
 </body>
 </html>
 """
+# --- LOGIC ---
 def get_all_urls():
+    urls = [SELF_PING_URL]
     if os.path.exists(URL_FILE):
         with open(URL_FILE, 'r') as f:
+            urls.extend([l.strip() for l in f if l.strip()])
+    return list(set(urls))
 def check_single_url(url):
+    timestamp = datetime.datetime.now().strftime("%H:%M:%S")
     try:
+        r = requests.get(url, timeout=10)
+        res = {'status': 'UP' if r.status_code < 400 else 'DOWN', 'code': r.status_code, 'time': timestamp, 'msg': 'OK' if r.status_code < 400 else 'Error'}
     except Exception as e:
+        res = {'status': 'DOWN', 'code': 'ERR', 'time': timestamp, 'msg': str(e)[:20]}
     with state_lock:
+        monitoring_state[url] = res
 def perform_full_scan():
     urls = get_all_urls()
     with state_lock:
+        scan_state.update({'is_scanning': True, 'total': len(urls), 'completed': 0})
     for url in urls:
         check_single_url(url)
         with state_lock:
             scan_state['completed'] += 1
     with state_lock:
+        scan_state.update({'is_scanning': False, 'last_scan_time': datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")})
 def background_worker():
     while True:
         perform_full_scan()
         time.sleep(INTERVAL_SECONDS)
+# --- ROUTES ---
 @app.route('/')
+def home(): return HTML_TEMPLATE
+@app.route('/api/status')
 def get_status():
     with state_lock:
+        down = [{'url': u, **s} for u, s in monitoring_state.items() if s['status'] == 'DOWN']
+        return jsonify({'scan': scan_state, 'down_urls': down})
 @app.route('/api/scan', methods=['POST'])
+def scan():
     threading.Thread(target=perform_full_scan, daemon=True).start()
     return jsonify({"status": "started"})
 @app.route('/api/retry', methods=['POST'])
+def retry():
+    threading.Thread(target=check_single_url, args=(request.json['url'],), daemon=True).start()
     return jsonify({"status": "retrying"})
 if __name__ == '__main__':
+    threading.Thread(target=background_worker, daemon=True).start()
+    print("Server starting on http://0.0.0.0:7860")
+    make_server('0.0.0.0', 7860, app).serve_forever()