csc-cia-stne 0.1.24__tar.gz → 0.1.26__tar.gz

{csc_cia_stne-0.1.24 → csc_cia_stne-0.1.26}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: csc_cia_stne
-Version: 0.1.24
+Version: 0.1.26
 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.24 → csc_cia_stne-0.1.26}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "csc_cia_stne"
-version = "0.1.24"
+version = "0.1.26"
 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.24 → csc_cia_stne-0.1.26}/src/csc_cia_stne/stne_admin.py

@@ -1,14 +1,9 @@
+import re
 import requests
 import jwt
 from datetime import datetime, timedelta
 import time
-#import subprocess
 import json
-#from modules.pdf import gerar_pdf_extrato
-#import csv
-#import io
-#import sys
-#import logging
 from pydantic import BaseModel, StrictStr, StrictInt, ValidationError, field_validator, FieldValidationInfo
 from typing import Literal
 
@@ -217,7 +212,6 @@ class StoneAdmin:
 
             raise Exception(f"Falha ao logar no Stone Admin:\n{response.json()}")        
 
-
     def renew_authorization(self):
         """
         Renova a autorização do usuário.
@@ -419,7 +413,7 @@ class StoneAdmin:
         response = requests.get(f"{self.base_url}/accounts/{account_id}", headers=self.authenticated_header)
         return response
 
-    def detalhar_titular_cpf(self,documento:str):
+    def __detalhar_titular_cpf(self,documento:str):
         """
         Detalha o titular do CPF fornecido.
         Args:
@@ -438,20 +432,31 @@ class StoneAdmin:
             DocumentoValidator(documento=documento)
         
         except ValidationError as e:
-        
-            raise ValueError("Erro na validação dos dados de input do método:", e.errors())
+
+            return {"success": False, "status_code": 400, "error": f"Erro na validação dos dados de input do método: {e.errors()}"}
 
         # Detalha o titular
         
         # Verificar na rota /users (CPF)
+        documento = re.sub(r"\D", "", documento)
         filtro = {'document': documento}
         param = {
-            'filter': json.dumps(filtro)  # Transforma o dicionário em uma string JSON
+            'filter': json.dumps(filtro),  # Transforma o dicionário em uma string JSON
+            'sort': json.dumps(['created_at','DESC'])
         }
-        response = requests.get(f"{self.base_url}/users", params=param, headers=self.authenticated_header)
-        return response
+        try:
+            response = requests.get(f"{self.base_url}/users", params=param, headers=self.authenticated_header)
+            if response.status_code == 401 and 'unauthenticated' in str(response.json()):
+                self.renew_authorization()
+                return self.__detalhar_titular_cpf(documento=documento)
+            elif response.status_code != 200:
+                return {"success": False, "status_code": response.status_code, "error": f"{response.text}"}
+            else:
+                return {"success": True, "status_code": response.status_code, "data": response.json()}
+        except Exception as e:
+            return {"success": False, "status_code": 500, "error": str(e)}
 
-    def detalhar_titular_cnpj(self,documento:str):
+    def __detalhar_titular_cnpj(self,documento:str):
         """
         Detalha o titular de um CNPJ.
         Args:
@@ -470,16 +475,36 @@ class StoneAdmin:
             DocumentoValidator(documento=documento)
         
         except ValidationError as e:
-        
-            raise ValueError("Erro na validação dos dados de input do método:", e.errors())
+            return {"success": False, "status_code": 400, "error": f"Erro na validação dos dados de input do método: {e.errors()}"}
 
         # Verificar na rota /organizations (CNPJ)
+        documento = re.sub(r"\D", "", documento)
         filtro = {'document': documento}
+        
         param = {
-            'filter': json.dumps(filtro)  # Transforma o dicionário em uma string JSON
+            'filter': json.dumps(filtro),  # Transforma o dicionário em uma string JSON
+            'sort': json.dumps(['created_at','DESC'])
         }
-        response = requests.get(f"{self.base_url}/organizations", params=param, headers=self.authenticated_header)
-        return response
+        try:
+            response = requests.get(f"{self.base_url}/organizations", params=param, headers=self.authenticated_header)
+            if response.status_code == 401 and 'unauthenticated' in str(response.json()):
+                self.renew_authorization()
+                return self.__detalhar_titular_cnpj(documento=documento)
+            elif response.status_code != 200:
+                return {"success": False, "status_code": response.status_code, "error": f"{response.text}"}
+            else:
+                return {"success": True, "status_code": response.status_code, "data": response.json()}
+        except Exception as e:
+            return {"success": False, "status_code": 500, "error": str(e)}
+
+    def detalhar_titular(self,tipo_documento:str,documento:str):
+        documento = re.sub(r"\D", "", documento)
+        if str(tipo_documento).strip().upper() == 'F':
+            return self.__detalhar_titular_cpf(documento=documento)
+        elif str(tipo_documento).strip().upper() == 'J':
+            return self.__detalhar_titular_cnpj(documento=documento)
+        else:
+            return {"success": False, "error": "Tipo de documento inválido. Use 'F' para CPF ou 'J' para CNPJ."}
 
     def extrair_extrato(self,account_id:str,data_inicio:datetime,data_fim:datetime,async_mode:bool=False):
         """
@@ -527,12 +552,12 @@ class StoneAdmin:
         
         except ValidationError as e:
         
-            raise ValueError("Erro na validação dos dados de input do método:", e.errors())
+            return {"success": False, "status_code": 400, "error": f"Erro na validação dos dados de input do método: {e.errors()}"}
         
         # Validação do async_mode
         if not isinstance(async_mode, bool):
-        
-            raise ValueError("async_mode deve ser um valor booleano.")
+
+            return {"success": False, "status_code": 400, "error": "async_mode deve ser um valor booleano."}
 
 
         data_inicio = data_inicio.strftime('%Y-%m-%d')
@@ -612,4 +637,68 @@ class StoneAdmin:
         
         except Exception as e:
             
-            return {'result': False, 'status_code': None, 'error': e, 'pdf_content': None}
+            return {'result': False, 'status_code': None, 'error': e, 'pdf_content': None}
+    
+    def legal_entities(self, documento: str) -> dict:
+        """
+        Consulta informações de entidades legais por documento.
+        
+        Args:
+            documento (str): Número do documento (CPF ou CNPJ) para consulta.
+                            Aceita documento com ou sem formatação.
+        
+        Returns:
+            dict: Dicionário contendo o resultado da consulta:
+                - success (bool): Indica se a operação foi bem-sucedida.
+                - data (list): Lista de entidades encontradas (se success=True).
+                - error (str): Mensagem de erro (se success=False).
+        
+        Raises:
+            ValueError: Se ocorrer erro na validação dos dados de entrada.
+        
+        Example:
+            >>> admin = StoneAdmin(client_id, user_agent, private_key, ambiente)
+            >>> resultado = admin.legal_entities("12345678901234")
+            >>> if resultado["success"]:
+            ...     print(f"Encontradas {len(resultado['data'])} entidades")
+            ...     for entidade in resultado["data"]:
+            ...         print(entidade)
+            ... else:
+            ...     print(f"Erro: {resultado['error']}")
+        """
+        
+        try:
+            # Validação usando a classe existente
+            DocumentoValidator(documento=documento)
+        
+        except ValidationError as e:
+            return {"success": False, "status_code": 400, "error": f"Erro na validação dos dados de input do método: {e.errors()}"}
+
+        # Remove formatação do documento
+        documento_limpo = re.sub(r"\D", "", documento)
+        
+        filtro = {'main_document_value': documento_limpo}
+        param = {
+            'filter': json.dumps(filtro)
+        }
+        
+        try:
+            response = requests.get(
+                f"{self.base_url}/legal-entities", 
+                params=param, 
+                headers=self.authenticated_header,
+                timeout=120
+            )
+            
+            # Verifica se houve erro de autenticação
+            if response.status_code == 401 and 'unauthenticated' in str(response.json()):
+                self.renew_authorization()
+                return self.legal_entities(documento=documento)
+            
+            if response.status_code == 200:
+                return {"success": True, "status_code": 200, "data": response.json()}
+            else:
+                return {"success": False, "status_code": response.status_code, "error": f"Status {response.status_code}: {response.text}"}
+
+        except Exception as e:
+            return {"success": False, "error": str(e)}

{csc_cia_stne-0.1.24 → csc_cia_stne-0.1.26}/src/csc_cia_stne.egg-info/PKG-INFO

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