feat: Phase 2 tenant isolation — RLS on 11 session tables #134
@@ -29,19 +29,35 @@ from app.models.session_branch import SessionBranch # noqa: F401
|
|||||||
from app.models.fork_point import ForkPoint # noqa: F401
|
from app.models.fork_point import ForkPoint # noqa: F401
|
||||||
from app.models.session_handoff import SessionHandoff # noqa: F401
|
from app.models.session_handoff import SessionHandoff # noqa: F401
|
||||||
from app.models.session_resolution_output import SessionResolutionOutput # noqa: F401
|
from app.models.session_resolution_output import SessionResolutionOutput # noqa: F401
|
||||||
|
import os
|
||||||
|
|
||||||
from app.core.config import settings
|
from app.core.config import settings
|
||||||
|
|
||||||
|
|
||||||
def _alembic_sync_url() -> str:
|
def _alembic_sync_url() -> str:
|
||||||
"""Return a psycopg2-compatible sync URL for Alembic.
|
"""Return a psycopg2-compatible sync URL for Alembic.
|
||||||
|
|
||||||
Prefers ADMIN_DATABASE_URL (resolutionflow_admin, BYPASSRLS) converted to
|
Priority order:
|
||||||
a sync driver. Falls back to DATABASE_URL_SYNC for local dev where the
|
1. Railway native PG env vars (PGHOST/PGPASSWORD/PGUSER/PGDATABASE) — these
|
||||||
admin URL may not be configured separately.
|
are auto-linked per-environment by the Railway PostgreSQL service, so they
|
||||||
|
always carry the correct superuser credentials for the current environment
|
||||||
|
(production, PR preview, etc.), including fresh DBs with no custom roles yet.
|
||||||
|
2. ADMIN_DATABASE_URL (resolutionflow_admin, BYPASSRLS) converted to a sync
|
||||||
|
driver — used for local dev and stable environments where the role exists.
|
||||||
|
3. DATABASE_URL_SYNC — last resort / legacy fallback.
|
||||||
"""
|
"""
|
||||||
|
pg_host = os.getenv("PGHOST")
|
||||||
|
pg_user = os.getenv("PGUSER")
|
||||||
|
pg_password = os.getenv("PGPASSWORD")
|
||||||
|
pg_db = os.getenv("PGDATABASE")
|
||||||
|
pg_port = os.getenv("PGPORT", "5432")
|
||||||
|
if all([pg_host, pg_user, pg_password, pg_db]):
|
||||||
|
return f"postgresql://{pg_user}:{pg_password}@{pg_host}:{pg_port}/{pg_db}"
|
||||||
|
|
||||||
admin_url = settings.ADMIN_DATABASE_URL
|
admin_url = settings.ADMIN_DATABASE_URL
|
||||||
if admin_url and "+asyncpg" in admin_url:
|
if admin_url and "+asyncpg" in admin_url:
|
||||||
return admin_url.replace("postgresql+asyncpg://", "postgresql://")
|
return admin_url.replace("postgresql+asyncpg://", "postgresql://")
|
||||||
|
|
||||||
return settings.DATABASE_URL_SYNC
|
return settings.DATABASE_URL_SYNC
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user