csc-cia-stne 0.1.19__py3-none-any.whl → 0.1.21__py3-none-any.whl

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,99 @@ 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
 
-        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,
-            )
+        try:
+            # 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:
+                # 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,
+                )
+
+                if response.status_code != 200:
+                    log.error(f"Erro ao listar arquivos: response code {response.status_code}\n{response.text}")
+                    return False
 
-            if response.status_code == 200:
-        
                 try:
-        
                     dados = xml_response_to_json(response.text)
+                    if not dados:
+                        break  # sem mais resultados
 
-                    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)
+                    # Normaliza dados para lista
+                    itens = dados if isinstance(dados, list) else [dados]
 
-                    # Verifica se o campo 'DataHoraDisponibilizacao' existe no último registro
-                    if dados and isinstance(dados, list) and 'DataHoraDisponibilizacao' in dados[-1]:
+                    # (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:
+                    # Se a próxima dt já ultrapassa o fim, encerramos sem nova chamada
+                    if dt_fim and proxima_dt > dt_fim:
+                        break
 
-                        log.error("Campo 'DataHoraDisponibilizacao' não encontrado ou estrutura inesperada.")
-                        return False
+                    ultima_dt = proxima_dt
 
                 except ET.ParseError as e:
-
                     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}")
-                return False
-        
-        return resultados
-    
+
+            return resultados
+        except Exception as e:
+            log.error(f"Erro em BC_STA:listar_arquivos: {e}")
+            return False
+
     def download_arquivo(self,protocolo:str,filename:str=None):
         """Faz o download de um arquivo de um protocolo especifico
 
@@ -631,7 +640,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.dist-info → csc_cia_stne-0.1.21.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: csc_cia_stne
-Version: 0.1.19
+Version: 0.1.21
 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.dist-info → csc_cia_stne-0.1.21.dist-info}/RECORD

@@ -1,6 +1,6 @@
 csc_cia_stne/__init__.py,sha256=jwLhGpOwFCow_6cqzwLn31WcIrMzutMZtEQpLL4bQtM,2638
 csc_cia_stne/bc_correios.py,sha256=s2XjJ0iokMlcUv0mAy9saU3pc_G-6X8wltb_lFHIL6o,24717
-csc_cia_stne/bc_sta.py,sha256=sE-aU-ZVSAqryQrT1-nor9eAFM5npNAKF1QSm-ChhGU,28945
+csc_cia_stne/bc_sta.py,sha256=U3Z9LZ7-_LZXDx5DAuvIHwHIkwGpBLSt9VJfonWcJTk,28990
 csc_cia_stne/email.py,sha256=y4xyPAe6_Mga5Wf6qAsDzYgn0f-zf2KshfItlWe58z8,8481
 csc_cia_stne/ftp.py,sha256=eNkOUEXdw-9NYfuZjNo6Oh7EduD54g8N0cfD0LuOiTU,11516
 csc_cia_stne/gcp_bigquery.py,sha256=foq8azvvv_f7uikMDslX9RcUIrx7RAS-Sn0AGW0QFQc,7231
@@ -38,8 +38,8 @@ csc_cia_stne/utilitarios/web_screen/__init__.py,sha256=5QcOPXKd95SvP2DoZiHS0gaU6
 csc_cia_stne/utilitarios/web_screen/web_screen_abstract.py,sha256=PjL8Vgfj_JdKidia7RFyCkro3avYLQu4RZRos41sh3w,3241
 csc_cia_stne/utilitarios/web_screen/web_screen_botcity.py,sha256=Xi5YJjl2pcxlX3OimqcBWRNXZEpAE7asyUjDJ4Oho5U,12297
 csc_cia_stne/utilitarios/web_screen/web_screen_selenium.py,sha256=JLIcPJE9ZX3Pd6zG6oTRMqqUAY063UzLY3ReRlxmiSM,15581
-csc_cia_stne-0.1.19.dist-info/licenses/LICENCE,sha256=LPGMtgKki2C3KEZP7hDhA1HBrlq5JCHkIeStUCLEMx4,1073
-csc_cia_stne-0.1.19.dist-info/METADATA,sha256=uqMxU4KDuh5YLlzkD7gI2oPgDIW00CfZSFR7QXGp3cI,1464
-csc_cia_stne-0.1.19.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-csc_cia_stne-0.1.19.dist-info/top_level.txt,sha256=ldo7GVv3tQx5KJvwBzdZzzQmjPys2NDVVn1rv0BOF2Q,13
-csc_cia_stne-0.1.19.dist-info/RECORD,,
+csc_cia_stne-0.1.21.dist-info/licenses/LICENCE,sha256=LPGMtgKki2C3KEZP7hDhA1HBrlq5JCHkIeStUCLEMx4,1073
+csc_cia_stne-0.1.21.dist-info/METADATA,sha256=obIzLQ4Cf6gGibFVcQgW1OlTuIcSsg7_d4Hw-s2s0a0,1464
+csc_cia_stne-0.1.21.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+csc_cia_stne-0.1.21.dist-info/top_level.txt,sha256=ldo7GVv3tQx5KJvwBzdZzzQmjPys2NDVVn1rv0BOF2Q,13
+csc_cia_stne-0.1.21.dist-info/RECORD,,