feat: Add history

main.py

@@ -1,12 +1,14 @@
 import time
+import signal
+import threading
 from flask import Flask
 from server import app
 import server
 from gunicorn.app.base import BaseApplication
 import os
 import dotenv
-import sys
-import json
+import concurrent.futures
+import schedule
 
 
 class GunicornApp(BaseApplication):
@@ -23,20 +25,63 @@ class GunicornApp(BaseApplication):
     def load(self):
         return self.application
 
-if __name__ == '__main__':
-    workers = os.getenv('WORKERS')
-    threads = os.getenv('THREADS')
-    if workers is None:
-        workers = 1
-    if threads is None:
-        threads = 2
+
+def check():
+    print('Checking nodes...', flush=True)
+    server.check_nodes()
+
+
+def run_scheduler(stop_event):
+    schedule.every(1).minutes.do(check)
+    while not stop_event.is_set():
+        schedule.run_pending()
+        time.sleep(1)
+
+
+def run_gunicorn():
+    workers = os.getenv('WORKERS', 1)
+    threads = os.getenv('THREADS', 2)
     workers = int(workers)
     threads = int(threads)
+
     options = {
         'bind': '0.0.0.0:5000',
         'workers': workers,
         'threads': threads,
     }
-    gunicorn_app = GunicornApp(app, options)
-    print('Starting server with ' + str(workers) + ' workers and ' + str(threads) + ' threads', flush=True)
+
+    gunicorn_app = GunicornApp(server.app, options)
+    print(f'Starting server with {workers} workers and {threads} threads', flush=True)
     gunicorn_app.run()
+
+
+def signal_handler(sig, frame):
+    print("Shutting down gracefully...", flush=True)
+    stop_event.set()
+
+
+if __name__ == '__main__':
+    dotenv.load_dotenv()
+    
+    stop_event = threading.Event()
+    signal.signal(signal.SIGINT, signal_handler)
+    signal.signal(signal.SIGTERM, signal_handler)
+    
+    with concurrent.futures.ThreadPoolExecutor() as executor:
+        # Start the scheduler
+        scheduler_future = executor.submit(run_scheduler, stop_event)
+        
+        try:
+            # Run the Gunicorn server
+            run_gunicorn()
+        except KeyboardInterrupt:
+            print("Shutting down server...", flush=True)
+        finally:
+            stop_event.set()
+            scheduler_future.cancel()
+            try:
+                scheduler_future.result(timeout=5)
+            except concurrent.futures.CancelledError:
+                print("Scheduler stopped.")
+            except Exception as e:
+                print(f"Scheduler did not stop cleanly: {e}")