backup

processors/validator.py

@@ -3,99 +3,64 @@
 import os
 import zipfile
 import rarfile
-from core.constants import IMAGE_EXTENSIONS, TRASH_FILES
 
 
-IMAGE_EXTENSIONS = {".jpg", ".jpeg", ".png", ".webp"}
-TRASH_FILES = {"thumbs.db", ".ds_store"}
-
 class ValidationResult:
     def __init__(self, path):
         self.path = path
-        self.is_valid = True
+        self.real_format = None     # "zip", "rar", None
+        self.extension = None       # ".cbz" o ".cbr"
         self.errors = []
         self.warnings = []
-        self.images = []
-        self.real_format = None  # "rar", "zip", None
-
-    def add_error(self, msg):
-        self.is_valid = False
-        self.errors.append(msg)
-
-    def add_warning(self, msg):
-        self.warnings.append(msg)
 
     def __str__(self):
-        status = "OK" if self.is_valid else "ERROR"
-        return f"[{status}] {self.path}"
+        msg = f"Validación de: {self.path}\n"
+        msg += f"  Formato real: {self.real_format}\n"
+        msg += f"  Extensión: {self.extension}\n"
+        if self.errors:
+            msg += "  Errores:\n"
+            for e in self.errors:
+                msg += f"    - {e}\n"
+        if self.warnings:
+            msg += "  Avisos:\n"
+            for w in self.warnings:
+                msg += f"    - {w}\n"
+        return msg
 
 
-def try_open_rar(path):
+def detect_real_format(path):
+    """Devuelve 'zip', 'rar' o None."""
     try:
-        archive = rarfile.RarFile(path, "r")
-        archive.namelist()  # fuerza lectura
-        return archive
+        zipfile.ZipFile(path).close()
+        return "zip"
     except:
-        return None
+        pass
 
-
-def try_open_zip(path):
     try:
-        archive = zipfile.ZipFile(path, "r")
-        archive.namelist()
-        return archive
+        rarfile.RarFile(path).close()
+        return "rar"
     except:
-        return None
+        pass
+
+    return None
 
 
 def validate_comic(path):
     result = ValidationResult(path)
     ext = os.path.splitext(path)[1].lower()
+    result.extension = ext
 
-    archive = None
+    real = detect_real_format(path)
+    result.real_format = real
 
-    # 1) Intentar abrir como RAR
-    if ext == ".cbr":
-        archive = try_open_rar(path)
-        if archive:
-            result.real_format = "rar"
-        else:
-            # 2) Intentar abrir como ZIP (CBZ renombrado)
-            archive = try_open_zip(path)
-            if archive:
-                result.real_format = "zip"
-                result.add_warning("El archivo tiene extensión .cbr pero es un ZIP (CBZ renombrado)")
-            else:
-                result.add_error("No se pudo abrir como RAR ni como ZIP")
-                return result
-
-    # 3) Si es CBZ, abrir como ZIP
-    elif ext == ".cbz":
-        archive = try_open_zip(path)
-        if archive:
-            result.real_format = "zip"
-        else:
-            result.add_error("No se pudo abrir el archivo ZIP")
-            return result
-
-    else:
-        result.add_error("Extensión no reconocida")
+    if real is None:
+        result.errors.append("Archivo corrupto o ilegible")
         return result
 
-    # 4) Validar contenido
-    file_list = archive.namelist()
+    if ext == ".cbz" and real == "rar":
+        result.warnings.append("Extensión incorrecta: debería ser .cbr")
 
-    for f in file_list:
-        name = f.lower()
-        _, fext = os.path.splitext(name)
-
-        if fext in IMAGE_EXTENSIONS:
-            result.images.append(f)
-
-        if os.path.basename(name) in TRASH_FILES:
-            result.add_warning(f"Archivo basura encontrado: {f}")
-
-    if not result.images:
-        result.add_error("No contiene imágenes válidas")
+    if ext == ".cbr" and real == "zip":
+        result.warnings.append("Extensión incorrecta: debería ser .cbz")
 
     return result