database.py

"""
database.py — Conexão e inicialização do banco de dados SQLite.
"""

import sqlite3
from pathlib import Path

DB_PATH = Path(__file__).parent / "helpdesk.db"


def get_connection() -> sqlite3.Connection:
    """Retorna uma conexão com o banco, com row_factory para acesso por nome."""
    conn = sqlite3.connect(DB_PATH)
    conn.row_factory = sqlite3.Row
    conn.execute("PRAGMA foreign_keys = ON")
    return conn


def inicializar_banco() -> None:
    """Cria as tabelas se ainda não existirem."""
    with get_connection() as conn:
        conn.executescript("""
            CREATE TABLE IF NOT EXISTS usuarios (
                id             INTEGER PRIMARY KEY AUTOINCREMENT,
                nome           TEXT    NOT NULL,
                email          TEXT    NOT NULL UNIQUE,
                setor          TEXT    NOT NULL,
                data_cadastro  TEXT    NOT NULL DEFAULT (datetime('now', 'localtime'))
            );

            CREATE TABLE IF NOT EXISTS chamados (
                id                INTEGER PRIMARY KEY AUTOINCREMENT,
                titulo            TEXT    NOT NULL,
                descricao         TEXT    NOT NULL,
                prioridade        TEXT    NOT NULL CHECK(prioridade IN ('Baixa','Média','Alta','Crítica')),
                status            TEXT    NOT NULL DEFAULT 'Aberto'
                                          CHECK(status IN ('Aberto','Em andamento','Resolvido','Fechado')),
                usuario_id        INTEGER NOT NULL REFERENCES usuarios(id),
                data_abertura     TEXT    NOT NULL DEFAULT (datetime('now', 'localtime')),
                data_atualizacao  TEXT    NOT NULL DEFAULT (datetime('now', 'localtime')),
                solucao           TEXT
            );
        """)