fix

LoadBalancer.py

@@ -95,10 +95,12 @@ class LoadBalancer:
         # logging.info(f"Cache Size: {cache_size}")
 
         # Process films
+        self.clear_json_file(self.FILM_STORE_JSON_PATH)
         for title, path in film_store.items():
             url = f"{instance_url}/api/film/{title.replace(' ', '%20')}"
             self.update_film_store_json(title, url)
 
+        self.clear_json_file(self.TV_STORE_JSON_PATH)
         # Process TV shows
         for title, seasons in tv_store.items():
             for season, episodes in seasons.items():
@@ -148,8 +150,7 @@ class LoadBalancer:
             print(f"Error getting system proxies: {e}")
             return {}
 
-    @staticmethod
-    def update_film_store_json(title, url):
+    def update_film_store_json(self,title, url):
         """
         Updates the film store JSON with the new file.
 
@@ -157,7 +158,7 @@ class LoadBalancer:
             title (str): The title of the film.
             url (str): The url.
         """
-        FILM_STORE_JSON_PATH = os.path.join(CACHE_DIR, "film_store.json")
+        FILM_STORE_JSON_PATH = self.FILM_STORE_JSON_PATH
 
         film_store_data = {}
         if os.path.exists(FILM_STORE_JSON_PATH):
@@ -170,18 +171,11 @@ class LoadBalancer:
             json.dump(film_store_data, json_file, indent=2)
         print(f'Film store updated with {title}.')
 
-    @staticmethod
-    def download_episode(title, season, episode):
-        """
-        Downloads a episode to a specific instance
-        """
-
-    @staticmethod
-    def update_tv_store_json(title, season, episode, url):
+    def update_tv_store_json(self, title, season, episode, url):
         """
         Updates the TV store JSON with the new file, organizing by title, season, and episode.
         """
-        TV_STORE_JSON_PATH = os.path.join(CACHE_DIR, "tv_store.json")
+        TV_STORE_JSON_PATH = self.TV_STORE_JSON_PATH
 
         tv_store_data = {}
         if os.path.exists(TV_STORE_JSON_PATH):
@@ -201,6 +195,26 @@ class LoadBalancer:
 
         print(f'TV store updated with {title}, {season}, {episode}.')
 
+    @staticmethod
+    def clear_json_file(file_path):
+        """
+        Clears the data in a JSON file by replacing its contents with an empty JSON object.
+
+        :param file_path: Path to the JSON file to be cleared
+        """
+        # Create an empty dictionary to replace the existing data
+        empty_data = {}
+
+        try:
+            # Open the file in write mode
+            with open(file_path, 'w') as json_file:
+                # Write the empty dictionary to the file
+                json.dump(empty_data, json_file)
+            print(f"Successfully cleared the data in {file_path}")
+        except Exception as e:
+            print(f"An error occurred while clearing the data in {file_path}: {e}")
+
+
     @staticmethod
     def is_valid_url(url):
         # Simple URL validation (could be more complex if needed)
@@ -298,7 +312,7 @@ class LoadBalancer:
             status = result["status"]
             progress_url = f'{best_instance}/api/progress/{episode_id}'
             response = {
-                "film_id":episode_id,
+                "episode_id":episode_id,
                 "status":status,
                 "progress_url":progress_url
             }