Cálculo de Desempenho de Deputados - Projeto Radar Cidadão

try:
    cursor.close()
except Exception:
    pass

cursor = conn.cursor(buffered=True)

pesos_temas_completos = {
    '39': 1.0, '44': 1.0, '46': 1.0, '52': 1.0, '56': 1.0, '57': 1.0, '58': 1.0, '61': 1.0, '64': 1.0, '67': 1.0,
    '34': 0.5, '35': 0.5, '37': 0.5, '40': 0.5, '41': 0.5, '42': 0.5, '43': 0.5, '48': 0.5, '51': 0.5, '53': 0.5,
    '54': 0.5, '55': 0.5, '60': 0.5, '62': 0.5, '66': 0.5, '68': 0.5, '70': 0.5, '74': 0.5, '76': 0.5, '85': 0.5,
    '86': 0.5, '72': 0.1
}

NOME_TABELA = "Lumina2.tema_proposicoes"
COLUNA_ID = "cd_tema"
COLUNA_PESO = "peso"

dados_update = []
for id_tema, peso in pesos_temas_completos.items():
    id_inteiro = int(id_tema)
    tupla_formatada = (peso, id_inteiro)
    dados_update.append(tupla_formatada)

try:
    query_update = f"""
        UPDATE {NOME_TABELA}
        SET {COLUNA_PESO} = %s
        WHERE {COLUNA_ID} = %s
    """
    print(f"Iniciando a atualização em lote na tabela '{NOME_TABELA}'...")
    cursor.executemany(query_update, dados_update)
    conn.commit()

    print(f" Sucesso! Foram processadas {cursor.rowcount} linhas no banco de dados.")

except mysql.connector.Error as erro:
    print(f"\n [FALHA] Ocorreu um erro técnico durante a operação no banco:")
    print(f"Detalhes do erro: {erro}")
    conn.rollback()
finally:
    cursor.close()

# Mapeamento de Códigos da API da Câmara para Pesos em Status

PESOS_SITUACAO = {
    # CONCLUÍDO POSITIVO (1.0)
    98: 1.0,   # Mapeamento Histórico / Concluída com sucesso
    114: 1.0,  # Mapeamento Histórico / Aprovada
    925: 1.0,  # Tramitando em Conjunto
    928: 1.0,  # Aguardando Análise de Parecer
    929: 1.0,  # Aguardando Redação Final
    930: 1.0,  # Enviada ao Arquivo
    931: 1.0,  # Aguardando Remessa ao Arquivo
    1140: 1.0, # Transformado em Norma Jurídica
    1230: 1.0, # Transformado em nova proposição
    1285: 1.0, # Tramitação Finalizada
    1299: 1.0, # Enviada ao Congresso Nacional
    1303: 1.0, # Enviada ao Senado Federal

    # EM TRAMITAÇÃO AVANÇADA (0.8)
    900: 0.8,  # Aguardando Autógrafos na Mesa
    914: 0.8,  # Aguardando Originais para Envio ao Arquivo
    915: 0.8,  # Aguardando Parecer
    916: 0.8,  # Aguardando Deliberação de Recurso
    924: 0.8,  # Pronta para Pauta
    926: 0.8,  # Aguardando Apreciação pelo Senado Federal
    927: 0.8,  # Aguardando Apensação
    932: 0.8,  # Aguardando Definição Encaminhamento
    934: 0.8,  # Aguardando despacho de Emenda
    941: 0.8,  # Recusado
    1070: 0.8, # Aguardando Envio ao Executivo
    1150: 0.8, # Aguardando Sanção
    1160: 0.8, # Aguardando Remessa à Sanção
    1270: 0.8, # Aguardando Envio à Redação Final
    1293: 0.8, # Aguardando Envio ao Senado Federal
    1294: 0.8, # Aguardando Promulgação
    1300: 0.8, # Aguardando Leitura do Parecer da Comissão Especial
    1301: 0.8, # Aguardando Interstício Regimental
    1305: 0.8, # Enviada ao TCU

    # EM TRAMITAÇÃO INICIAL (0.5)
    907: 0.5,  # Aguardando Designação de Relator(a)
    908: 0.5,  # Mapeamento Histórico / Em comissão (Inicial)
    909: 0.5,  # Mapeamento Histórico / Aguardando parecer em comissão
    910: 0.5,  # Aguardando Encaminhamento
    911: 0.5,  # Aguardando Instalação de Comissão Temporária
    912: 0.5,  # Aguardando Leitura e Publicação
    913: 0.5,  # Mapeamento Histórico / Publicada
    935: 0.5,  # Aguardando despacho de Substitutivo
    936: 0.5,  # Aguardando Providências Internas
    937: 0.5,  # Vetado totalmente
    940: 0.5,  # Aguardando Despacho de Arquivamento
    1000: 0.5, # Aguardando Recebimento para Publicação - Relatadas
    1030: 0.5, # Aguardando Informações do DCD - Relatadas
    1050: 0.5, # Aguardando análise de prazo recursal
    1052: 0.5, # Aguardando Abertura de Prazo para Recurso
    1060: 0.5, # Encaminhada à Publicação
    1090: 0.5, # Aguardando Análise
    1091: 0.5, # Ag. Análise de Inconstitucionalidade
    1280: 0.5, # Commission em funcionamento
    1295: 0.5, # Aguardando Reformulação de Parecer
    1297: 0.5, # Aguardando Parecer da Comissão Especial
    1298: 0.5, # Aguardando Manifestação do(a)(s) Acusado(a)(s)
    1310: 0.5, # Em tramitação no Conselho de Ética
    1313: 0.5, # Aguardando Parecer Preliminar no Conselho de Ética
    1314: 0.5, # Aguardando Defesa Escrita no Conselho de Ética
    1355: 0.5, # Em Instrução Probatória
    1380: 0.5, # Aguardando Elaboração do Parecer pelo(a) Relator(a)

    # AGUARDANDO ANDAMENTO (0.3)
    901: 0.3,  # Aguardando Constitution de Comissão Temporária
    902: 0.3,  # Aguardando Criação de Comissão Temporária
    903: 0.3,  # Aguardando Deliberação
    904: 0.3,  # Aguardando Deliberação de Recurso
    905: 0.3,  # Aguardando Despacho do Presidente da Câmara dos Deputados
    906: 0.3,  # Aguardando Distribuição
    917: 0.3,  # Aguardando Recebimento
    918: 0.3,  # Aguardando Recurso
    933: 0.3,  # Aguardando Conhecimento
    939: 0.3,  # Aguardando Apreciação do Veto
    1010: 0.3, # Aguardando Informações do DCD - Novas
    1020: 0.3, # Aguardando Encaminhamento à Publicação
    1040: 0.3, # Aguardando Encaminhamento à CCP para Publicação
    1080: 0.3, # Aguardando Recebimento para Publicação - Novas
    1097: 0.3, # Triagem de Documentos Internos
    1098: 0.3, # Processamento de Proposições Similares
    1110: 0.3, # Aguardando Despacho do Presidente
    1170: 0.3, # Aguardando Designação - Devolução de Relator(a) que deixou de ser Membro
    1180: 0.3, # Aguardando Apoiamento - Conferência de Assinaturas
    1185: 0.3, # Aguardando Apoiamento - Coleta de Assinaturas
    1200: 0.3, # Aguardando Autorização do Despacho
    1201: 0.3, # Aguardando Despacho do Presidente da Câmara dos Deputados (Chancela)
    1210: 0.3, # Aguardando Chancela e Publicação do Despacho
    1220: 0.3, # Aguardando Despacho do Presidente da Câmara dos Deputados (Análise)
    1221: 0.3, # Aguardando Despacho do Presidente da Câmara dos Deputados (Autorização)
    1223: 0.3, # Aguardando Despacho - Requerimentos
    1290: 0.3, # Aguardando Indexação
    1291: 0.3, # Aguardando Indexação (Substituição de Versão)
    1296: 0.3, # Aguardando Eleição do(a) Presidente da Comissão Especial e do(a) Relator(a)
    1302: 0.3, # Aguardando Publicação
    1304: 0.3, # Aguardando Solicitação de Relatoria na CMO
    1311: 0.3, # <Não Definido>
    1312: 0.3, # Aguardando Instauração do Processo
    1350: 0.3, # Aguardando Notificação do(a) Representado(a)
    1381: 0.3, # Aguardando análise do Presidente da Câmara - (Decisão)
    1382: 0.3, # Aguardando Despacho do Presidente da Câmara dos Deputados
    1383: 0.3, # Recebido pelo Protocolo Digital

    # CONCLUÍDO NEGATIVO (0.05)
    32: 0.05,  # Mapeamento Histórico / Rejeitada
    134: 0.05, # Mapeamento Histórico / Arquivada definitivamente
    920: 0.05, # Aguardando Deliberação sobre Prejudicialidade
    921: 0.05, # Aguardando Resposta
    922: 0.05, # Aguardando Vistas
    923: 0.05, # Arquivada
    942: 0.05, # Mapeamento Histórico / Retirada pelo autor antiga
    943: 0.05, # Mapeamento Histórico / Inativa antiga
    950: 0.05, # Retirado pelo(a) Autor(a)
    1120: 0.05,# Devolvida ao(à) Autor(a)
    1222: 0.05,# Prejudicialidade
    1250: 0.05,# Inativa Sinopse (Carga Jan/2001)
    1260: 0.05,# Desmembrada
    1292: 0.05,# Perdeu a Eficácia
    1360: 0.05, # Tramitação Suspensa
    0: 0.05  # Sem Situação Cadastrada / Dados Legados Neutros
}

try:
    #transforma o dicionário em uma lista organizada para o SQL: (peso, cod_situacao)
    lote_dados = [(peso, cod) for cod, peso in PESOS_SITUACAO.items()]

    query = "UPDATE proposicoes SET peso_status = %s WHERE codSituacao = %s;"
    cursor.executemany(query, lote_dados)

    conn.commit()
    print(f"Sucesso! {cursor.rowcount} linhas foram atualizadas.")

except Exception as erro:
    print(f"Erro: {erro}")

# Mapeamento de Códigos da API da Câmara para Pesos em Tipo

PESOS_CODIGOS_SUPERIORES = {
    # PESO 1.0 (Impacto Constitucional e Estrutural Máximo)
    139: 1.0,  # PEC (Proposta de Emenda à Constituição)
    140: 1.0,  # PLP (Projeto de Lei Complementar)

    # PESO 0.8 (Leis Ordinárias, Decretos e Medidas de Alto Impacto)
    141: 0.8,  # PL (Projeto de Lei)
    291: 0.8,  # MPV (Medida Provisória)
    142: 0.8,  # PLN (Projeto de Lei Orçamentária)
    143: 0.8,  # PDL (Projeto de Decreto Legislativo)
    144: 0.8,  # PDC (Projeto de Decreto Legislativo de Concessão)

    # PESO 0.5 (Instrumentos de Fiscalização, Controle e Investigação)
    145: 0.5,  # RIC (Requerimento de Informação)
    146: 0.5,  # PFC (Proposta de Fiscalização e Controle)
    147: 0.5,  # RCP (Requerimento de Criação de CPI)

    # PESO 0.3 (Regimentos e Resoluções de Impacto Interno)
    148: 0.3,  # PRC (Projeto de Resolução da Câmara dos Deputados)
    149: 0.3   # PRN (Projeto de Resolução do Congresso Nacional)
}

NOME_TABELA = "tipo_peso"
COLUNA_ID = "cd_tipo"
COLUNA_PESO = "peso"

try:
    print("1. Buscando todos os IDs existentes na tabela...")
    # Busca apenas a coluna de IDs para saber quem precisa ser atualizado
    query_select = f"SELECT {COLUNA_ID} FROM {NOME_TABELA};"
    cursor.execute(query_select)
    todos_os_ids = [linha[0] for linha in cursor.fetchall()]

    dados_update = []

    for id_tipo in todos_os_ids:
        peso_final = PESOS_CODIGOS_SUPERIORES.get(id_tipo, 0.10)

        # Organiza na ordem do SQL: (peso, cd_tipo)
        dados_update.append((peso_final, id_tipo))

    print(f"2. Iniciando a atualização em lote de {len(dados_update)} registros...")
    query_update = f"""
        UPDATE {NOME_TABELA}
        SET {COLUNA_PESO} = %s
        WHERE {COLUNA_ID} = %s;
    """

    cursor.executemany(query_update, dados_update)
    conn.commit()

    print(f"Sucesso! {cursor.rowcount} linhas tiveram seus pesos alterados.")

except Exception as erro:
    print(f"\n[FALHA] Erro durante o processamento: {erro}")

cursor = conn.cursor()

query_select = """
SELECT t.fk_deputado, t.taxa_assiduidade
FROM taxa_presenca t
JOIN deputado d ON t.fk_deputado = d.cd_deputado
"""
cursor.execute(query_select)
dados = cursor.fetchall()

query_insert = "INSERT INTO desempenho (fk_deputado, score_presenca) VALUES (%s, %s)"

for (id_dep, taxa_assiduidade) in dados:
    valor_dividido = taxa_assiduidade / 100
    try:
        cursor.execute(query_insert, (id_dep, valor_dividido))
    except mysql.connector.Error as err:
        print(f"Pulei o deputado {id_dep} devido ao erro: {err}")

conn.commit()
cursor.close()

cursor = conn.cursor()

query_select = "SELECT fk_deputado, despesa_total, economia FROM economia"
cursor.execute(query_select)
dados_economia = cursor.fetchall()

query_update = "UPDATE desempenho SET score_gastos = %s WHERE fk_deputado = %s"

for (id_dep, despesa, economia) in dados_economia:
    total_geral = despesa + economia

    if total_geral > 0:
        score_gastos = (economia / total_geral)

        # Garante que não fique negativo (se gastou mais que o teto, fica 0)
        score_gastos = max(0, score_gastos)
    else:
        score_gastos = 0

    cursor.execute(query_update, (score_gastos, id_dep))

conn.commit()
cursor.close()

print("Coluna score_gastos atualizada com sucesso!")

try:
    # 1. Calcula o score de cada projeto individualmente
    query_update_score = """
        UPDATE Lumina2.proposicao_deputados
        INNER JOIN Lumina2.proposicoes
            ON proposicao_deputados.fk_proposicao = proposicoes.cd_proposicoes
        LEFT JOIN Lumina2.tipo_peso
            ON proposicoes.fk_tipo = tipo_peso.cd_tipo
        LEFT JOIN Lumina2.tema_proposicoes
            ON proposicoes.cd_proposicoes = tema_proposicoes.id_proposicao
        LEFT JOIN Lumina2.tema_peso
            ON tema_proposicoes.id_tema = tema_peso.cd_tema
        SET proposicao_deputados.score_projetos = (
            (IFNULL(tema_peso.peso, 0.1) * 0.4) +
            (IFNULL(tipo_peso.peso, 0.1) * 0.3) +
            (IFNULL(proposicoes.peso_status, 0.1) * 0.2) +
            (IFNULL(proposicao_deputados.peso, 0.5) * 0.1)
        );
    """

    # 2. agrupa os projetos por deputado, exclui projetos com tema 72 (homenagens, datas) e calcula a média da nota de cada deputado
    query_projetos_unificados = """
        UPDATE Lumina2.desempenho d
        INNER JOIN (
            SELECT
                pd.fk_deputado,
                AVG(pd.score_projetos) AS media_pura,
                COUNT(DISTINCT pd.fk_proposicao) AS volume_valido
            FROM Lumina2.proposicao_deputados pd
            INNER JOIN Lumina2.tema_proposicoes tp ON pd.fk_proposicao = tp.id_proposicao
            WHERE pd.fk_deputado IS NOT NULL
              AND tp.id_tema <> 72
            GROUP BY pd.fk_deputado
        ) m ON d.fk_deputado = m.fk_deputado
        SET d.score_projetos = CASE
            WHEN m.volume_valido <= 3 THEN 0.0
            WHEN m.volume_valido < 112 THEN m.media_pura * (m.volume_valido / 112.0)
            ELSE m.media_pura
        END;
    """

    print("Processando cálculo do score_projetos...")
    cursor.execute(query_update_score)
    cursor.execute(query_projetos_unificados)

    conn.commit()
    print("score_projetos calculado e atualizado com sucesso na tabela desempenho!")

except mysql.connector.Error as erro:
    print(f"[ERRO NA EXECUÇÃO]: {erro}")
    conn.rollback()

# Passo 2
# SCORE FINAL

try:
    cursor.close()
except Exception:
    pass
cursor = conn.cursor(buffered=True)

try:
    #(0.3 + 0.7 = 1.0) + Bônus
    query_calculo_lumina = """
        SELECT
            d.fk_deputado,

        -- 1. COMPONENTE BASE: PRESENÇA (Peso Original 0.3)
            (0.3 * IFNULL(d.score_presenca, 0.0)) AS componente_presenca,

        -- 2. COMPONENTE BASE: PROJETOS (Peso Original 0.7)
            (0.7 * IFNULL(d.score_projetos, 0.0)) AS componente_projetos,

        -- 3. BÔNUS DE GASTOS (Adicional de até 0.10)
            (0.10 * IFNULL(d.score_gastos, 0.0)) AS bonus_gastos,

        -- 4. BÔNUS DE APROVAÇÃO (Adicional de até 0.05)
            CASE
                WHEN ap.fk_deputado IS NOT NULL THEN 0.05
                WHEN av.fk_deputado IS NOT NULL THEN 0.02
                ELSE 0.0
            END AS bonus_aprovacao,

        -- 5. BÔNUS DE COMISSÃO / ÓRGÃO (Adicional de até 0.10)
            (0.10 * IFNULL(lid.maior_peso, 0.00)) AS bonus_orgao

        FROM Lumina2.desempenho d

        LEFT JOIN (
            SELECT DISTINCT pd.fk_deputado
            FROM Lumina2.proposicao_deputados pd
            INNER JOIN Lumina2.proposicoes p ON pd.fk_proposicao = p.cd_proposicoes
            INNER JOIN Lumina2.tema_proposicoes tp ON p.cd_proposicoes = tp.id_proposicao
            WHERE p.codSituacao IN (98, 114, 1140, 1230)
              AND tp.id_tema <> 72
              AND pd.fk_deputado IS NOT NULL
        ) ap ON d.fk_deputado = ap.fk_deputado

        LEFT JOIN (
            SELECT DISTINCT pd.fk_deputado
            FROM Lumina2.proposicao_deputados pd
            INNER JOIN Lumina2.proposicoes p ON pd.fk_proposicao = p.cd_proposicoes
            WHERE p.codSituacao IN (920, 924, 930, 1100, 1120)
              AND pd.fk_deputado IS NOT NULL
        ) av ON d.fk_deputado = av.fk_deputado

        LEFT JOIN (
            SELECT
                fk_deputado,
                MAX(peso_cargo) AS maior_peso
            FROM Lumina2.lideranca_orgaos
            GROUP BY fk_deputado
        ) lid ON d.fk_deputado = lid.fk_deputado;
    """

    cursor.execute(query_calculo_lumina)
    resultados_finais = cursor.fetchall()

    query_update_final = """
        UPDATE Lumina2.desempenho
        SET score_final = %s
        WHERE fk_deputado = %s
    """

    lote_updates = []
    for fk_deputado, comp_pres, comp_proj, b_gastos, b_aprov, b_orgao in resultados_finais:
        # Soma direta: Base (Máximo 1.0) + Bônus Extras (Máximo 0.25)
        score_total = float(comp_pres) + float(comp_proj) + float(b_gastos) + float(b_aprov) + float(b_orgao)

        # Se passar de 1.0 (Nota 10.0 no site), a função min corta e fecha em 1.0
        score_normalizado = max(0.0, min(1.0, score_total))
        lote_updates.append((score_normalizado, fk_deputado))

    if lote_updates:
        print(f" Gravando score_final consolidado para {len(lote_updates)} deputados...")
        cursor.executemany(query_update_final, lote_updates)
        conn.commit()
        print("Sucesso!")
    else:
        print("Nenhum registro modificado.")

except mysql.connector.Error as erro:
    print(f"\n[ERRO]: {erro}")
    conn.rollback()
finally:
    cursor.close()