csc-cia-stne 0.1.19__tar.gz → 0.1.20__tar.gz

{csc_cia_stne-0.1.19 → csc_cia_stne-0.1.20}/PKG-INFO

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

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "csc_cia_stne"
-version = "0.1.19"
+version = "0.1.20"
 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.19 → csc_cia_stne-0.1.20}/src/csc_cia_stne/bc_sta.py

@@ -6,7 +6,7 @@ import xml.etree.ElementTree as ET
 from xml.dom.minidom import parseString
 from pydantic import BaseModel, ValidationError, field_validator, Field, HttpUrl
 from typing import Literal, Dict, Union, Optional, List
-from datetime import datetime
+from datetime import datetime, timedelta
 
 log = logging.getLogger('__main__')
 
@@ -499,90 +499,95 @@ 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
-        """
+    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):
         resultados = []
-        ultima_data = inicio
+
+        # Parse seguro para datetime (assume formato ISO 'AAAA-MM-DDTHH:MM:SS')
+        dt_inicio = datetime.fromisoformat(inicio)
+        dt_fim = datetime.fromisoformat(fim) if fim else None
+        ultima_dt = dt_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}"
+            # Se já passamos do fim, encerra antes de chamar a API
+            if dt_fim and ultima_dt >= dt_fim:
+                break
+
+            # Monta a URL usando a data atual do cursor
+            _inicio_str = ultima_dt.strftime("%Y-%m-%dT%H:%M:%S")
+            url = f"{self.base_url}/arquivos?tipoConsulta=AVANC&nivelDetalhe={nivel}"
+            url += f"&dataHoraInicio={_inicio_str}"
+            if situacao:
+                url += f"&situacaoTransmissao={situacao}"
+            if identificadorDocumento:
+                url += f"&identificadorDocumento={identificadorDocumento}"
+            if dt_fim:
+                _fim_str = dt_fim.strftime("%Y-%m-%dT%H:%M:%S")
+                url += f"&dataHoraFim={_fim_str}"
+            url += f"&qtdMaxResultados={qtd}"
 
             response = requests.get(
-                url,
-                headers=self.headers,
-                auth=self.auth, 
-                timeout=60,
+                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:
-
-                        if isinstance(dados, list):
-                            dados_filtrados = [arquivo for arquivo in dados if arquivo.get("TipoArquivo") in tipo_arquivo]
-                        elif dados.get("TipoArquivo") in tipo_arquivo:
-                            dados_filtrados = [dados]
-
-                        resultados.extend(dados_filtrados)
-
-                    else:
-
-                        resultados.extend(dados)
+            if response.status_code != 200:
+                log.error(f"Erro ao listar arquivos: response code {response.status_code}\n{response.text}")
+                return False
 
-                    # Verifica se o campo 'DataHoraDisponibilizacao' existe no último registro
-                    if dados and isinstance(dados, list) and 'DataHoraDisponibilizacao' in dados[-1]:
+            try:
+                dados = xml_response_to_json(response.text)
+                if not dados:
+                    break  # sem mais resultados
+
+                # Normaliza dados para lista
+                itens = dados if isinstance(dados, list) else [dados]
+
+                # (opcional) filtra por tipo_arquivo
+                if tipo_arquivo is not None:
+                    itens = [a for a in itens if a.get("TipoArquivo") in tipo_arquivo]
+
+                resultados.extend(itens)
+
+                # Atualiza o cursor a partir do último item retornado
+                # Busca o último com DataHoraDisponibilizacao
+                ultimo = None
+                for cand in reversed(itens):
+                    if "DataHoraDisponibilizacao" in cand and cand["DataHoraDisponibilizacao"]:
+                        ultimo = cand["DataHoraDisponibilizacao"]
+                        break
+
+                if not ultimo:
+                    # fallback: se o JSON bruto vier como dict/list diferente, tente no original
+                    if isinstance(dados, list) and dados and "DataHoraDisponibilizacao" in dados[-1]:
+                        ultimo = dados[-1]["DataHoraDisponibilizacao"]
+                    elif isinstance(dados, dict) and "DataHoraDisponibilizacao" in dados:
+                        ultimo = dados["DataHoraDisponibilizacao"]
+
+                if not ultimo:
+                    log.error("Campo 'DataHoraDisponibilizacao' não encontrado ou estrutura inesperada.")
+                    return False
 
-                        ultima_data = dados[-1]['DataHoraDisponibilizacao']  # Atualiza a data para a próxima requisição
+                # Cursor: último + 1s para evitar repetir o mesmo registro
+                proxima_dt = datetime.fromisoformat(ultimo) + timedelta(seconds=1)
 
-                    elif dados and isinstance(dados, dict) and 'DataHoraDisponibilizacao' in dados:
-                        
-                        ultima_data = dados['DataHoraDisponibilizacao']  # Atualiza a data para a próxima requisição
-                    
-                    else:
-
-                        log.error("Campo 'DataHoraDisponibilizacao' não encontrado ou estrutura inesperada.")
-                        return False
+                # Se a próxima dt já ultrapassa o fim, encerramos sem nova chamada
+                if dt_fim and proxima_dt > dt_fim:
+                    break
 
-                except ET.ParseError as e:
+                ultima_dt = proxima_dt
 
-                    log.error(f"Erro ao processar XML: {e}")
-                    return False
-        
-            else:
-                
-                log.error(f"Erro ao listar arquivos: response code {response.status_code}\n{response.text}")
+            except ET.ParseError as e:
+                log.error(f"Erro ao processar XML: {e}")
                 return False
-        
+
         return resultados
-    
+
     def download_arquivo(self,protocolo:str,filename:str=None):
         """Faz o download de um arquivo de um protocolo especifico
 
@@ -631,7 +636,6 @@ class BC_STA:
         
             return {"success": False, "status_code": int(response.status_code), "content": response.text}
 
-
     def qs_gerar_xml_string_responder_nao_cliente(self,protocolo_inicial_ordem:str,numctrlccs:str,cnpj_cpf:str,numctrlenvio:str,numprocjud:str,identdemissor:str,identddestinatario:str,domsist:str,nuop:str,dtmovto:datetime,cnpjbaseentrespons:str):
         
         """Gera a string do arquivo XML de resposta para nao-cliente para ser enviado ao STA

{csc_cia_stne-0.1.19 → csc_cia_stne-0.1.20}/src/csc_cia_stne.egg-info/PKG-INFO

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