csc-cia-stne 0.1.34__tar.gz → 0.1.35__tar.gz

{csc_cia_stne-0.1.34 → csc_cia_stne-0.1.35}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: csc_cia_stne
-Version: 0.1.34
+Version: 0.1.35
 Summary: Biblioteca do time CSC-CIA utilizada no desenvolvimento de RPAs
 License: MIT
 Keywords: karavela,csc,cia,stone,rpa,botcity,stne

{csc_cia_stne-0.1.34 → csc_cia_stne-0.1.35}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "csc_cia_stne"
-version = "0.1.34"
+version = "0.1.35"
 license = { text = "MIT" }
 description = "Biblioteca do time CSC-CIA utilizada no desenvolvimento de RPAs"
 keywords = ["karavela", "csc", "cia", "stone", "rpa", "botcity", "stne"]

{csc_cia_stne-0.1.34 → csc_cia_stne-0.1.35}/src/csc_cia_stne/utilitarios/functions/func_pdf_extract.py

@@ -1,29 +1,39 @@
 from PyPDF2 import PdfReader, PdfWriter
 from io import BytesIO
-from typing import Optional
+from typing import Optional, Dict, Any
 
-def extrair_x_paginas_pdf(file_path: str, pages_limit: int = 15) -> Optional[bytes]:
+def extrair_x_paginas_pdf(file_path: str, pages_limit: int = 15) -> Dict[str, Any]:
     """
     Extrai as primeiras X páginas de um arquivo PDF e retorna os bytes dessas páginas.
 
     Args:
         file_path (str): Caminho completo do arquivo PDF original.
         pages_limit (int, optional): Número máximo de páginas a serem extraídas. 
-                                    Defaults to 15.
+                                    Padrão: 15.
 
     Returns:
-        Optional[bytes]: Bytes do novo PDF contendo as primeiras X páginas,
-                        ou None em caso de erro.
+        Dict[str, Any]: Dicionário contendo:
+            - success (bool): True se a operação foi bem-sucedida, False caso contrário
+            - error (str|None): Mensagem de erro se houver falha, None se sucesso
+            - data (bytes|None): Bytes do PDF extraído se sucesso, None se erro
+            - total_doc_pages (int): Número total de páginas do documento original (apenas se success=True)
+            - extracted_pages (int): Número de páginas efetivamente extraídas (apenas se success=True)
     
     Note:
         - Se o PDF tiver menos páginas que pages_limit, todas serão extraídas
         - Utiliza PyPDF2 para manipulação do arquivo PDF
-        - Retorna None em caso de erro no processamento
+        - Retorna success=False em caso de erro no processamento
     
     Example:
-        >>> pdf_bytes = extrair_x_paginas_pdf("documento.pdf", 10)
-        >>> if pdf_bytes["success"]:
-        ...     print(f"PDF extraído com {len(pdf_bytes["data"])} bytes")
+        >>> resultado = extrair_x_paginas_pdf("documento.pdf", 10)
+        >>> if resultado["success"]:
+        ...     print(f"PDF extraído com {len(resultado['data'])} bytes")
+        ...     print(f"Páginas extraídas: {resultado['extracted_pages']}/{resultado['total_doc_pages']}")
+        ... else:
+        ...     print(f"Erro: {resultado['error']}")
+    
+    Raises:
+        Não levanta exceções diretamente - erros são capturados e retornados no dicionário.
     """
     try:
         # Lê o arquivo PDF original
@@ -31,20 +41,21 @@ def extrair_x_paginas_pdf(file_path: str, pages_limit: int = 15) -> Optional[byt
         writer = PdfWriter()
 
         # Extrai as primeiras 'pages_limit' páginas ou menos, caso o PDF tenha menos de 'pages_limit' páginas
-        for page_num in range(min(pages_limit, len(reader.pages))):
+        total_doc_pages = len(reader.pages)
+        for page_num in range(min(pages_limit, total_doc_pages)):
             writer.add_page(reader.pages[page_num])
 
         # Salva o novo PDF em um objeto BytesIO
         pdf_bytes = BytesIO()
         writer.write(pdf_bytes)
         pdf_bytes.seek(0)  # Move o cursor para o início do buffer
-        resposta = {"success": True, "error": None, "data": pdf_bytes.read()}
+        resposta = {"success": True, "error": None, "data": pdf_bytes.read(), "total_doc_pages": total_doc_pages, "extracted_pages": min(pages_limit, total_doc_pages)}
         return resposta
     except Exception as e:
         resposta = {"success": False, "error": f"Erro ao extrair as primeiras {pages_limit} páginas do PDF: {str(e)}", "data": None}
         return resposta
 
-def extrair_paginas_intervalo_pdf(file_path: str, page_start: int = 1, pages_limit: int = 15) -> Optional[bytes]:
+def extrair_paginas_intervalo_pdf(file_path: str, page_start: int = 1, pages_limit: int = 15) -> Dict[str, Any]:
     """
     Extrai um número específico de páginas de um arquivo PDF a partir de uma página inicial.
 

{csc_cia_stne-0.1.34 → csc_cia_stne-0.1.35}/src/csc_cia_stne.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: csc_cia_stne
-Version: 0.1.34
+Version: 0.1.35
 Summary: Biblioteca do time CSC-CIA utilizada no desenvolvimento de RPAs
 License: MIT
 Keywords: karavela,csc,cia,stone,rpa,botcity,stne