databricks-tellr-app 0.3.10.dev2__tar.gz → 0.3.10.dev3__tar.gz

{databricks_tellr_app-0.3.10.dev2 → databricks_tellr_app-0.3.10.dev3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: databricks-tellr-app
-Version: 0.3.10.dev2
+Version: 0.3.10.dev3
 Summary: Tellr application package for Databricks Apps
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown

{databricks_tellr_app-0.3.10.dev2 → databricks_tellr_app-0.3.10.dev3}/databricks_tellr_app.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: databricks-tellr-app
-Version: 0.3.10.dev2
+Version: 0.3.10.dev3
 Summary: Tellr application package for Databricks Apps
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown

{databricks_tellr_app-0.3.10.dev2 → databricks_tellr_app-0.3.10.dev3}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "databricks-tellr-app"
-version = "0.3.10.dev2"
+version = "0.3.10.dev3"
 description = "Tellr application package for Databricks Apps"
 readme = "README.md"
 requires-python = ">=3.10"

{databricks_tellr_app-0.3.10.dev2 → databricks_tellr_app-0.3.10.dev3}/src/core/database.py

@@ -521,6 +521,45 @@ def _run_migrations(engine, schema: str | None = None):
         # --- image_assets.tags: json → jsonb (PostgreSQL) for native @> queries ---
         _migrate_image_assets_tags_json_to_jsonb(conn, schema, _qual, is_sqlite)
 
+        # --- THROWAWAY devloop ownership probe (NOT FOR MERGE) ---
+        # Exercises the shared-owner unblock on a fork: ALTER an *inherited*
+        # app_data_prod table (the case that used to fail "must be owner of
+        # table") + CREATE a new table (which the reassign step below re-homes).
+        if not is_sqlite:
+            logger.info("PROBE: ALTER inherited config_profiles + CREATE _probe_tbl")
+            conn.execute(text(
+                f"ALTER TABLE {_qual('config_profiles')} ADD COLUMN IF NOT EXISTS _probe text"
+            ))
+            conn.execute(text(
+                f"CREATE TABLE IF NOT EXISTS {_qual('_probe_tbl')} (col1 text)"
+            ))
+
+        # --- keep newly created objects owned by the shared role (prod forks) ---
+        _reassign_new_objects_to_shared_owner(conn, is_sqlite)
+
+
+def _reassign_new_objects_to_shared_owner(
+    conn, is_sqlite: bool, owning_role: str = "tellr_app_owners"
+) -> None:
+    """Re-home objects this connection's role just created onto the shared owner.
+
+    No-op on SQLite and when the owning role is absent (static-schema envs).
+    Wrapped in a SAVEPOINT so a failure cannot poison the outer migration
+    transaction. Idempotent: REASSIGN OWNED is a no-op when nothing is owned.
+    """
+    from sqlalchemy import text
+
+    if is_sqlite:
+        return
+    exists = conn.execute(
+        text("SELECT 1 FROM pg_roles WHERE rolname = :r"), {"r": owning_role}
+    ).scalar()
+    if not exists:
+        return
+    logger.info(f"Migration: reassigning new objects to shared owner {owning_role}")
+    with conn.begin_nested():
+        conn.execute(text(f'REASSIGN OWNED BY CURRENT_USER TO "{owning_role}"'))
+
 
 def _ensure_llm_judge_backend_default(
     conn, table_name, qualified_table, schema, is_sqlite