snowflake-cli 3.7.0__tar.gz → 3.7.1__tar.gz

{snowflake_cli-3.7.0 → snowflake_cli-3.7.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: snowflake-cli
-Version: 3.7.0
+Version: 3.7.1
 Summary: Snowflake CLI
 Project-URL: Source code, https://github.com/snowflakedb/snowflake-cli
 Project-URL: Bug Tracker, https://github.com/snowflakedb/snowflake-cli/issues
@@ -217,6 +217,7 @@ Classifier: Programming Language :: Python :: 3 :: Only
 Classifier: Programming Language :: SQL
 Classifier: Topic :: Database
 Requires-Python: >=3.10
+Requires-Dist: certifi==2025.1.31
 Requires-Dist: gitpython==3.1.44
 Requires-Dist: jinja2==3.1.6
 Requires-Dist: packaging

{snowflake_cli-3.7.0 → snowflake_cli-3.7.1}/RELEASE-NOTES.md

@@ -23,6 +23,16 @@
 ## Fixes and improvements
 
 
+# v3.7.1
+
+## Deprecations
+
+## New additions
+
+## Fixes and improvements
+* Fix certificate connection issues.
+* Fix `snow spcs image-registry login` slow query problem.
+
 # v3.7.0
 
 ## Deprecations

{snowflake_cli-3.7.0 → snowflake_cli-3.7.1}/pyproject.toml

@@ -25,6 +25,7 @@ requires-python = ">=3.10"
 description = "Snowflake CLI"
 readme = "README.md"
 dependencies = [
+  "certifi==2025.1.31",
   "GitPython==3.1.44",
   "jinja2==3.1.6",
   "packaging",

{snowflake_cli-3.7.0 → snowflake_cli-3.7.1}/snyk/requirements.txt

@@ -1,3 +1,4 @@
+certifi==2025.1.31
 GitPython==3.1.44
 jinja2==3.1.6
 packaging

{snowflake_cli-3.7.0 → snowflake_cli-3.7.1}/src/snowflake/cli/__about__.py

@@ -16,7 +16,7 @@ from __future__ import annotations
 
 from enum import Enum, unique
 
-VERSION = "3.7.0"
+VERSION = "3.7.1"
 
 
 @unique

{snowflake_cli-3.7.0 → snowflake_cli-3.7.1}/src/snowflake/cli/_plugins/spcs/image_registry/manager.py

@@ -74,12 +74,21 @@ class RegistryManager(SqlExecutionMixin):
         return re.fullmatch(r"^.*//.+", url) is not None
 
     def get_registry_url(self) -> str:
-        repositories_query = "show image repositories in account"
-        result_set = self.execute_query(repositories_query, cursor_class=DictCursor)
-        results = result_set.fetchall()
-        if len(results) == 0:
-            raise NoImageRepositoriesFoundError()
-        sample_repository_url = results[0]["repository_url"]
+        images_query = "show image repositories in schema snowflake.images;"
+        images_result = self.execute_query(images_query, cursor_class=DictCursor)
+
+        results = images_result.fetchone()
+
+        if not results:
+            # fallback to account level query - slower one, so we try to avoid it if possible
+            repositories_query = "show image repositories in account"
+            result_set = self.execute_query(repositories_query, cursor_class=DictCursor)
+            results = result_set.fetchone()
+
+            if not results:
+                raise NoImageRepositoriesFoundError()
+
+        sample_repository_url = results["repository_url"]
         if not self._has_url_scheme(sample_repository_url):
             sample_repository_url = f"//{sample_repository_url}"
         return urlparse(sample_repository_url).netloc

{snowflake_cli-3.7.0 → snowflake_cli-3.7.1}/tests/spcs/test_registry.py

@@ -73,11 +73,40 @@ def test_get_registry_url(mock_execute, mock_conn, mock_cursor):
     ]
 
     mock_execute.return_value = mock_cursor(
-        rows=[{col: row for col, row in zip(MOCK_REPO_COLUMNS, mock_row)}],
+        rows=[{}, {col: row for col, row in zip(MOCK_REPO_COLUMNS, mock_row)}],
         columns=MOCK_REPO_COLUMNS,
     )
     result = RegistryManager().get_registry_url()
     expected_query = "show image repositories in account"
+    assert mock_execute.call_count == 2
+    mock_execute.assert_any_call(expected_query, cursor_class=DictCursor)
+    assert result == "orgname-alias.registry.snowflakecomputing.com"
+
+
+@mock.patch("snowflake.cli._plugins.spcs.image_registry.manager.RegistryManager._conn")
+@mock.patch(
+    "snowflake.cli._plugins.spcs.image_registry.manager.RegistryManager.execute_query"
+)
+def test_get_registry_url_with_schema_query(mock_execute, mock_conn, mock_cursor):
+    mock_row = [
+        "2023-01-01 00:00:00",
+        "IMAGES",
+        "DB",
+        "SCHEMA",
+        "orgname-alias.registry.snowflakecomputing.com/DB/SCHEMA/IMAGES",
+        "TEST_ROLE",
+        "ROLE",
+        "",
+    ]
+
+    mock_execute.return_value = mock_cursor(
+        rows=[{col: row for col, row in zip(MOCK_REPO_COLUMNS, mock_row)}],
+        columns=MOCK_REPO_COLUMNS,
+    )
+
+    result = RegistryManager().get_registry_url()
+    expected_query = "show image repositories in schema snowflake.images;"
+
     mock_execute.assert_called_once_with(expected_query, cursor_class=DictCursor)
     assert result == "orgname-alias.registry.snowflakecomputing.com"
 
@@ -88,14 +117,17 @@ def test_get_registry_url(mock_execute, mock_conn, mock_cursor):
 )
 def test_get_registry_url_no_repositories(mock_execute, mock_conn, mock_cursor):
     mock_execute.return_value = mock_cursor(
-        rows=[],
+        rows=[{}, {}],
         columns=MOCK_REPO_COLUMNS,
     )
     with pytest.raises(NoImageRepositoriesFoundError):
         RegistryManager().get_registry_url()
 
-    expected_query = "show image repositories in account"
-    mock_execute.assert_called_once_with(expected_query, cursor_class=DictCursor)
+    expected_query1 = "show image repositories in schema snowflake.images;"
+    expected_query2 = "show image repositories in account"
+    assert mock_execute.call_count == 2
+    mock_execute.assert_any_call(expected_query1, cursor_class=DictCursor)
+    mock_execute.assert_any_call(expected_query2, cursor_class=DictCursor)
 
 
 @mock.patch(