36 lines
1.1 KiB
Python
36 lines
1.1 KiB
Python
"""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")
|