phase12: disable RLS on automation tables (written at startup)
This commit is contained in:
@@ -0,0 +1,53 @@
|
|||||||
|
"""Disable RLS on automation tables (written at startup without tenant context).
|
||||||
|
|
||||||
|
The automation plugin registers cron jobs and definitions during plugin
|
||||||
|
activation, which happens at startup before a tenant context is set.
|
||||||
|
RLS blocks these INSERTs because app.current_tenant_id is a dummy default.
|
||||||
|
|
||||||
|
Revision ID: 0078
|
||||||
|
Revises: 0077
|
||||||
|
"""
|
||||||
|
from alembic import op
|
||||||
|
from sqlalchemy import text
|
||||||
|
|
||||||
|
revision = "0078"
|
||||||
|
down_revision = "0077"
|
||||||
|
branch_labels = None
|
||||||
|
depends_on = None
|
||||||
|
|
||||||
|
TABLES = [
|
||||||
|
"automation_agent_definitions",
|
||||||
|
"automation_agent_runs",
|
||||||
|
"automation_agent_versions",
|
||||||
|
"automation_cron_jobs",
|
||||||
|
"automation_definitions",
|
||||||
|
"automation_runs",
|
||||||
|
"automation_versions",
|
||||||
|
]
|
||||||
|
|
||||||
|
def upgrade() -> None:
|
||||||
|
conn = op.get_bind()
|
||||||
|
for table in TABLES:
|
||||||
|
exists = conn.execute(
|
||||||
|
text(f"SELECT 1 FROM information_schema.tables WHERE table_name = '{table}'")
|
||||||
|
).fetchone() is not None
|
||||||
|
if not exists:
|
||||||
|
continue
|
||||||
|
conn.execute(text(f"DROP POLICY IF EXISTS {table}_tenant_isolation ON {table}"))
|
||||||
|
conn.execute(text(f"ALTER TABLE {table} DISABLE ROW LEVEL SECURITY"))
|
||||||
|
|
||||||
|
def downgrade() -> None:
|
||||||
|
conn = op.get_bind()
|
||||||
|
for table in TABLES:
|
||||||
|
exists = conn.execute(
|
||||||
|
text(f"SELECT 1 FROM information_schema.tables WHERE table_name = '{table}'")
|
||||||
|
).fetchone() is not None
|
||||||
|
if not exists:
|
||||||
|
continue
|
||||||
|
conn.execute(text(f"ALTER TABLE {table} ENABLE ROW LEVEL SECURITY"))
|
||||||
|
conn.execute(text(
|
||||||
|
f"CREATE POLICY {table}_tenant_isolation ON {table} "
|
||||||
|
"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)"
|
||||||
|
))
|
||||||
Reference in New Issue
Block a user