sqlbuild 0.63.3__tar.gz → 0.63.4__tar.gz

{sqlbuild-0.63.3 → sqlbuild-0.63.4}/Cargo.lock

@@ -572,7 +572,7 @@ checksum = "f8fadd59c855ef2080decdef8ff161eb6661b86933c9d82e5ba29dc602a55aba"
 
 [[package]]
 name = "sqlbuild-kata-native"
-version = "0.63.3"
+version = "0.63.4"
 dependencies = [
  "fensu-policy",
  "globset",

{sqlbuild-0.63.3 → sqlbuild-0.63.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sqlbuild
-Version: 0.63.3
+Version: 0.63.4
 Requires-Dist: duckdb>=1.3.2
 Requires-Dist: filelock>=3.20.3
 Requires-Dist: polyglot-sql>=0.9.2

{sqlbuild-0.63.3 → sqlbuild-0.63.4}/crates/sqlbuild-kata-native/Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "sqlbuild-kata-native"
-version = "0.63.3"
+version = "0.63.4"
 edition.workspace = true
 license.workspace = true
 publish.workspace = true

{sqlbuild-0.63.3 → sqlbuild-0.63.4}/pyproject.toml

@@ -7,7 +7,7 @@ build-backend = "maturin"
 
 [project]
 name = "sqlbuild"
-version = "0.63.3"
+version = "0.63.4"
 description = "Typed, test-first SQL pipelines with local E2E testing"
 readme = "README.md"
 requires-python = ">=3.12"

{sqlbuild-0.63.3 → sqlbuild-0.63.4}/src/sqlbuild/adapters/snowflake/classes/snowflake_adapter.py

@@ -63,6 +63,8 @@ from sqlbuild.adapters.snowflake.constants import (
     MFA_AUTHENTICATOR,
     NUMBER_TYPE_NAME,
     OAUTH_AUTHORIZATION_CODE_AUTHENTICATOR,
+    SECONDARY_ROLES_ALL,
+    SECONDARY_ROLES_NONE,
     STATUS_COLUMN_NAME,
     SUCCESS_STATUS_TOKENS,
     TEXT_TYPE_NAMES,
@@ -1478,10 +1480,14 @@ class SnowflakeAdapter(MicrobatchMixin, BaseAdapter):
         warehouse: object | None = connect_config.get("warehouse")
         database: object | None = connect_config.get("database")
         schema: object | None = connect_config.get("schema")
+        secondary_roles: str = self._normalize_secondary_roles(
+            connect_config.pop("secondary_roles", None)
+        )
         raw_connection: Any = snowflake.connector.connect(**connect_config)
         connection: _SnowflakeConnection = _SnowflakeConnection(raw_connection)
         self._initialize_session(
             connection=connection,
+            secondary_roles=secondary_roles,
             role=role,
             warehouse=warehouse,
             database=database,
@@ -2876,12 +2882,13 @@ class SnowflakeAdapter(MicrobatchMixin, BaseAdapter):
         self,
         *,
         connection: _SnowflakeConnection,
+        secondary_roles: str,
         role: object | None,
         warehouse: object | None,
         database: object | None,
         schema: object | None,
     ) -> None:
-        statements: list[str] = []
+        statements: list[str] = [f"USE SECONDARY ROLES {secondary_roles}"]
         normalized_role: str | None = self._normalize_session_value(role)
         normalized_warehouse: str | None = self._normalize_session_value(warehouse)
         normalized_database: str | None = self._normalize_session_value(database)
@@ -2902,6 +2909,18 @@ class SnowflakeAdapter(MicrobatchMixin, BaseAdapter):
         for statement in statements:
             self.execute(connection=connection, sql=statement)
 
+    @staticmethod
+    def _normalize_secondary_roles(value: object | None) -> str:
+        if value is None:
+            return SECONDARY_ROLES_NONE
+        normalized: str = str(value).strip().upper()
+        if normalized not in {SECONDARY_ROLES_ALL, SECONDARY_ROLES_NONE}:
+            raise AdapterUserError(
+                message="Snowflake connection secondary_roles must be 'ALL' or 'NONE'",
+                code="A301",
+            )
+        return normalized
+
     @staticmethod
     def _normalize_session_value(value: object | None) -> str | None:
         if not isinstance(value, str):

{sqlbuild-0.63.3 → sqlbuild-0.63.4}/src/sqlbuild/adapters/snowflake/constants.py

@@ -5,6 +5,8 @@ VIEW_RELATION_TYPE_TOKEN: str = "view"
 EXTERNAL_BROWSER_AUTHENTICATOR: str = "externalbrowser"
 MFA_AUTHENTICATOR: str = "username_password_mfa"
 OAUTH_AUTHORIZATION_CODE_AUTHENTICATOR: str = "oauth_authorization_code"
+SECONDARY_ROLES_ALL: str = "ALL"
+SECONDARY_ROLES_NONE: str = "NONE"
 NUMBER_TYPE_NAME: str = "NUMBER"
 NORMALIZED_LTZ_INPUT_TYPE_NAME: str = "TIMESTAMPLTZ"
 NORMALIZED_NTZ_INPUT_TYPE_NAMES: frozenset[str] = frozenset({"TIMESTAMP", "TIMESTAMPNTZ"})