import sqlite3
from config import DB_PATH

def get_db():
    """Get database connection (staging or prod)."""
    conn = sqlite3.connect(DB_PATH)
    conn.row_factory = sqlite3.Row
    return conn

def init_db():
    """Initialize tables (safe to run multiple times)."""
    conn = get_db()
    # Create tables if they don't exist
    # Documents table, events table, maintenance table
    conn.close()