snowflake-cli 3.6.0__py3-none-any.whl → 3.7.1__py3-none-any.whl

snowflake/cli/api/exceptions.py

@@ -22,46 +22,87 @@ from snowflake.cli.api.constants import ObjectType
 from snowflake.connector.compat import IS_WINDOWS
 
 
-class EnvironmentVariableNotFoundError(ClickException):
+class BaseCliError(ClickException):
+    """Base Cli Exception.
+
+    0 Everything ran smoothly.
+    1 Something went wrong with the client.
+    2 Something went wrong with command line arguments.
+    3 Cli could not connect to server.
+    4 Cli could not communicate properly with server.
+    5 The enhanced_exit_codes parameter was set and Cli exited because of error.
+    """
+
+    def __init__(self, *args, **kwargs):
+        from snowflake.cli.api.cli_global_context import get_cli_context
+
+        if not get_cli_context().enhanced_exit_codes:
+            self.exit_code = kwargs.pop("exit_code", 1)
+        super().__init__(*args, **kwargs)
+
+
+class CliError(BaseCliError):
+    """Generic Cli Error - to be used in favour of ClickException."""
+
+    exit_code = 1
+
+
+class CliArgumentError(BaseCliError):
+    exit_code = 2
+
+
+class CliConnectionError(BaseCliError):
+    exit_code = 3
+
+
+class CliCommunicationError(BaseCliError):
+    exit_code = 4
+
+
+class CliSqlError(BaseCliError):
+    exit_code = 5
+
+
+class EnvironmentVariableNotFoundError(CliError):
     def __init__(self, env_variable_name: str):
         super().__init__(f"Environment variable {env_variable_name} not found")
 
 
-class MissingConfiguration(ClickException):
+class MissingConfigurationError(CliError):
     pass
 
 
-class CycleDetectedError(ClickException):
+class CycleDetectedError(CliError):
     pass
 
 
-class InvalidTemplate(ClickException):
+class InvalidTemplateError(CliError):
     pass
 
 
-class InvalidConnectionConfiguration(ClickException):
+class InvalidConnectionConfigurationError(CliError):
     def format_message(self):
         return f"Invalid connection configuration. {self.message}"
 
 
-class InvalidLogsConfiguration(ClickException):
+class InvalidLogsConfigurationError(CliError):
     def format_message(self):
         return f"Invalid logs configuration. {self.message}"
 
 
-class InvalidPluginConfiguration(ClickException):
+class InvalidPluginConfigurationError(CliError):
     def format_message(self):
         return f"Invalid plugin configuration. {self.message}"
 
 
-class PluginNotInstalledError(ClickException):
+class PluginNotInstalledError(CliError):
     def __init__(self, plugin_name, installed_plugins: List[str]):
         super().__init__(
             f"Plugin {plugin_name} is not installed. Available plugins: {', '.join(installed_plugins)}."
         )
 
 
-class SnowflakeConnectionError(ClickException):
+class SnowflakeConnectionError(CliError):
     def __init__(self, snowflake_err: Exception):
         super().__init__(f"Could not connect to Snowflake. Reason: {snowflake_err}")
 
@@ -71,17 +112,17 @@ class UnsupportedConfigSectionTypeError(Exception):
         super().__init__(f"Unsupported configuration section type {section_type}")
 
 
-class OutputDataTypeError(ClickException):
+class OutputDataTypeError(CliError):
     def __init__(self, got_type: type, expected_type: type):
         super().__init__(f"Got {got_type} type but expected {expected_type}")
 
 
-class CommandReturnTypeError(ClickException):
+class CommandReturnTypeError(CliError):
     def __init__(self, got_type: type):
         super().__init__(f"Commands have to return OutputData type, but got {got_type}")
 
 
-class SnowflakeSQLExecutionError(ClickException):
+class SnowflakeSQLExecutionError(CliError):
     """
     Could not successfully execute the Snowflake SQL statements.
     """
@@ -95,7 +136,7 @@ class SnowflakeSQLExecutionError(ClickException):
         )
 
 
-class ObjectAlreadyExistsError(ClickException):
+class ObjectAlreadyExistsError(CliError):
     def __init__(
         self,
         object_type: ObjectType,
@@ -108,45 +149,45 @@ class ObjectAlreadyExistsError(ClickException):
         super().__init__(msg)
 
 
-class NoProjectDefinitionError(ClickException):
+class NoProjectDefinitionError(CliError):
     def __init__(self, project_type: str, project_root: str | Path):
         super().__init__(
             f"No {project_type} project definition found in {project_root}"
         )
 
 
-class InvalidProjectDefinitionVersionError(ClickException):
+class InvalidProjectDefinitionVersionError(CliError):
     def __init__(self, expected_version: str, actual_version: str):
         super().__init__(
             f"This command only supports definition version {expected_version}, got {actual_version}."
         )
 
 
-class InvalidSchemaError(ClickException):
+class InvalidSchemaError(CliError):
     def __init__(self, schema: str):
         super().__init__(f"Invalid schema {schema}")
 
 
-class SecretsWithoutExternalAccessIntegrationError(ClickException):
+class SecretsWithoutExternalAccessIntegrationError(CliError):
     def __init__(self, object_name: str):
         super().__init__(
             f"{object_name} defined with secrets but without external integration."
         )
 
 
-class FileTooLargeError(ClickException):
+class FileTooLargeError(CliError):
     def __init__(self, path: Path, size_limit_in_kb: int):
         super().__init__(
             f"File {path} is too large (size limit: {size_limit_in_kb} KB)"
         )
 
 
-class DirectoryIsNotEmptyError(ClickException):
+class DirectoryIsNotEmptyError(CliError):
     def __init__(self, path: Path):
         super().__init__(f"Directory '{path}' is not empty")
 
 
-class ConfigFileTooWidePermissionsError(ClickException):
+class ConfigFileTooWidePermissionsError(CliError):
     def __init__(self, path: Path):
         change_permissons_command = (
             f'icacls "{path}" /deny <USER_ID>:F'
@@ -162,26 +203,26 @@ class ConfigFileTooWidePermissionsError(ClickException):
         super().__init__(msg)
 
 
-class DatabaseNotProvidedError(ClickException):
+class DatabaseNotProvidedError(CliError):
     def __init__(self):
         super().__init__(
             "Database not specified. Please update connection to add `database` parameter, or re-run command using `--database` option. Use `snow connection list` to list existing connections."
         )
 
 
-class SchemaNotProvidedError(ClickException):
+class SchemaNotProvidedError(CliError):
     def __init__(self):
         super().__init__(
             "Schema not specified. Please update connection to add `schema` parameter, or re-run command using `--schema` option. Use `snow connection list` to list existing connections."
         )
 
 
-class FQNNameError(ClickException):
+class FQNNameError(CliError):
     def __init__(self, name: str):
         super().__init__(f"Specified name '{name}' is not valid name.")
 
 
-class FQNInconsistencyError(ClickException):
+class FQNInconsistencyError(CliError):
     def __init__(self, part: str, name: str):
         super().__init__(
             f"{part.capitalize()} provided but name '{name}' is fully qualified name."
@@ -206,7 +247,7 @@ class UnmetParametersError(UsageError):
         )
 
 
-class NoWarehouseSelectedInSessionError(ClickException):
+class NoWarehouseSelectedInSessionError(CliError):
     def __init__(self, msg: str):
         super().__init__(
             "Received the following error message while executing SQL statement:\n"
@@ -215,7 +256,7 @@ class NoWarehouseSelectedInSessionError(ClickException):
         )
 
 
-class DoesNotExistOrUnauthorizedError(ClickException):
+class DoesNotExistOrUnauthorizedError(CliError):
     def __init__(self, msg: str):
         super().__init__(
             "Received the following error message while executing SQL statement:\n"
@@ -224,7 +265,7 @@ class DoesNotExistOrUnauthorizedError(ClickException):
         )
 
 
-class CouldNotUseObjectError(ClickException):
+class CouldNotUseObjectError(CliError):
     def __init__(self, object_type: ObjectType, name: str):
         super().__init__(
             f"Could not use {object_type} {name}. Object does not exist, or operation cannot be performed."
@@ -238,7 +279,7 @@ class ShowSpecificObjectMultipleRowsError(RuntimeError):
         )
 
 
-class CouldNotSetKeyPairError(ClickException):
+class CouldNotSetKeyPairError(CliError):
     def __init__(self):
         super().__init__(
             "The public key is set already. Use the rotate command instead."

snowflake/cli/api/identifiers.py

@@ -122,6 +122,8 @@ class FQN:
         name = stage
         if stage.startswith("@"):
             name = stage[1:]
+        if stage.startswith("snow://"):
+            name = stage[7:]
         if stage.startswith("~"):
             return cls(name="~", database=None, schema=None)
         return cls.from_string(name)

snowflake/cli/api/plugins/plugin_config.py

@@ -25,7 +25,7 @@ from snowflake.cli.api.config import (
     get_config_value,
     get_plugins_config,
 )
-from snowflake.cli.api.exceptions import InvalidPluginConfiguration
+from snowflake.cli.api.exceptions import InvalidPluginConfigurationError
 
 
 @dataclass
@@ -66,6 +66,6 @@ class PluginConfigProvider:
 
 def _assert_value_is_bool(value, *, value_name: str, plugin_name: str) -> None:
     if type(value) is not bool:
-        raise InvalidPluginConfiguration(
+        raise InvalidPluginConfigurationError(
             f'[{plugin_name}]: "{value_name}" must be a boolean'
         )

snowflake/cli/api/project/schemas/template.py

@@ -19,7 +19,7 @@ from typing import Any, List, Literal, Optional, Union
 import typer
 from click import ClickException
 from pydantic import BaseModel, Field
-from snowflake.cli.api.exceptions import InvalidTemplate
+from snowflake.cli.api.exceptions import InvalidTemplateError
 from snowflake.cli.api.secure_path import SecurePath
 
 
@@ -68,10 +68,10 @@ class Template(BaseModel):
         for path_in_template in self.files_to_render:
             full_path = template_root / path_in_template
             if not full_path.exists():
-                raise InvalidTemplate(
+                raise InvalidTemplateError(
                     f"[files_to_render] contains not-existing file: {path_in_template}"
                 )
             if full_path.is_dir():
-                raise InvalidTemplate(
+                raise InvalidTemplateError(
                     f"[files_to_render] contains a dictionary: {path_in_template}"
                 )

snowflake/cli/api/rendering/project_templates.py

@@ -24,7 +24,7 @@ from jinja2 import (
     UndefinedError,
     loaders,
 )
-from snowflake.cli.api.exceptions import InvalidTemplate
+from snowflake.cli.api.exceptions import InvalidTemplateError
 from snowflake.cli.api.rendering.jinja import IgnoreAttrEnvironment, env_bootstrap
 from snowflake.cli.api.secure_path import SecurePath
 
@@ -90,9 +90,9 @@ def render_template_files(
             full_path = template_root / path
             full_path.write_text(rendered_result)
         except TemplateSyntaxError as err:
-            raise InvalidTemplate(
+            raise InvalidTemplateError(
                 f"Invalid template syntax in line {err.lineno} of file {path}:\n"
                 f"{err.message}"
             )
         except UndefinedError as err:
-            raise InvalidTemplate(err.message)
+            raise InvalidTemplateError(err.message)

snowflake/cli/api/rendering/sql_templates.py

@@ -20,7 +20,7 @@ from click import ClickException
 from jinja2 import Environment, StrictUndefined, loaders, meta
 from snowflake.cli.api.cli_global_context import get_cli_context
 from snowflake.cli.api.console.console import cli_console
-from snowflake.cli.api.exceptions import InvalidTemplate
+from snowflake.cli.api.exceptions import InvalidTemplateError
 from snowflake.cli.api.metrics import CLICounterField
 from snowflake.cli.api.rendering.jinja import (
     CONTEXT_KEY,
@@ -72,7 +72,7 @@ def choose_sql_jinja_env_based_on_template_syntax(
     has_new_syntax = _does_template_have_env_syntax(new_syntax_env, template_content)
     reference_name_str = f" in {reference_name}" if reference_name else ""
     if has_old_syntax and has_new_syntax:
-        raise InvalidTemplate(
+        raise InvalidTemplateError(
             f"The SQL query{reference_name_str} mixes {_OLD_SQL_TEMPLATE_START} ... {_OLD_SQL_TEMPLATE_END} syntax"
             f" and {_SQL_TEMPLATE_START} ... {_SQL_TEMPLATE_END} syntax."
         )

snowflake/cli/api/sql_execution.py

@@ -89,7 +89,7 @@ class BaseSqlExecutor:
         """Executes a single SQL query and returns the results"""
         return self._execute_string(query, **kwargs)
 
-    def execute_query(self, query: str, **kwargs):
+    def execute_query(self, query: str, **kwargs) -> SnowflakeCursor:
         """Executes a single SQL query and returns the last result"""
         *_, last_result = list(self.execute_string(dedent(query), **kwargs))
         return last_result

snowflake/cli/api/utils/definition_rendering.py

@@ -21,7 +21,7 @@ from jinja2 import Environment, TemplateSyntaxError, nodes
 from packaging.version import Version
 from snowflake.cli.api.cli_global_context import get_cli_context
 from snowflake.cli.api.console import cli_console as cc
-from snowflake.cli.api.exceptions import CycleDetectedError, InvalidTemplate
+from snowflake.cli.api.exceptions import CycleDetectedError, InvalidTemplateError
 from snowflake.cli.api.metrics import CLICounterField
 from snowflake.cli.api.project.schemas.project_definition import (
     ProjectProperties,
@@ -60,7 +60,7 @@ class TemplatedEnvironment:
         try:
             ast = self._jinja_env.parse(template_str)
         except TemplateSyntaxError as e:
-            raise InvalidTemplate(
+            raise InvalidTemplateError(
                 f"Error parsing template from project definition file. Value: '{template_str}'. Error: {e}"
             ) from e
 
@@ -98,7 +98,9 @@ class TemplatedEnvironment:
             )
             or current_attr_chain is not None
         ):
-            raise InvalidTemplate(f"Unexpected template syntax in {template_value}")
+            raise InvalidTemplateError(
+                f"Unexpected template syntax in {template_value}"
+            )
 
         for child_node in ast_node.iter_child_nodes():
             all_referenced_vars.update(
@@ -182,16 +184,20 @@ class TemplateVar:
                 not isinstance(current_dict_level, dict)
                 or key not in current_dict_level
             ):
-                raise InvalidTemplate(f"Could not find template variable {self.key}")
+                raise InvalidTemplateError(
+                    f"Could not find template variable {self.key}"
+                )
             current_dict_level = current_dict_level[key]
 
         value = current_dict_level
 
         if value is None:
-            raise InvalidTemplate(f"Template variable {self.key} does not have a value")
+            raise InvalidTemplateError(
+                f"Template variable {self.key} does not have a value"
+            )
 
         if isinstance(value, (dict, list)):
-            raise InvalidTemplate(
+            raise InvalidTemplateError(
                 f"Template variable {self.key} does not have a scalar value"
             )
 
@@ -245,12 +251,12 @@ def _render_graph_node(env: TemplatedEnvironment, node: Node[TemplateVar]) -> No
 
 def _validate_env_section(env_section: dict):
     if not isinstance(env_section, dict):
-        raise InvalidTemplate(
+        raise InvalidTemplateError(
             "env section in project definition file should be a mapping"
         )
     for variable, value in env_section.items():
         if value is None or isinstance(value, (dict, list)):
-            raise InvalidTemplate(
+            raise InvalidTemplateError(
                 f"Variable {variable} in env section of project definition file should be a scalar"
             )
 

snowflake/cli/api/utils/templating_functions.py

@@ -16,7 +16,7 @@ from __future__ import annotations
 
 from typing import Any, List, Optional
 
-from snowflake.cli.api.exceptions import InvalidTemplate
+from snowflake.cli.api.exceptions import InvalidTemplateError
 from snowflake.cli.api.project.util import (
     concat_identifiers,
     get_env_username,
@@ -41,18 +41,18 @@ class TemplatingFunctions:
         max_count: Optional[int] = None,
     ):
         if min_count is not None and len(args) < min_count:
-            raise InvalidTemplate(
+            raise InvalidTemplateError(
                 f"{func_name} requires at least {min_count} argument(s)"
             )
 
         if max_count is not None and len(args) > max_count:
-            raise InvalidTemplate(
+            raise InvalidTemplateError(
                 f"{func_name} supports at most {max_count} argument(s)"
             )
 
         for arg in args:
             if not isinstance(arg, str):
-                raise InvalidTemplate(f"{func_name} only accepts String values")
+                raise InvalidTemplateError(f"{func_name} only accepts String values")
 
     @staticmethod
     def concat_ids(*args):

{snowflake_cli-3.6.0.dist-info → snowflake_cli-3.7.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: snowflake-cli
-Version: 3.6.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,32 +217,33 @@ 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
 Requires-Dist: pip
 Requires-Dist: pluggy==1.5.0
-Requires-Dist: pydantic==2.10.6
+Requires-Dist: pydantic==2.11.2
 Requires-Dist: pyyaml==6.0.2
 Requires-Dist: requests==2.32.3
 Requires-Dist: requirements-parser==0.11.0
-Requires-Dist: rich==13.9.4
-Requires-Dist: setuptools==76.0.0
+Requires-Dist: rich==14.0.0
+Requires-Dist: setuptools==78.1.0
 Requires-Dist: snowflake-connector-python[secure-local-storage]==3.14.0
-Requires-Dist: snowflake-core==1.0.2; python_version < '3.12'
+Requires-Dist: snowflake-core==1.2.0; python_version < '3.12'
 Requires-Dist: snowflake-snowpark-python<1.26.0,>=1.15.0; python_version < '3.12'
 Requires-Dist: tomlkit==0.13.2
 Requires-Dist: typer==0.15.2
 Requires-Dist: urllib3<2.4,>=1.24.3
 Provides-Extra: development
-Requires-Dist: coverage==7.6.12; extra == 'development'
+Requires-Dist: coverage==7.8.0; extra == 'development'
 Requires-Dist: factory-boy==3.3.3; extra == 'development'
-Requires-Dist: faker==37.0.0; extra == 'development'
+Requires-Dist: faker==37.1.0; extra == 'development'
 Requires-Dist: pre-commit>=3.5.0; extra == 'development'
 Requires-Dist: pytest-httpserver==1.1.2; extra == 'development'
 Requires-Dist: pytest-randomly==3.16.0; extra == 'development'
 Requires-Dist: pytest==8.3.5; extra == 'development'
-Requires-Dist: syrupy==4.9.0; extra == 'development'
+Requires-Dist: syrupy==4.9.1; extra == 'development'
 Provides-Extra: packaging
 Description-Content-Type: text/markdown