Added a significant overhaul to the server manager.

2025-12-29 16:07:30 -05:00
parent 4f4e5276a6
commit 216e18a7b4
796 changed files with 528631 additions and 2671 deletions
@@ -0,0 +1,127 @@
+import os
+import re
+from collections import defaultdict
+import logging
+from .samba_manager import SambaManager
+from .config import Settings
+from smb.smb_structs import OperationFailure
+
+# --- Dedicated logger for ComicsManager ---
+comics_logger = logging.getLogger('comics_manager')
+comics_logger.setLevel(logging.INFO)
+comics_logger.propagate = False
+if not comics_logger.handlers:
+    comics_log_handler = logging.FileHandler("logs/comics_organization.log", mode='a')
+    comics_log_handler.setFormatter(logging.Formatter('%(asctime)s - %(levelname)s - %(message)s'))
+    comics_logger.addHandler(comics_log_handler)
+    comics_logger.addHandler(logging.StreamHandler())
+
+class ComicsManager:
+
+    @staticmethod
+    def _get_series_name(filename):
+        base_name, _ = os.path.splitext(filename)
+        pattern = re.compile(r'[-_\s]*(v(ol)?|c(h)?|chapter|issue|ep|episode)[-_\s]*\d+.*|[-_\s]+\d+$', re.IGNORECASE)
+        match = pattern.search(base_name)
+        if match:
+            return base_name[:match.start()].strip('-_ ')
+        return base_name.strip('-_ ')
+
+    def cleanup_toberead(self, samba_manager: SambaManager):
+        comics_logger.info("Starting cleanup of 'toberead' directory...")
+        comics_root = "/comics/toberead"
+        try:
+            items_in_toberead = samba_manager.list_path(comics_root)
+            if "error" in items_in_toberead:
+                raise Exception(f"Failed to list files in 'toberead': {items_in_toberead['error']}")
+
+            for item in items_in_toberead:
+                item_path = item["path"]
+                item_name = item["name"]
+                if item["is_directory"]:
+                    comics_logger.info(f"Deleting directory '{item_name}' and all its contents.")
+                    try:
+                        samba_manager.delete_directory_recursive(item_path)
+                        comics_logger.info(f"Successfully deleted directory: {item_name}")
+                    except OperationFailure as e:
+                        comics_logger.error(f"Failed to delete directory {item_name}: {e}")
+                else:
+                    comics_logger.info(f"Deleting file '{item_name}'.")
+                    try:
+                        samba_manager.delete_file(item_path)
+                        comics_logger.info(f"Successfully deleted file: {item_name}")
+                    except OperationFailure as e:
+                        comics_logger.error(f"Failed to delete file {item_name}: {e}")
+            
+            comics_logger.info("Cleanup of 'toberead' finished.")
+
+        except Exception as e:
+            comics_logger.error(f"An error occurred during 'toberead' cleanup: {e}", exc_info=True)
+
+    def organize_comics(self, samba_manager: SambaManager):
+        comics_logger.info("Starting comics organization process...")
+        comics_root = "/comics/toberead"
+        comics_dest_root = "/comics/manga"
+        try:
+            items_in_toberead = samba_manager.list_path(comics_root)
+            if "error" in items_in_toberead:
+                raise Exception(f"Failed to list files in 'toberead': {items_in_toberead['error']}")
+
+            series_chapters = defaultdict(list)
+            for item in items_in_toberead:
+                if not item["is_directory"]:
+                    series_name = self._get_series_name(item["name"])
+                    series_chapters[series_name].append(item)
+            
+            comics_logger.info(f"Found {len(series_chapters)} series to process.")
+
+            for series_name, chapters in series_chapters.items():
+                comics_logger.info(f"Processing series: {series_name}")
+                final_series_folder_path = f"{comics_dest_root}/{series_name}".replace("\\", "/")
+                
+                try:
+                    samba_manager.create_directory(final_series_folder_path)
+                except OperationFailure as e:
+                    if "FILE_OBJECT_NAME_COLLISION" not in str(e):
+                        comics_logger.error(f"Could not create directory {final_series_folder_path}: {e}")
+                
+                for chapter_item in chapters:
+                    original_path = chapter_item["path"].replace("\\", "/")
+                    original_filename = chapter_item["name"]
+                    
+                    path_to_move = ""
+                    filename_to_move = ""
+
+                    if original_filename.lower().endswith('.zip'):
+                        cbz_filename = os.path.splitext(original_filename)[0] + '.cbz'
+                        temp_cbz_path = f"{comics_root}/{cbz_filename}".replace("\\", "/")
+                        
+                        try:
+                            samba_manager.rename_file(original_path, temp_cbz_path)
+                            path_to_move = temp_cbz_path
+                            filename_to_move = cbz_filename
+                            comics_logger.info(f"Renamed {original_filename} to {cbz_filename}")
+                        except OperationFailure as e:
+                            comics_logger.error(f"Failed to rename ZIP {original_filename}: {e}")
+                            continue
+                    
+                    elif original_filename.lower().endswith('.cbz'):
+                        path_to_move = original_path
+                        filename_to_move = original_filename
+                    else:
+                        continue
+
+                    final_cbz_path = f"{final_series_folder_path}/{filename_to_move}".replace("\\", "/")
+                    
+                    try:
+                        samba_manager.rename_file(path_to_move, final_cbz_path)
+                        comics_logger.info(f"Moved {filename_to_move} to {final_series_folder_path}")
+                    except OperationFailure as e:
+                        comics_logger.error(f"Failed to move {filename_to_move} to {final_series_folder_path}: {e}")
+
+        except Exception as e:
+            comics_logger.error(f"An error occurred during comics organization: {e}", exc_info=True)
+        
+        comics_logger.info("Comics organization process finished.")
+            
+        return