database.py

Rama main · réplica de lectura del repositorio original.

"""
database.py – Manages database connections
"""

import os
import json
import logging
import sqlite3
from pathlib import Path
from contextlib import contextmanager

from runtime_paths import CONFIG_DIR

DB_PATH = CONFIG_DIR / "mixerbee.db"

@contextmanager
def get_db_connection():
    """Yields a database connection and guarantees it is closed afterward."""
    DB_PATH.parent.mkdir(parents=True, exist_ok=True)
    conn = sqlite3.connect(DB_PATH, check_same_thread=False, timeout=10.0)
    conn.execute("PRAGMA journal_mode=WAL")
    conn.row_factory = sqlite3.Row
    conn.execute("PRAGMA foreign_keys=ON")
    try:
        yield conn
    finally:
        conn.close()

def init_db():
    """Initializes the database, creating tables and running migrations if needed."""
    with get_db_connection() as conn:
        initialize_schema(conn)


def initialize_schema(conn):
    """Create tables and apply additive migrations on ``conn``, then commit.

    Takes a connection (with ``sqlite3.Row`` rows) so a backup restore can migrate
    a staged copy of the database before it replaces the live one.
    """
    cursor = conn.cursor()

    cursor.execute("""
        CREATE TABLE IF NOT EXISTS presets (
            name TEXT PRIMARY KEY,
            data TEXT NOT NULL,
            created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
        )
    """)
    cursor.execute("""
        CREATE TABLE IF NOT EXISTS schedules (
            id TEXT PRIMARY KEY,
            playlist_name TEXT NOT NULL,
            user_id TEXT NOT NULL,
            job_type TEXT NOT NULL,
            crontab TEXT NOT NULL,
            config_data TEXT,
            last_run TEXT,
            created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
        )
    """)
    
    cursor.execute("""
        CREATE TABLE IF NOT EXISTS settings (
            key TEXT PRIMARY KEY,
            value TEXT,
            updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
        )
    """)
    settings_columns = {r['name'] for r in conn.execute('PRAGMA table_info(settings)')}
    if 'updated_at' not in settings_columns:
        conn.execute('ALTER TABLE settings ADD COLUMN updated_at TIMESTAMP')
    
    import accounts
    accounts.initialize_schema(conn)
    import connections
    connections.initialize_schema(conn)
    from app.build_history import init_history_schema
    init_history_schema(conn)
    conn.commit()