fix

LoadBalancer.py

@@ -18,8 +18,8 @@ CACHE_DIR = os.getenv("CACHE_DIR")
 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.5 V Beta"
+    def __init__(self, cache_dir, index_file, token, repo, polling_interval=1, max_retries=3, initial_delay=1):
+        self.version = "0.0.2.4 V Beta"
         self.instances = []
         self.instances_health = {}
         self.polling_interval = polling_interval
@@ -93,57 +93,31 @@ class LoadBalancer:
         cache_size = report.get('cache_size')
 
         logging.info(f"Processing report from {instance_url}")
-
-        temp_film_store_path = os.path.join(self.CACHE_DIR, "temp_film_store.json")
-        temp_tv_store_path = os.path.join(self.CACHE_DIR, "temp_tv_store.json")
-
-        temp_film_store_data = {}
-        temp_tv_store_data = {}
-
-        if os.path.exists(temp_film_store_path):
-            with open(temp_film_store_path, 'r') as json_file:
-                temp_film_store_data = json.load(json_file)
-
-        if os.path.exists(temp_tv_store_path):
-            with open(temp_tv_store_path, 'r') as json_file:
-                temp_tv_store_data = json.load(json_file)
+        # logging.info(f"Film Store: {film_store}")
+        # logging.info(f"TV Store: {tv_store}")
+        # logging.info(f"Cache Size: {cache_size}")
 
         # Process films
         for title, path in film_store.items():
             url = f"{instance_url}/api/film/{title.replace(' ', '%20')}"
-            temp_film_store_data[title] = url
+            self.update_film_store_json(title, url)
 
+ 
         # Process TV shows
         for title, seasons in tv_store.items():
             for season, episodes in seasons.items():
                 for episode, path in episodes.items():
                     url = f"{instance_url}/api/tv/{title.replace(' ', '%20')}/{season.replace(' ', '%20')}/{episode.replace(' ', '%20')}"
-                    if title not in temp_tv_store_data:
-                        temp_tv_store_data[title] = {}
-                    if season not in temp_tv_store_data[title]:
-                        temp_tv_store_data[title][season] = {}
-                    temp_tv_store_data[title][season][episode] = url
-
-        # Write temporary JSON files
-        with open(temp_film_store_path, 'w') as json_file:
-            json.dump(temp_film_store_data, json_file, indent=2)
-
-        with open(temp_tv_store_path, 'w') as json_file:
-            json.dump(temp_tv_store_data, json_file, indent=2)
-
-        logging.info("Temporary Film and TV Stores processed successfully.")
-
-        # Replace actual JSON files with temporary JSON files
-        os.replace(temp_film_store_path, self.FILM_STORE_JSON_PATH)
-        os.replace(temp_tv_store_path, self.TV_STORE_JSON_PATH)
+                    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.")
         while not self.stop_event.is_set():
             self.get_reports()
-            #time.sleep(self.polling_interval)
+            time.sleep(self.polling_interval)
         logging.info("Polling stopped.")
 
     def stop_polling(self):