Upload 58 files

app/services/tse_api.py

@@ -227,38 +227,44 @@ async def buscar_politico(nome: str) -> Dict[str, Any]:
         "ufs": set()
     }
     
-    # Search in recent elections
+    # Search in recent elections - continue through ALL years
     for ano in [2024, 2022, 2020, 2018]:
-        candidatos = await buscar_candidatos(nome, ano=ano)
-        
-        for c in candidatos:
-            if nome.lower() in c.nome.lower():
-                resultado["encontrado"] = True
-                resultado["candidaturas"].append({
-                    "ano": ano,
-                    "cargo": c.cargo,
-                    "partido": c.partido_sigla,
-                    "uf": c.uf,
-                    "situacao": c.situacao,
-                    "patrimonio": c.total_bens
-                })
-                
-                if c.partido_sigla:
-                    resultado["partidos"].add(c.partido_sigla)
-                if c.uf:
-                    resultado["ufs"].add(c.uf)
-                
-                if c.total_bens > resultado["total_patrimonio"]:
-                    resultado["total_patrimonio"] = c.total_bens
-                
-                if not resultado["ultimo_cargo"]:
-                    resultado["ultimo_cargo"] = f"{c.cargo} ({ano})"
-        
-        if resultado["encontrado"]:
-            break  # Found in most recent election
+        try:
+            candidatos = await buscar_candidatos(nome, ano=ano)
+            print(f"TSE: Buscando '{nome}' em {ano} - encontrados: {len(candidatos)}")
+            
+            for c in candidatos:
+                # Match if nome is in the candidate's full name
+                if nome.lower() in c.nome.lower() or nome.lower() in c.nome_urna.lower():
+                    resultado["encontrado"] = True
+                    resultado["candidaturas"].append({
+                        "ano": ano,
+                        "cargo": c.cargo,
+                        "partido": c.partido_sigla,
+                        "uf": c.uf,
+                        "situacao": c.situacao,
+                        "patrimonio": c.total_bens
+                    })
+                    
+                    if c.partido_sigla:
+                        resultado["partidos"].add(c.partido_sigla)
+                    if c.uf:
+                        resultado["ufs"].add(c.uf)
+                    
+                    if c.total_bens > resultado["total_patrimonio"]:
+                        resultado["total_patrimonio"] = c.total_bens
+                    
+                    if not resultado["ultimo_cargo"]:
+                        resultado["ultimo_cargo"] = f"{c.cargo} ({ano})"
+        except Exception as e:
+            print(f"TSE search {ano} error: {e}")
+            continue
     
     # Convert sets to lists for JSON
     resultado["partidos"] = list(resultado["partidos"])
     resultado["ufs"] = list(resultado["ufs"])
     
+    print(f"TSE resultado para '{nome}': encontrado={resultado['encontrado']}, candidaturas={len(resultado['candidaturas'])}")
+    
     return resultado
+