Versão em produção - código extraído do container

processor.py

@@ -0,0 +1,93 @@
+import logging
+import os
+import shutil
+import asyncio
+from sqlalchemy.future import select
+from utils import extrair_dados_pdf
+from database import AsyncSessionLocal
+from models import Fatura, LogProcessamento
+
+logger = logging.getLogger(__name__)
+
+UPLOADS_DIR = os.path.join("app", "uploads")
+TEMP_DIR = os.path.join(UPLOADS_DIR, "temp")
+
+fila_processamento = asyncio.Queue()
+status_arquivos = {}
+
+def remover_arquivo_temp(caminho_pdf):
+    try:
+        if os.path.exists(caminho_pdf) and TEMP_DIR in caminho_pdf:
+            os.remove(caminho_pdf)
+            logger.info(f"Arquivo temporário removido: {os.path.basename(caminho_pdf)}")
+    except Exception as e:
+        logger.warning(f"Falha ao remover arquivo temporário: {e}")
+
+def salvar_em_uploads(caminho_pdf_temp, nome_original, nota_fiscal):
+    try:
+        extensao = os.path.splitext(nome_original)[1].lower()
+        nome_destino = f"{nota_fiscal}{extensao}"
+        destino_final = os.path.join(UPLOADS_DIR, nome_destino)
+        shutil.copy2(caminho_pdf_temp, destino_final)
+        return destino_final
+    except Exception as e:
+        logger.error(f"Erro ao salvar em uploads: {e}")
+        return caminho_pdf_temp
+
+async def process_single_file(caminho_pdf_temp: str, nome_original: str):
+    async with AsyncSessionLocal() as session:
+        try:
+            dados = extrair_dados_pdf(caminho_pdf_temp)
+            dados['arquivo_pdf'] = nome_original
+
+            existente_result = await session.execute(
+                select(Fatura).filter_by(nota_fiscal=dados['nota_fiscal'], unidade_consumidora=dados['unidade_consumidora'])
+            )
+            if existente_result.scalar_one_or_none():
+                remover_arquivo_temp(caminho_pdf_temp)
+                return {"status": "Duplicado", "dados": dados}
+
+            caminho_final = salvar_em_uploads(caminho_pdf_temp, nome_original, dados['nota_fiscal'])
+            dados['link_arquivo'] = caminho_final
+
+
+            fatura = Fatura(**dados)
+            session.add(fatura)
+
+            session.add(LogProcessamento(
+                status="Sucesso",
+                mensagem="Fatura processada com sucesso",
+                nome_arquivo=nome_original,
+                acao="PROCESSAMENTO"
+            ))
+
+            await session.commit()
+            remover_arquivo_temp(caminho_pdf_temp)
+            return {"status": "Concluído", "dados": dados}
+
+        except Exception as e:
+            await session.rollback()
+            session.add(LogProcessamento(
+            status="Erro",
+            mensagem=str(e),
+            nome_arquivo=nome_original,
+            acao="PROCESSAMENTO"
+        ))
+            await session.commit()
+            logger.error(f"Erro ao processar fatura: {e}", exc_info=True)
+            remover_arquivo_temp(caminho_pdf_temp)
+            return {"status": "Erro", "mensagem": str(e)}
+
+async def processar_em_lote():
+    resultados = []
+    while not fila_processamento.empty():
+        item = await fila_processamento.get()
+        resultado = await process_single_file(item['caminho_pdf'], item['nome_original'])
+        status_arquivos[item['nome_original']] = resultado.get("status", "Erro")
+        resultados.append(resultado)
+    return resultados
+
+def limpar_arquivos_processados():
+    status_arquivos.clear()
+    while not fila_processamento.empty():
+        fila_processamento.get_nowait()