0.0.2.2 V Beta

LoadBalancer.py

@@ -19,8 +19,9 @@ download_progress = {}
 
 class LoadBalancer:
     def __init__(self, cache_dir, index_file, token, repo, polling_interval=10, max_retries=3, initial_delay=1):
-        self.version = "0.0.2.1 V Beta"
+        self.version = "0.0.2.2 V Beta"
         self.instances = []
+        self.instances_health = {}
         self.polling_interval = polling_interval
         self.max_retries = max_retries
         self.initial_delay = initial_delay
@@ -106,6 +107,7 @@ class LoadBalancer:
                     self.update_tv_store_json(title, season, episode, url)
 
         logging.info("Film and TV Stores processed successfully.")
+        self.update_instances_health(instance=instance_url, cache_size=cache_size)
 
     def start_polling(self):
         logging.info("Starting polling.")
@@ -222,7 +224,13 @@ class LoadBalancer:
             r'(?:/?|[/?]\S+)$', re.IGNORECASE)
         return re.match(regex, url) is not None
 
+#################################################################
 
+    def update_instances_health(self, instance, cache_size):
+        self.instances_health[instance] = cache_size
+        logging.info(f"Updated instance {instance} with cache size {cache_size}")
+
+#################################################################
     def load_json(self, file_path):
         """Load JSON data from a file."""
         with open(file_path, 'r') as file:

app.py

@@ -242,6 +242,10 @@ def register_instance():
 def get_instances():
     return load_balancer.instances
 
+@app.route('/api/instances/health',methods=["GET"])
+def get_instances_health():
+    return load_balancer.instances_health
+
 #############################################################
 # Routes
 @app.route('/')