26 lines
786 B
Python
26 lines
786 B
Python
|
|
"""Disable RLS on tax_rates table (read at startup without tenant context).
|
||
|
|
|
||
|
|
Revision ID: 0077
|
||
|
|
Revises: 0076
|
||
|
|
"""
|
||
|
|
from alembic import op
|
||
|
|
from sqlalchemy import text
|
||
|
|
|
||
|
|
revision = "0077"
|
||
|
|
down_revision = "0076"
|
||
|
|
branch_labels = None
|
||
|
|
depends_on = None
|
||
|
|
|
||
|
|
def upgrade() -> None:
|
||
|
|
op.execute("DROP POLICY IF EXISTS tax_rates_tenant_isolation ON tax_rates")
|
||
|
|
op.execute("ALTER TABLE tax_rates DISABLE ROW LEVEL SECURITY")
|
||
|
|
|
||
|
|
def downgrade() -> None:
|
||
|
|
op.execute("ALTER TABLE tax_rates ENABLE ROW LEVEL SECURITY")
|
||
|
|
op.execute(
|
||
|
|
"CREATE POLICY tax_rates_tenant_isolation ON tax_rates "
|
||
|
|
"FOR ALL "
|
||
|
|
"USING (tenant_id = current_setting('app.current_tenant_id', true)::uuid) "
|
||
|
|
"WITH CHECK (tenant_id = current_setting('app.current_tenant_id', true)::uuid)"
|
||
|
|
)
|