try fix

LoadBalancer.py

@@ -37,30 +37,41 @@ class LoadBalancer:
         self.file_structure = None
 
         # Ensure CACHE_DIR exists
-        if not os.path.exists(self.CACHE_DIR):
-            os.makedirs(self.CACHE_DIR)
+        os.makedirs(self.CACHE_DIR, exist_ok=True)
 
-        for path in [self.FILM_STORE_JSON_PATH, self.TV_STORE_JSON_PATH]:
-            if not os.path.exists(path):
-                with open(path, 'w') as json_file:
-                    json.dump({}, json_file)
+        # Initialize JSON files
+        self.initialize_json_file(self.FILM_STORE_JSON_PATH)
+        self.initialize_json_file(self.TV_STORE_JSON_PATH)
 
         # Index the file structure
         indexer()
 
         # Load the file structure JSON
-        if not os.path.exists(self.INDEX_FILE):
-            raise FileNotFoundError(f"{self.INDEX_FILE} not found. Please make sure the file exists.")
-
-        with open(self.INDEX_FILE, 'r') as f:
-            self.file_structure = json.load(f)
+        self.load_file_structure()
 
+        # Start prefetching
         prefetch_thread = Thread(target=self.start_prefetching)
         prefetch_thread.daemon = True
         prefetch_thread.start()
 
+    def initialize_json_file(self, path):
+        """Initialize a JSON file with an empty object if it does not exist."""
+        if not os.path.exists(path):
+            with open(path, 'w') as json_file:
+                json.dump({}, json_file)
+            logging.info(f"Created JSON file: {path}")
+
+    def load_file_structure(self):
+        """Load the file structure from the index file."""
+        if not os.path.exists(self.INDEX_FILE):
+            raise FileNotFoundError(f"{self.INDEX_FILE} not found. Please make sure the file exists.")
+        
+        with open(self.INDEX_FILE, 'r') as f:
+            self.file_structure = json.load(f)
+        logging.info(f"Loaded file structure from {self.INDEX_FILE}")
 
     def register_instance(self, instance_url):
+        """Register a new instance."""
         if instance_url not in self.instances:
             self.instances.append(instance_url)
             logging.info(f"Registered instance {instance_url}")
@@ -68,6 +79,7 @@ class LoadBalancer:
             logging.info(f"Instance {instance_url} is already registered.")
 
     def remove_instance(self, instance_url):
+        """Remove an instance."""
         if instance_url in self.instances:
             self.instances.remove(instance_url)
             self.instances_health.pop(instance_url, None)
@@ -78,13 +90,14 @@ class LoadBalancer:
             logging.info(f"Instance {instance_url} not found for removal.")
 
     def get_reports(self):
+        """Fetch reports from instances and update JSON files."""
         reports = self.instances_api.fetch_reports()
         temp_film_store_path = 'temp_film_store.json'
         temp_tv_store_path = 'temp_tv_store.json'
 
-        # Create temporary JSON files
-        open(temp_film_store_path, 'w').close()
-        open(temp_tv_store_path, 'w').close()
+        # Initialize temporary JSON files
+        self.initialize_json_file(temp_film_store_path)
+        self.initialize_json_file(temp_tv_store_path)
 
         for instance_url in self.instances[:]:
             if instance_url in reports:
@@ -100,6 +113,7 @@ class LoadBalancer:
         self.replace_json_files(temp_tv_store_path, self.TV_STORE_JSON_PATH)
 
     def process_report(self, instance_url, report, temp_film_store_path, temp_tv_store_path):
+        """Process the report from an instance and update temporary JSON files."""
         film_store = report.get('film_store', {})
         tv_store = report.get('tv_store', {})
         cache_size = report.get('cache_size')
@@ -122,14 +136,7 @@ class LoadBalancer:
         self.update_instances_health(instance=instance_url, cache_size=cache_size)
 
     def update_film_store_json(self, title, url, temp_film_store_path):
-        """
-        Updates the temporary film store JSON with the new file.
-
-        Args:
-            title (str): The title of the film.
-            url (str): The url.
-            temp_film_store_path (str): Path to the temporary film store JSON.
-        """
+        """Update the temporary film store JSON with the new file."""
         film_store_data = {}
         if os.path.exists(temp_film_store_path):
             with open(temp_film_store_path, 'r') as json_file:
@@ -140,19 +147,10 @@ class LoadBalancer:
         with open(temp_film_store_path, 'w') as json_file:
             json.dump(film_store_data, json_file, indent=2)
 
-        print(f'Film store updated with {title}.')
+        logging.info(f'Film store updated with {title}. Content: {film_store_data}')
 
     def update_tv_store_json(self, title, season, episode, url, temp_tv_store_path):
-        """
-        Updates the temporary 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_tv_store_path (str): Path to the temporary TV store JSON.
-        """
+        """Update the temporary TV store JSON with the new file."""
         tv_store_data = {}
         if os.path.exists(temp_tv_store_path):
             with open(temp_tv_store_path, 'r') as json_file:
@@ -169,23 +167,22 @@ class LoadBalancer:
         with open(temp_tv_store_path, 'w') as json_file:
             json.dump(tv_store_data, json_file, indent=2)
 
-        print(f'TV store updated with {title}, {season}, {episode}.')
+        logging.info(f'TV store updated with {title}, {season}, {episode}. Content: {tv_store_data}')
 
     def replace_json_files(self, temp_file_path, actual_file_path):
-        """
-        Replaces the actual JSON file with the temporary JSON file.
-
-        Args:
-            temp_file_path (str): Path to the temporary JSON file.
-            actual_file_path (str): Path to the actual JSON file.
-        """
-        if os.path.exists(temp_file_path):
-            shutil.copy(temp_file_path, actual_file_path)
-            os.remove(temp_file_path)
-            print(f'Updated {actual_file_path} with new data.')
-
+        """Replace the actual JSON file with the temporary JSON file."""
+        try:
+            if os.path.exists(temp_file_path):
+                shutil.copy(temp_file_path, actual_file_path)
+                os.remove(temp_file_path)
+                logging.info(f'Updated {actual_file_path} with new data.')
+            else:
+                logging.error(f'Temporary file {temp_file_path} does not exist.')
+        except Exception as e:
+            logging.error(f'Error replacing JSON files: {e}')
 
     def start_polling(self):
+        """Start polling for reports."""
         logging.info("Starting polling.")
         while not self.stop_event.is_set():
             self.get_reports()
@@ -193,6 +190,7 @@ class LoadBalancer:
         logging.info("Polling stopped.")
 
     def stop_polling(self):
+        """Stop polling."""
         logging.info("Stopping polling.")
         self.stop_event.set()