csc-cia-stne 0.0.65__py3-none-any.whl → 0.0.67__py3-none-any.whl

csc_cia_stne/bc_sta.py

@@ -3,8 +3,11 @@ import requests
 import xml.etree.ElementTree as ET
 import hashlib
 from pydantic import BaseModel, ValidationError, field_validator, Field, HttpUrl
-from typing import Literal, Dict, Union, Optional
+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."""
@@ -170,16 +173,26 @@ class ListarArquivosParams(BaseModel):
     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")
@@ -485,52 +498,31 @@ class BC_STA:
             #return f"Failed to create protocol. Status code: {response.status_code}, Reason: {response.reason}"
             return resposta
 
-    def listar_arquivos_old(self, nivel: Literal['RES', 'BAS', 'COMPL'], inicio: str, situacao: Optional[Literal['REC', 'A_REC']] = None, identificadorDocumento: Optional[str] = None, qtd: int = 100):
-        """
-        Lista os arquivos disponíveis na API STA.
+    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 (str): Nível de detalhe da consulta ('RES', 'BAS' ou 'COMPL').
-            inicio (str): Data e hora de início no formato ISO 8601 (yyyy-MM-ddTHH:mm:ss).
-            situacao (Optional[str]): Situação da transmissão ('REC' ou 'A_REC').
-            identificadorDocumento (Optional[str]): Identificador do documento.
-            qtd (int): Quantidade máxima de resultados (padrão: 100, máximo: 100).
-        
+            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:
-            dict: Resposta da API em formato JSON.
+            list: Lista de arquivos do STA
         """
-        params = ListarArquivosParams(nivel=nivel, inicio=inicio, situacao=situacao, identificadorDocumento=identificadorDocumento, qtd=qtd)
-        
-        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"&qtdMaxResultados={params.qtd}"
-
-        response = requests.get(
-            url,
-            headers=self.headers,
-            auth=self.auth, 
-            timeout=60,
-        )
-
-        if response.status_code == 200:
-            return xml_response_to_json(response.text)  # Retorna um dicionário
-
-
-        else:
-            print(f"Erro ao listar arquivos: {response.status_code}")
-            return None
-
-    def listar_arquivos(self, nivel: Literal['RES', 'BAS', 'COMPL'], inicio: str, situacao: Optional[Literal['REC', 'A_REC']] = None, identificadorDocumento: Optional[str] = None, qtd: int = 100):
         resultados = []
         ultima_data = inicio
 
         while True:
-            params = ListarArquivosParams(nivel=nivel, inicio=ultima_data, situacao=situacao, identificadorDocumento=identificadorDocumento, qtd=qtd)
+            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(
@@ -541,27 +533,45 @@ class BC_STA:
             )
 
             if response.status_code == 200:
+        
                 try:
+        
                     dados = xml_response_to_json(response.text)
-                    #print(dados)
-                    #print(type(dados))
 
                     if not dados:  # Verifica se a lista está vazia
-                        #print("lista ta vaziaaaa")
+        
                         break  # Sai do loop se não houver mais dados
-                    resultados.extend(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:
-                        print("Campo 'DataHoraDisponibilizacao' não encontrado ou estrutura inesperada.")
-                        break
+
+                        log.error("Campo 'DataHoraDisponibilizacao' não encontrado ou estrutura inesperada.")
+                        return False
+
                 except ET.ParseError as e:
-                    print(f"Erro ao processar XML: {e}")
-                    break
+
+                    log.error(f"Erro ao processar XML: {e}")
+                    return False
+        
             else:
-                print(f"Erro ao listar arquivos: {response.status_code}")
-                break
+                
+                log.error(f"Erro ao listar arquivos: {response.status_code}")
+                return False
+        
         return resultados
     
     def download_arquivo(self,protocolo:str,filename:str=None):

{csc_cia_stne-0.0.65.dist-info → csc_cia_stne-0.0.67.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: csc_cia_stne
-Version: 0.0.65
+Version: 0.0.67
 Summary: Biblioteca do time CSC-CIA utilizada no desenvolvimento de RPAs
 License: MIT
 Keywords: karavela,csc,cia,stone,rpa,botcity,stne
@@ -27,6 +27,7 @@ Requires-Dist: botcity-framework-web
 Requires-Dist: email-validator
 Requires-Dist: botcity-maestro-sdk
 Requires-Dist: psutil
+Requires-Dist: cryptography
 Requires-Dist: pyautogui
 
 Essa biblioteca é desenvolvida e atualizada pelo time **CSC-CIA** da **Stone**

{csc_cia_stne-0.0.65.dist-info → csc_cia_stne-0.0.67.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=bVj7o2oOrAR3V_fSD9ex0CGahLc1U6EgeYetQGUw1-0,25282
+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.65.dist-info/LICENCE,sha256=LPGMtgKki2C3KEZP7hDhA1HBrlq5JCHkIeStUCLEMx4,1073
-csc_cia_stne-0.0.65.dist-info/METADATA,sha256=hq9XBuN_o_bOOi8jaqlctvRBd6Ej8MZ3sZ3gXOJiELQ,1312
-csc_cia_stne-0.0.65.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-csc_cia_stne-0.0.65.dist-info/top_level.txt,sha256=ldo7GVv3tQx5KJvwBzdZzzQmjPys2NDVVn1rv0BOF2Q,13
-csc_cia_stne-0.0.65.dist-info/RECORD,,
+csc_cia_stne-0.0.67.dist-info/LICENCE,sha256=LPGMtgKki2C3KEZP7hDhA1HBrlq5JCHkIeStUCLEMx4,1073
+csc_cia_stne-0.0.67.dist-info/METADATA,sha256=5yKmOI6aa-7_32-o8vylgl_9mvx4UttDvEERbSw_v04,1340
+csc_cia_stne-0.0.67.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+csc_cia_stne-0.0.67.dist-info/top_level.txt,sha256=ldo7GVv3tQx5KJvwBzdZzzQmjPys2NDVVn1rv0BOF2Q,13
+csc_cia_stne-0.0.67.dist-info/RECORD,,