data_insert/inserir_processo.py

# Manipulação do banco de dados de adição de processos, adicionar do banco de dados

import sqlite3

DATABASE_FILE = "monitoramento.db"

def initialize_db():
    conn = sqlite3.connect(DATABASE_FILE)
    cursor = conn.cursor()
    cursor.execute('''
        CREATE TABLE IF NOT EXISTS monitor_processos (
            numero_processo TEXT PRIMARY KEY,
            credencial TEXT NOT NULL,
            uf TEXT NOT NULL
        )
    ''')
    conn.commit()
    conn.close()

initialize_db()

async def insert_processo(numero_processo: str, uf: str, credencial: str):
    """
    Insere ou atualiza um número de processo no banco de dados de monitoramento.
    
    Args:
        numero_processo (str): Número do processo a ser monitorado.
        uf (str): Unidade federativa relacionada.
        credencial (str): Credencial do responsável pela requisição.
    
    Returns:
        dict: Mensagem com o status da operação.
    """
    try:
        conn = sqlite3.connect(DATABASE_FILE)
        cursor = conn.cursor()
        cursor.execute('''
            INSERT OR REPLACE INTO monitor_processos (numero_processo, credencial, uf)
            VALUES (?, ?, ?)
        ''', (numero_processo, credencial, uf))
        conn.commit()
        conn.close()
        return {"code": 200, "message": "Processo inserido com sucesso."}
    except Exception as e:
        return {"code": 500, "message": f"Erro ao inserir processo: {str(e)}"}