from db.database import conn, cursor


def get_maintenance() -> bool:
    cursor.execute("SELECT value FROM settings WHERE key = 'maintenance_mode'")
    row = cursor.fetchone()
    return bool(row and row[0] == '1')


def set_maintenance(active: bool):
    cursor.execute("""
        INSERT INTO settings (key, value) VALUES ('maintenance_mode', ?)
        ON CONFLICT(key) DO UPDATE SET value = excluded.value
    """, ('1' if active else '0',))
    conn.commit()