csc-cia-stne 0.0.64__py3-none-any.whl → 0.0.66__py3-none-any.whl

csc_cia_stne/bc_sta.py

@@ -2,8 +2,55 @@ from requests.auth import HTTPBasicAuth
 import requests
 import xml.etree.ElementTree as ET
 import hashlib
-from pydantic import BaseModel, ValidationError, field_validator
-from typing import Literal, Dict, Union, Optional
+from pydantic import BaseModel, ValidationError, field_validator, Field, HttpUrl
+from typing import Literal, Dict, Union, Optional, List
+import json
+import logging
+
+log = logging.getLogger('__main__')
+
+def xml_to_dict(element):
+    """Converte um elemento XML recursivamente para um dicionário."""
+    if not list(element):  # Se não tem filhos, retorna apenas o texto
+        return element.text.strip() if element.text else ""
+
+    result = {}
+    for child in element:
+        child_data = xml_to_dict(child)
+        if child.tag in result:
+            if isinstance(result[child.tag], list):
+                result[child.tag].append(child_data)
+            else:
+                result[child.tag] = [result[child.tag], child_data]
+        else:
+            result[child.tag] = child_data
+
+    if element.attrib:
+        result["@atributos"] = element.attrib  # Adiciona atributos XML se existirem
+
+    return result
+
+def xml_response_to_json(response_text):
+    """Converte a resposta XML para um dicionário JSON válido."""
+
+    root = ET.fromstring(response_text)
+    lista = xml_to_dict(root)
+    if not isinstance(lista, dict):
+        return []
+    return list(lista.values())[0]  # Agora retorna um dicionário em vez de uma string JSON
+
+def print_element(element, indent=0):
+    """Função recursiva para exibir campos e subcampos"""
+    prefix = " " * (indent * 2)  # Indentação para visualização hierárquica
+    print(f"{prefix}- {element.tag}: {element.text.strip() if element.text else ''}")
+    
+    # Se o elemento tiver atributos, exibir
+    if element.attrib:
+        print(f"{prefix}  Atributos: {element.attrib}")
+    
+    # Percorrer subelementos
+    for child in element:
+        print_element(child, indent + 1)
 
 # Validações dos inputs
 class InitParamsValidator(BaseModel):
@@ -118,7 +165,44 @@ class EnviarArquivoValidator(BaseModel):
                     raise ValueError("Cada destinatário deve conter as chaves 'unidade', 'dependencia' e 'operador'. Verifique a documentação da API BC STA para entender o que colocar cada campo")
         
         return value
-    
+
+class ListarArquivosParams(BaseModel):
+    """
+    Parâmetros para listar arquivos disponíveis na API STA.
+
+    Atributos:
+        nivel (str): Nível de detalhe da consulta. Aceita apenas 'RES', 'BAS' ou 'COMPL'.
+        inicio (str): Data e hora de início no formato ISO 8601 (yyyy-MM-ddTHH:mm:ss).
+        fim (str): Data e hora de fim no formato ISO 8601 (yyyy-MM-ddTHH:mm:ss).
+        situacao (Optional[str]): Situação da transmissão, podendo ser 'REC' ou 'A_REC'.
+        identificadorDocumento (Optional[str]): Identificador do documento, se aplicável.
+        qtd (int): Quantidade máxima de resultados (valor padrão: 100, máximo permitido: 100).
+        tipo_arquivo (list): lista de tipos de arquivo para filtrar ['ACCS002','ACCS003','AJUD301','AJUD302','AJUD303','AJUD304','AJUD305','AJUD308','AJUD309','AJUD310','AJUD331','AMES102','AMTF102','ASVR9810','ATXB001']
+    """
+    nivel: Literal['RES', 'BAS', 'COMPL']
+    #inicio: str = Field(..., regex=r"^\d{4}-\d{2}-\d{2}T\d{2}:\d{2}:\d{2}$")
+    inicio: str = Field(..., pattern=r"^\d{4}-\d{2}-\d{2}T\d{2}:\d{2}:\d{2}$")
+    fim: Optional[str] = Field(None, pattern=r"^\d{4}-\d{2}-\d{2}T\d{2}:\d{2}:\d{2}$")
+    situacao: Optional[Literal['REC', 'A_REC']] = None
+    identificadorDocumento: Optional[str] = None
+    qtd: int = Field(default=100, le=100)
+    #tipo_arquivo: Optional[str] = None
+    tipo_arquivo: Optional[List[
+        Literal[
+            'ACCS002', 'ACCS003', 'AJUD301', 'AJUD302', 'AJUD303', 'AJUD304', 'AJUD305', 'AJUD308',
+            'AJUD309', 'AJUD310', 'AJUD331', 'AMES102', 'AMTF102', 'ASVR9810', 'ATXB001'
+        ]
+    ]] = None
+
+class DownloadArquivoParams(BaseModel):
+    protocolo: str = Field(..., min_length=1, description="Código do protocolo")
+    filename: Optional[str] = Field(None, description="Nome e caminho do arquivo")
+
+class DownloadArquivoResponse(BaseModel):
+    success: bool = Field(..., description="Indica se o download foi bem-sucedido")
+    status_code: int = Field(..., ge=100, le=599, description="Código de status HTTP")
+    content: Union[bytes, str] = Field(..., description="Conteúdo do arquivo (em bytes se sucesso, em string se erro)")
+
 class BC_STA:
     
     def __init__(self, usuario:str, senha:str, ambiente:str):
@@ -414,3 +498,128 @@ class BC_STA:
             #return f"Failed to create protocol. Status code: {response.status_code}, Reason: {response.reason}"
             return resposta
 
+    def listar_arquivos(self, nivel: Literal['RES', 'BAS', 'COMPL'], inicio: str, fim: str = None,situacao: Optional[Literal['REC', 'A_REC']] = None, identificadorDocumento: Optional[str] = None, qtd: int = 100, tipo_arquivo:list=None):
+        """Lista os arquivos do STA com os parâmetros informados.
+
+        Args:
+            nivel (Literal['RES', 'BAS', 'COMPL']): nivel de detalhamento da consulta
+            inicio (str): data de inicio, no padrao 'AAAA-MM-DDTHH:MM:SS'
+            fim (str, optional): data de fim da consulta, no padrao 'AAAA-MM-DDTHH:MM:SS'
+            situacao (Optional[Literal['REC', 'A_REC']], optional): Arquivos já recebidos ou ainda não recebidos. Default=None (Todos).
+            identificadorDocumento (Optional[str], optional): Identificador documento. Default=None (Todos).
+            qtd (int, optional): Quantidade de arquivos na resposta. Default=100 (máximo=100).
+            tipo_arquivo (list, optional): Tipo de arquivo para fazer download. Default=None (Todos). Opções: ['ACCS002', 'ACCS003', 'AJUD301', 'AJUD302', 'AJUD303', 'AJUD304', 'AJUD305', 'AJUD308', 'AJUD309', 'AJUD310', 'AJUD331', 'AMES102', 'AMTF102', 'ASVR9810', 'ATXB001'].
+
+        Returns:
+            list: Lista de arquivos do STA
+        """
+        resultados = []
+        ultima_data = inicio
+
+        while True:
+            params = ListarArquivosParams(nivel=nivel, inicio=ultima_data, fim=fim, situacao=situacao, identificadorDocumento=identificadorDocumento, qtd=qtd, tipo_arquivo=tipo_arquivo)
+            
+            url = f"{self.base_url}/arquivos?tipoConsulta=AVANC&nivelDetalhe={params.nivel}"
+            url += f"&dataHoraInicio={params.inicio}&situacaoTransmissao={params.situacao}" if params.situacao else ""
+            url += f"&identificadorDocumento={params.identificadorDocumento}" if params.identificadorDocumento else ""
+            url += f"&dataHoraFim={params.fim}" if params.fim else ""
+            url += f"&qtdMaxResultados={params.qtd}"
+
+            response = requests.get(
+                url,
+                headers=self.headers,
+                auth=self.auth, 
+                timeout=60,
+            )
+
+            if response.status_code == 200:
+        
+                try:
+        
+                    dados = xml_response_to_json(response.text)
+
+                    if not dados:  # Verifica se a lista está vazia
+        
+                        break  # Sai do loop se não houver mais dados
+                    
+                    # Filtra apenas os arquivos do tipo 'AJUD308'
+                    if tipo_arquivo is not None:
+
+                        dados_filtrados = [arquivo for arquivo in dados if arquivo.get("TipoArquivo") in tipo_arquivo]
+                        resultados.extend(dados_filtrados)
+
+                    else:
+
+                        resultados.extend(dados)
+
+                    # Verifica se o campo 'DataHoraDisponibilizacao' existe no último registro
+                    if dados and isinstance(dados, list) and 'DataHoraDisponibilizacao' in dados[-1]:
+
+                        ultima_data = dados[-1]['DataHoraDisponibilizacao']  # Atualiza a data para a próxima requisição
+
+                    else:
+
+                        log.error("Campo 'DataHoraDisponibilizacao' não encontrado ou estrutura inesperada.")
+                        return False
+
+                except ET.ParseError as e:
+
+                    log.error(f"Erro ao processar XML: {e}")
+                    return False
+        
+            else:
+                
+                log.error(f"Erro ao listar arquivos: {response.status_code}")
+                return False
+        
+        return resultados
+    
+    def download_arquivo(self,protocolo:str,filename:str=None):
+        """Faz o download de um arquivo de um protocolo especifico
+
+        Args:
+            protocolo (str): protocolo
+            filename (str): path+nome do arquivo
+
+        Returns:
+            dict: {"success": bool, "status_code": int, "content": bytes/str}
+        """
+        
+        # Validação dos parâmetros
+        try:
+        
+            params = DownloadArquivoParams(protocolo=protocolo, filename=filename)
+        
+        except ValidationError as e:
+        
+            return Exception(str(e))
+        
+        url = f"/arquivos/{protocolo}/conteudo"
+        response = requests.get(
+            self.base_url + url,
+            auth=self.auth, 
+            timeout=60,
+            headers={"Connection": "keep-alive"},
+        )
+
+        if response.status_code == 200:
+            
+            if filename is not None:
+                
+                try:
+                
+                    with open(filename, "wb") as arquivo:
+                
+                        arquivo.write(response.content)
+                
+                except Exception as e:
+                
+                    raise Exception(f"Falha ao salvar o arquivo em disco\n{str(e)}")
+                
+            return {"success": True, "status_code": int(response.status_code), "content": response.content }
+        
+        else:
+        
+            return {"success": False, "status_code": int(response.status_code), "content": response.text}
+
+

{csc_cia_stne-0.0.64.dist-info → csc_cia_stne-0.0.66.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: csc_cia_stne
-Version: 0.0.64
+Version: 0.0.66
 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.0.64.dist-info → csc_cia_stne-0.0.66.dist-info}/RECORD

@@ -1,6 +1,6 @@
 csc_cia_stne/__init__.py,sha256=Io-gKis1evws5cHUqyOrcsZKNCQRviYj3zbp__5lgKU,2512
 csc_cia_stne/bc_correios.py,sha256=pQAnRrcXEMrx3N1MWydZVIhEQLerh3x8-0B045zZIzk,24174
-csc_cia_stne/bc_sta.py,sha256=f75HJ7FLIDSJFLDTvvSvCYo9z0HchzP7rDY5WIdiKXY,16830
+csc_cia_stne/bc_sta.py,sha256=KWHTytM3msAFayt8y5_97RwWHE8Dv8ssfwyh6zU7xI0,25836
 csc_cia_stne/email.py,sha256=RK_TzWBVnUfpP-s5NvjTJJjzhICy8e2fME9EuaiySMY,8162
 csc_cia_stne/gcp_bigquery.py,sha256=jYxvqrWDOPkxc05U4aef7V5lL8ptqsE93lfn0dLFyvc,7385
 csc_cia_stne/google_drive.py,sha256=lgcOd27vk2Mb_wP_fAWIbec-S3MIBKyh4TpRth6REXc,12788
@@ -26,8 +26,8 @@ csc_cia_stne/utilitarios/validations/GoogleDriveValidator.py,sha256=PBo-AV2bjR__
 csc_cia_stne/utilitarios/validations/ServiceNowValidator.py,sha256=yleKUIo1ZfyloP9fDPNjv3JJXdLcocT81WIgRSYmqEA,14423
 csc_cia_stne/utilitarios/validations/__init__.py,sha256=O_qyEU2ji3u6LHUXZCXvUFsMpoMWL625qqHTXyXivTA,106
 csc_cia_stne/utilitarios/validations/web_validator.py,sha256=HYKYSpDv1RvRjZIuwTPt-AbEz-9392MxM_O329iYuSA,5722
-csc_cia_stne-0.0.64.dist-info/LICENCE,sha256=LPGMtgKki2C3KEZP7hDhA1HBrlq5JCHkIeStUCLEMx4,1073
-csc_cia_stne-0.0.64.dist-info/METADATA,sha256=KYbJwXirJeL6A39Cz8ohmuBV8BD7y_M2E4bCVs3yBBY,1312
-csc_cia_stne-0.0.64.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-csc_cia_stne-0.0.64.dist-info/top_level.txt,sha256=ldo7GVv3tQx5KJvwBzdZzzQmjPys2NDVVn1rv0BOF2Q,13
-csc_cia_stne-0.0.64.dist-info/RECORD,,
+csc_cia_stne-0.0.66.dist-info/LICENCE,sha256=LPGMtgKki2C3KEZP7hDhA1HBrlq5JCHkIeStUCLEMx4,1073
+csc_cia_stne-0.0.66.dist-info/METADATA,sha256=xTDZdZe5LPIZRHdvcJ3Vhk0C6KLrIXMopa7JXmSdXaY,1312
+csc_cia_stne-0.0.66.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+csc_cia_stne-0.0.66.dist-info/top_level.txt,sha256=ldo7GVv3tQx5KJvwBzdZzzQmjPys2NDVVn1rv0BOF2Q,13
+csc_cia_stne-0.0.66.dist-info/RECORD,,