Merge pull request #2717 from prefeiturasp/hotfix/8.19.1

Hotfix/8.19.1
prefeiturasp · Mar 4, 2024 · e7ece8f · e7ece8f
2 parents ec845d2 + aee2f0e
commit e7ece8f
Show file tree

Hide file tree

Showing 3 changed files with 25 additions and 19 deletions.
diff --git a/sme_ptrf_apps/__init__.py b/sme_ptrf_apps/__init__.py
@@ -1,4 +1,4 @@
-__version__ = "8.19.0"
+__version__ = "8.19.1"
 
 __version_info__ = tuple(
     [

diff --git a/sme_ptrf_apps/sme/services/exporta_dados_creditos_service.py b/sme_ptrf_apps/sme/services/exporta_dados_creditos_service.py
@@ -66,13 +66,16 @@ def __init__(self, **kwargs) -> None:
         self.data_final = kwargs.get('data_final', None)
         self.nome_arquivo = kwargs.get('nome_arquivo', None)
         self.user = kwargs.get('user', None)
+        self.objeto_arquivo_download = None
 
     def exporta_creditos_principal(self):
+        self.cria_registro_central_download()
         self.cabecalho = CABECALHO_RECEITA[0]
         self.filtra_range_data('data')
         self.exporta_credito_csv()
 
     def exporta_creditos_motivos_estorno(self):
+        self.cria_registro_central_download()
         self.cabecalho = CABECALHO_MOTIVOS_ESTORNO[0]
         self.filtra_range_data('data')
         self.exporta_credito_csv()
@@ -144,26 +147,28 @@ def filtra_range_data(self, field) -> QuerySet:
                 **{f'{field}__lt': self.data_final}
             )
         return self.queryset
+
+    def cria_registro_central_download(self): 
+        logger.info(f"Criando registro na central de download")
+
+        obj = gerar_arquivo_download( 
+            self.user, 
+            self.nome_arquivo ) 
+        self.objeto_arquivo_download = obj
 
     def envia_arquivo_central_download(self, tmp) -> None:
-        logger.info("Gerando arquivo download...")
-        obj_arquivo_download = gerar_arquivo_download(
-            self.user,
-            self.nome_arquivo
-        )
-
         try:
             logger.info("Salvando arquivo download...")
-            obj_arquivo_download.arquivo.save(
-                name=obj_arquivo_download.identificador,
+            self.objeto_arquivo_download.arquivo.save(
+                name=self.objeto_arquivo_download.identificador,
                 content=File(tmp)
             )
-            obj_arquivo_download.status = ArquivoDownload.STATUS_CONCLUIDO
-            obj_arquivo_download.save()
+            self.objeto_arquivo_download.status = ArquivoDownload.STATUS_CONCLUIDO
+            self.objeto_arquivo_download.save()
             logger.info("Arquivo salvo com sucesso...")
 
         except Exception as e:
-            obj_arquivo_download.status = ArquivoDownload.STATUS_ERRO
-            obj_arquivo_download.msg_erro = str(e)
-            obj_arquivo_download.save()
+            self.objeto_arquivo_download.status = ArquivoDownload.STATUS_ERRO
+            self.objeto_arquivo_download.msg_erro = str(e)
+            self.objeto_arquivo_download.save()
             logger.error("Erro arquivo download...")
diff --git a/sme_ptrf_apps/sme/tests/tests_services/test_exportacoes_dados_creditos.py b/sme_ptrf_apps/sme/tests/tests_services/test_exportacoes_dados_creditos.py
@@ -49,7 +49,7 @@ def test_exporta_credito_csv(receita_queryset, usuario_para_teste):
         nome_arquivo='creditos_motivos_estorno.csv',
         queryset=receita_queryset,
         data_inicio=None,
-        data_fim=None,
+        data_final=None,
         user=usuario_para_teste.username,
     )
     service.cabecalho = [
@@ -58,12 +58,11 @@ def test_exporta_credito_csv(receita_queryset, usuario_para_teste):
         ('Descrição do motivo de estorno', 'motivo')
     ],
     service.cabecalho = service.cabecalho[0]
-    service.exporta_credito_csv()
+    service.exporta_creditos_principal()
 
     assert ArquivoDownload.objects.first().arquivo.name == 'creditos_motivos_estorno.csv'
     assert ArquivoDownload.objects.count() == 1
 
-
 def test_envia_arquivo_central_download(usuario_para_teste):
     with NamedTemporaryFile(
         mode="r+",
@@ -73,10 +72,12 @@ def test_envia_arquivo_central_download(usuario_para_teste):
         suffix='.txt'
     ) as file:
         file.write("testando central de download")
-    ExportacoesDadosCreditosService(
+    service = ExportacoesDadosCreditosService(
         nome_arquivo='usuario_para_test.txt',
         user=usuario_para_teste.username
-    ).envia_arquivo_central_download(file)
+    )
+    service.cria_registro_central_download()
+    service.envia_arquivo_central_download(file)
 
     assert ArquivoDownload.objects.count() == 1