"""disable_rls_temporarily

Revision ID: 0bffea34ee5f
Revises: 14b4aa5f0f5d
Create Date: 2026-09-05 12:01:30.634229

"""
from typing import Sequence, Union

from alembic import op
import sqlalchemy as sa


# revision identifiers, used by Alembic.
revision: str = '0bffea34ee5f'
down_revision: Union[str, None] = '14b4aa5f0f5d'
branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None


RLS_TABLES = [
    "user_org_units", "org_unit_leads", "user_roles", "access_groups", "user_access_groups",
    "org_units", "tenant_subscriptions", "tenant_limit_overrides", "chatbot_documents",
    "drive_activities", "drive_files", "drive_folders", "notifications", "tenant_smtp_configs",
    "user_files", "projects", "roles", "activity_logs", "tenant_storage_configs",
    "chat_sessions", "signing_requests", "tenant_signature_configs", "users"
]

def upgrade() -> None:
    for table in RLS_TABLES:
        op.execute(f"ALTER TABLE {table} DISABLE ROW LEVEL SECURITY")

def downgrade() -> None:
    for table in RLS_TABLES:
        op.execute(f"ALTER TABLE {table} ENABLE ROW LEVEL SECURITY")