Create core.py

core.py

@@ -0,0 +1,99 @@
+# Versión 1.0: Lógica para Unir, Dividir y Proteger PDFs
+import os
+from pypdf import PdfWriter, PdfReader
+from config import TEMP_DIR
+import uuid
+
+class PDFEngine:
+    """
+    Motor de procesamiento de PDFs.
+    Principio SRP: Solo se encarga de manipular bytes y archivos.
+    """
+
+    @staticmethod
+    def _get_output_path(filename: str) -> str:
+        """Genera una ruta única para evitar colisiones."""
+        unique_name = f"{uuid.uuid4().hex[:8]}_{filename}"
+        return os.path.join(TEMP_DIR, unique_name)
+
+    def merge_pdfs(self, file_paths: list) -> str:
+        """
+        Une múltiples PDFs en uno solo.
+        Args:
+            file_paths: Lista de rutas de archivos temporales.
+        Returns:
+            Ruta del archivo generado.
+        """
+        if not file_paths:
+            raise ValueError("No se han proporcionado archivos.")
+
+        merger = PdfWriter()
+        
+        try:
+            for path in file_paths:
+                merger.append(path)
+            
+            output_path = self._get_output_path("unido.pdf")
+            with open(output_path, "wb") as f:
+                merger.write(f)
+            
+            return output_path
+        except Exception as e:
+            raise RuntimeError(f"Error al unir PDFs: {str(e)}")
+        finally:
+            merger.close()
+
+    def split_pdf(self, file_path: str) -> list:
+        """
+        Divide un PDF en páginas individuales.
+        Nota: Devuelve una lista de archivos.
+        """
+        if not file_path:
+            raise ValueError("Archivo no proporcionado.")
+
+        output_files = []
+        try:
+            reader = PdfReader(file_path)
+            base_name = os.path.basename(file_path).replace(".pdf", "")
+
+            for i, page in enumerate(reader.pages):
+                writer = PdfWriter()
+                writer.add_page(page)
+                
+                out_name = f"{base_name}_pag_{i+1}.pdf"
+                out_path = self._get_output_path(out_name)
+                
+                with open(out_path, "wb") as f:
+                    writer.write(f)
+                
+                output_files.append(out_path)
+            
+            return output_files
+        except Exception as e:
+            raise RuntimeError(f"Error al dividir PDF: {str(e)}")
+
+    def protect_pdf(self, file_path: str, password: str) -> str:
+        """
+        Encripta un PDF con contraseña.
+        """
+        if not file_path or not password:
+            raise ValueError("Faltan el archivo o la contraseña.")
+
+        try:
+            reader = PdfReader(file_path)
+            writer = PdfWriter()
+
+            # Copiar todas las páginas
+            for page in reader.pages:
+                writer.add_page(page)
+
+            # Encriptar
+            writer.encrypt(password)
+
+            output_path = self._get_output_path("protegido.pdf")
+            with open(output_path, "wb") as f:
+                writer.write(f)
+            
+            return output_path
+        except Exception as e:
+            raise RuntimeError(f"Error al proteger PDF: {str(e)}")