csc-cia-stne 0.1.18__py3-none-any.whl → 0.1.20__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,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/ftp.py

@@ -8,7 +8,7 @@ from .utilitarios.validations.ftp import InitParamsValidator,UploadDownloadValid
 
 class FTP():
 
-    def __init__(self, host:str , user:str , password:str , timeout:int=30, port:int=21, tls:bool=True, ssl:bool=True, verify_ssl:bool=False, verify_host:bool=False, tryouts:int=3, sftp:bool=False):
+    def __init__(self, host:str , user:str , password:str , timeout:int=30, port:int=21, tls:bool=True, ssl:bool=True, verify_ssl:bool=False, verify_host:bool=False, tryouts:int=3, sftp:bool=False, ssh_password:bool=False):
 
         self.host = host
         self.user = user
@@ -21,6 +21,7 @@ class FTP():
         self.verify_host = verify_host
         self.tryouts = tryouts
         self.sftp = sftp
+        self.ssh_password = ssh_password
 
         try:
 
@@ -95,6 +96,10 @@ class FTP():
 
             connection_ftp.setopt(connection_ftp.USERPWD, f"{self.user}:{self.password}".encode("utf-8"))
 
+            if self.ssh_password:
+
+                connection_ftp.setopt(connection_ftp.SSH_AUTH_TYPES, connection_ftp.SSH_AUTH_PASSWORD)
+
             connection_ftp.setopt(connection_ftp.CONNECTTIMEOUT, self.timeout)  # Tempo limite para conexão
                     
             connection_ftp.setopt(connection_ftp.TIMEOUT, self.timeout*2)  # Tempo limite para operação

{csc_cia_stne-0.1.18.dist-info → csc_cia_stne-0.1.20.dist-info}/METADATA

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

@@ -1,8 +1,8 @@
 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=urDC_apWYMiJSzgtfhpIDp42SEzdtYtTZrbDQe_485M,28611
 csc_cia_stne/email.py,sha256=y4xyPAe6_Mga5Wf6qAsDzYgn0f-zf2KshfItlWe58z8,8481
-csc_cia_stne/ftp.py,sha256=WSs28s7NO-mvEURIsK6VyqqIK_CVSv35ncscajyGu3o,11311
+csc_cia_stne/ftp.py,sha256=eNkOUEXdw-9NYfuZjNo6Oh7EduD54g8N0cfD0LuOiTU,11516
 csc_cia_stne/gcp_bigquery.py,sha256=foq8azvvv_f7uikMDslX9RcUIrx7RAS-Sn0AGW0QFQc,7231
 csc_cia_stne/gcp_bucket.py,sha256=vMALWiW7IoBCuJAR8bUCpOV6BuBzI9AhRRk3b72OdMk,11515
 csc_cia_stne/google_drive.py,sha256=sfq2arBYrv8OLIfRJTnj067EPbem2gofwIfpbqJ475o,44500
@@ -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.18.dist-info/licenses/LICENCE,sha256=LPGMtgKki2C3KEZP7hDhA1HBrlq5JCHkIeStUCLEMx4,1073
-csc_cia_stne-0.1.18.dist-info/METADATA,sha256=E61T5EEsvJQW1qLzbjCZl-EdUD3WVBdqsTfHEdBZsfM,1464
-csc_cia_stne-0.1.18.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-csc_cia_stne-0.1.18.dist-info/top_level.txt,sha256=ldo7GVv3tQx5KJvwBzdZzzQmjPys2NDVVn1rv0BOF2Q,13
-csc_cia_stne-0.1.18.dist-info/RECORD,,
+csc_cia_stne-0.1.20.dist-info/licenses/LICENCE,sha256=LPGMtgKki2C3KEZP7hDhA1HBrlq5JCHkIeStUCLEMx4,1073
+csc_cia_stne-0.1.20.dist-info/METADATA,sha256=klKwNP7-_o1IQIOzS6MTgEjDRkZ2InEKbC0epf_e7hM,1464
+csc_cia_stne-0.1.20.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+csc_cia_stne-0.1.20.dist-info/top_level.txt,sha256=ldo7GVv3tQx5KJvwBzdZzzQmjPys2NDVVn1rv0BOF2Q,13
+csc_cia_stne-0.1.20.dist-info/RECORD,,