0.0.2.6 V Beta fix

LoadBalancer.py

@@ -12,13 +12,14 @@ import logging
 from threading import Thread, Event, Timer
 from api import InstancesAPI
 
+
 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"
+        self.version = "0.0.2.6 V Beta"
         self.instances = []
         self.instances_health = {}
         self.polling_interval = polling_interval
@@ -95,39 +96,21 @@ class LoadBalancer:
 
         logging.info(f"Processing report from {instance_url}")
 
-        # Clear temporary JSON files before processing
-        self.clear_json_file(self.TEMP_FILM_STORE_JSON_PATH)
-        self.clear_json_file(self.TEMP_TV_STORE_JSON_PATH)
-
         # Process films
         for title, path in film_store.items():
             url = f"{instance_url}/api/film/{title.replace(' ', '%20')}"
-            self.update_temp_film_store_json(title, url)
+            self.update_film_store_json(title, url, temp=True)
 
         # 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')}"
-                    self.update_temp_tv_store_json(title, season, episode, url)
+                    self.update_tv_store_json(title, season, episode, url, temp=True)
 
         logging.info("Film and TV Stores processed successfully.")
-        self.replace_store_jsons()
         self.update_instances_health(instance=instance_url, cache_size=cache_size)
-
-    def replace_store_jsons(self):
-        """Replace the actual JSON contents with the temporary JSON contents."""
-        self.replace_json_content(self.TEMP_FILM_STORE_JSON_PATH, self.FILM_STORE_JSON_PATH)
-        self.replace_json_content(self.TEMP_TV_STORE_JSON_PATH, self.TV_STORE_JSON_PATH)
-
-    @staticmethod
-    def replace_json_content(temp_file_path, actual_file_path):
-        """Replace the content of the actual JSON file with the content of the temporary JSON file."""
-        with open(temp_file_path, 'r') as temp_file:
-            temp_data = json.load(temp_file)
-
-        with open(actual_file_path, 'w') as actual_file:
-            json.dump(temp_data, actual_file, indent=2)
+        self.replace_store_files()
 
     def start_polling(self):
         logging.info("Starting polling.")
@@ -168,26 +151,45 @@ class LoadBalancer:
             print(f"Error getting system proxies: {e}")
             return {}
 
-    def update_temp_film_store_json(self, title, url):
+    def update_film_store_json(self, title, url, temp=False):
         """
-        Updates the temporary film store JSON with the new file.
+        Updates the film store JSON with the new file.
 
         Args:
             title (str): The title of the film.
             url (str): The url.
+            temp (bool): If True, update the temporary JSON file. Defaults to False.
         """
-        film_store_data = self.read_json(self.TEMP_FILM_STORE_JSON_PATH)
+        file_path = self.TEMP_FILM_STORE_JSON_PATH if temp else self.FILM_STORE_JSON_PATH
+
+        film_store_data = {}
+        if os.path.exists(file_path):
+            with open(file_path, 'r') as json_file:
+                film_store_data = json.load(json_file)
+
         film_store_data[title] = url
 
-        with open(self.TEMP_FILM_STORE_JSON_PATH, 'w') as json_file:
+        with open(file_path, 'w') as json_file:
             json.dump(film_store_data, json_file, indent=2)
-        print(f'Temporary film store updated with {title}.')
+        print(f'Film store updated with {title}.')
 
-    def update_temp_tv_store_json(self, title, season, episode, url):
+    def update_tv_store_json(self, title, season, episode, url, temp=False):
         """
-        Updates the temporary TV store JSON with the new file, organizing by title, season, and episode.
+        Updates the TV store JSON with the new file, organizing by title, season, and episode.
+
+        Args:
+            title (str): The title of the TV show.
+            season (str): The season of the TV show.
+            episode (str): The episode of the TV show.
+            url (str): The url.
+            temp (bool): If True, update the temporary JSON file. Defaults to False.
         """
-        tv_store_data = self.read_json(self.TEMP_TV_STORE_JSON_PATH)
+        file_path = self.TEMP_TV_STORE_JSON_PATH if temp else self.TV_STORE_JSON_PATH
+
+        tv_store_data = {}
+        if os.path.exists(file_path):
+            with open(file_path, 'r') as json_file:
+                tv_store_data = json.load(json_file)
 
         if title not in tv_store_data:
             tv_store_data[title] = {}
@@ -197,10 +199,10 @@ class LoadBalancer:
 
         tv_store_data[title][season][episode] = url
 
-        with open(self.TEMP_TV_STORE_JSON_PATH, 'w') as json_file:
+        with open(file_path, 'w') as json_file:
             json.dump(tv_store_data, json_file, indent=2)
 
-        print(f'Temporary TV store updated with {title}, {season}, {episode}.')
+        print(f'TV store updated with {title}, {season}, {episode}.')
 
     @staticmethod
     def clear_json_file(file_path):
@@ -209,10 +211,13 @@ class LoadBalancer:
 
         :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:
@@ -220,6 +225,7 @@ class LoadBalancer:
 
     @staticmethod
     def is_valid_url(url):
+        # Simple URL validation (could be more complex if needed)
         regex = re.compile(
             r'^(?:http|ftp)s?://' # http:// or https://
             r'(?:(?:[A-Z0-9](?:[A-Z0-9-]{0,61}[A-Z0-9])?\.)+(?:[A-Z]{2,6}\.?|[A-Z0-9-]{2,}\.?)|' # domain...