From b64068cfb60dc8d1f8fd8d2fa163f3d1c430fa06 Mon Sep 17 00:00:00 2001
From: "ewerton.almeida" <ewerton.almeida@localhost.com>
Date: Thu, 14 Aug 2025 11:22:38 -0300
Subject: [PATCH] Busca correta da unidade cosumidora na fatura

---
 app/layouts/equatorial_go.py | 15 ++++++++++-----
 1 file changed, 10 insertions(+), 5 deletions(-)

diff --git a/app/layouts/equatorial_go.py b/app/layouts/equatorial_go.py
index a0ada00..ec1b4c8 100755
--- a/app/layouts/equatorial_go.py
+++ b/app/layouts/equatorial_go.py
@@ -29,14 +29,19 @@ def extrair_dados(texto_final):
 
     nota_fiscal = extrair_seguro(r'NOTA FISCAL Nº\s*(\d+)', texto_final)
 
+    # --- Unidade Consumidora (UC): 8–12 dígitos, SEM hífen ---
     uc = extrair_seguro([
-        r'(\d{7,10}-\d)',
-        r'UNIDADE\s+CONSUMIDORA\s*[:\-]?\s*(\d{6,})',
-        r'(\d{6,})\s+FAZENDA',
-        r'(\d{6,})\s+AVENIDA',
-        r'(\d{6,})\s+RUA'
+        r'UNIDADE\s*CONSUMIDORA\D*?(\d{8,12})',
+        r'\bUC\D*?(\d{8,12})',
+        r'INSTALA[ÇC][ÃA]O\D*?(\d{8,12})',
     ], texto_final)
 
+    # fallback: maior sequência "solta" de 8–10 dígitos sem hífen
+    if not uc:
+        seqs = re.findall(r'(?<!\d)(\d{8,10})(?![\d-])', texto_final)
+        if seqs:
+            uc = max(seqs, key=len)
+
     logging.debug("TEXTO PDF:\n" + texto_final)
 
     referencia = extrair_seguro([