snowflake-cli-labs 2.2.0rc0__tar.gz → 2.3.0__tar.gz

{snowflake_cli_labs-2.2.0rc0 → snowflake_cli_labs-2.3.0}/.pre-commit-config.yaml

@@ -60,7 +60,7 @@ repos:
         entry: "snowcli"
         pass_filenames: true
         files: ^src/.*\.py$
-        exclude: ^src/snowflake/cli/app/telemetry.py$
+        exclude: ^src/snowflake/cli/app/constants.py$
       - id: dependencies-sync
         name: "Copy dependencies from pyproject.toml to requirements.txt"
         language: system

{snowflake_cli_labs-2.2.0rc0 → snowflake_cli_labs-2.3.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: snowflake-cli-labs
-Version: 2.2.0rc0
+Version: 2.3.0
 Summary: Snowflake CLI
 Project-URL: Source code, https://github.com/snowflakedb/snowflake-cli
 Project-URL: Bug Tracker, https://github.com/snowflakedb/snowflake-cli/issues
@@ -218,27 +218,33 @@ Classifier: Programming Language :: SQL
 Classifier: Topic :: Database
 Requires-Python: >=3.8
 Requires-Dist: gitpython==3.1.43
-Requires-Dist: jinja2==3.1.3
+Requires-Dist: jinja2==3.1.4
 Requires-Dist: pip
-Requires-Dist: pluggy==1.4.0
-Requires-Dist: pydantic==2.7.0
+Requires-Dist: pluggy==1.5.0
+Requires-Dist: pydantic==2.7.1
 Requires-Dist: pyyaml==6.0.1
 Requires-Dist: requests==2.31.0
 Requires-Dist: requirements-parser==0.9.0
 Requires-Dist: rich==13.7.1
 Requires-Dist: setuptools==69.5.1
-Requires-Dist: snowflake-connector-python[secure-local-storage]==3.8.1
+Requires-Dist: snowflake-connector-python[secure-local-storage]==3.10.0
 Requires-Dist: tomlkit==0.12.3
 Requires-Dist: typer==0.12.3
 Requires-Dist: urllib3<2.3,>=1.21.1
 Provides-Extra: development
-Requires-Dist: coverage==7.4.4; extra == 'development'
+Requires-Dist: coverage==7.5.0; extra == 'development'
 Requires-Dist: pre-commit>=3.5.0; extra == 'development'
 Requires-Dist: pytest-randomly==3.15.0; extra == 'development'
-Requires-Dist: pytest==8.1.1; extra == 'development'
+Requires-Dist: pytest==8.2.0; extra == 'development'
 Requires-Dist: syrupy==4.6.1; extra == 'development'
 Description-Content-Type: text/markdown
 
+[![Code quality checks](https://github.com/snowflakedb/snowflake-cli/actions/workflows/lint.yaml/badge.svg)](https://github.com/snowflakedb/snowflake-cli/actions/workflows/lint.yaml)
+[![Integration testing](https://github.com/snowflakedb/snowflake-cli/actions/workflows/integration_test.yaml/badge.svg)](https://github.com/snowflakedb/snowflake-cli/actions/workflows/integration_test.yaml)
+[![CLI Action testing](https://github.com/snowflakedb/snowflake-cli/actions/workflows/test_cli_action.yaml/badge.svg?branch=main)](https://github.com/snowflakedb/snowflake-cli/actions/workflows/test_cli_action.yaml)
+
+[//]: # ([![Python 3.11]&#40;https://img.shields.io/badge/python-3.11-blue.svg&#41;]&#40;https://www.python.org/downloads/release/python-311/&#41;)
+
 # Snowflake CLI
 
 Snowflake CLI is an open-source command-line tool explicitly designed for developer-centric workloads in addition to SQL operations. It is a flexible and extensible tool that can accommodate modern development practices and technologies.

{snowflake_cli_labs-2.2.0rc0 → snowflake_cli_labs-2.3.0}/README.md

@@ -1,3 +1,9 @@
+[![Code quality checks](https://github.com/snowflakedb/snowflake-cli/actions/workflows/lint.yaml/badge.svg)](https://github.com/snowflakedb/snowflake-cli/actions/workflows/lint.yaml)
+[![Integration testing](https://github.com/snowflakedb/snowflake-cli/actions/workflows/integration_test.yaml/badge.svg)](https://github.com/snowflakedb/snowflake-cli/actions/workflows/integration_test.yaml)
+[![CLI Action testing](https://github.com/snowflakedb/snowflake-cli/actions/workflows/test_cli_action.yaml/badge.svg?branch=main)](https://github.com/snowflakedb/snowflake-cli/actions/workflows/test_cli_action.yaml)
+
+[//]: # ([![Python 3.11](https://img.shields.io/badge/python-3.11-blue.svg)](https://www.python.org/downloads/release/python-311/))
+
 # Snowflake CLI
 
 Snowflake CLI is an open-source command-line tool explicitly designed for developer-centric workloads in addition to SQL operations. It is a flexible and extensible tool that can accommodate modern development practices and technologies.

{snowflake_cli_labs-2.2.0rc0 → snowflake_cli_labs-2.3.0}/RELEASE-NOTES.md

@@ -1,3 +1,38 @@
+# Unreleased version
+
+## Backward incompatibility
+
+## Deprecations
+
+## New additions
+
+## Fixes and improvements
+
+
+# v2.3.0
+
+## Backward incompatibility
+
+## Deprecations
+
+## New additions
+* New `snow sql` functionality:
+  * `-D/--variable` allows variable substitutions in a SQL input (client-side query templating)
+* New `snow app deploy` functionality:
+  * Passing files and directories as arguments syncs these only: `snow app deploy some-file some-dir`.
+  * `--recursive` syncs all files and subdirectories recursively.
+  * `--prune` deletes specified files from the stage if they don't exist locally.
+
+## Fixes and improvements
+* More human-friendly errors in case of corrupted `config.toml` file.
+* Fixed a bug in `snow app` that caused files to be re-uploaded unnecessarily.
+* Optimize snowpark dependency search to lower the size of .zip artifacts and
+  the number of anaconda dependencies for snowpark projects.
+* Added support for fully qualified stage names in stage and git execute commands.
+* Fixed a bug where `snow app run` was not upgrading the application when the local state and remote stage are identical (for example immediately after `snow app deploy`).
+* Fixed handling of stage path separators on Windows
+* The `--info` callback returns info about configured feature flags.
+
 # v2.2.0
 
 ## Backward incompatibility
@@ -12,7 +47,7 @@
   * `--allow-native-libraries` is deprecated in favour of boolean flag `--allow-shared-libraries`.
     Using old flag will cause a warning.
 * `snow snowpark build`:
-  * `--pypi-download` flag is deprecated, have no effect and will cause a warning. Create command always check against PyPi.
+  * `--pypi-download` flag is deprecated, have no effect and will cause a warning. Build command always check against PyPi.
   * `--check-anaconda-for-pypi-depts` is deprecated and using it will cause warning, the functionality is replaced by `--ignore-anaconda`
   * `--package-native-libraries` is deprecated and using it will cause warning, the functionality is replaced by `--allow-shared-libraries`
 * `snow object stage` commands are deprecated and using them will cause a warning.
@@ -37,7 +72,7 @@
 * Added support for snowgit. New commands:
   * `snow git setup` - wizard setting up a git repository stage and creating all necessary objects
   * `snow git fetch` - fetches latest changes from the origin repository into Snowflake repository
-  * `snow git list-brahces` - lists all branches in the repository
+  * `snow git list-branches` - lists all branches in the repository
   * `snow git list-tags` - lists all tags in the repository
   * `snow git list-files` - lists all files on provided branch/tag/commit
   * `snow git copy` - copies files from provided branch/tag/commit into stage or local directory
@@ -54,7 +89,7 @@
 * Fixed errors during `spcs image-registry login` not being formatted correctly.
 * Project definition no longer accept extra fields. Any extra field will cause an error.
 * Changing imports in function/procedure section in `snowflake.yml` will cause the definition update on replace
-* Adding `--pattern` flag to `stage list` command for filtering out results with regex.
+* Adding `--pattern` flag to `stage list-files` command for filtering out results with regex.
 * Fixed snowpark build paths for builds with --project option (fixed empty zip issue).
 * More clear error messages in `snow snowpark build` command
 * Adding support for any source supported by `pip` in `snow snowpark`.
@@ -62,6 +97,7 @@
 * Fix handling database/schema/role identifiers containing dashes
 * Fix schema override bug in `snow connection test`
 * Hidden incorrectly working config permissions warning on Windows
+* Make errors from `snow connection test` more meaningful when role, warehouse or database does not exist.
 
 # v2.1.2
 

snowflake_cli_labs-2.3.0/conftest.py

@@ -0,0 +1,24 @@
+from unittest import mock
+
+import pytest
+
+
+@pytest.fixture(name="_patch_app_version_in_tests", autouse=True)
+def mock_app_version_in_tests(request):
+    """Set predefined Snowflake-CLI for testing.
+
+    Marker `app_version_patch` can be used in tests to skip it.
+
+    @pytest.mark.app_version_patch(False)
+    def test_case():
+        ...
+    """
+    marker = request.node.get_closest_marker("app_version_patch")
+
+    if marker and marker.kwargs.get("use") is False:
+        yield
+    else:
+        with mock.patch(
+            "snowflake.cli.__about__.VERSION", "0.0.0-test_patched"
+        ) as _fixture:
+            yield _fixture

{snowflake_cli_labs-2.2.0rc0 → snowflake_cli_labs-2.3.0}/pyproject.toml

@@ -11,20 +11,20 @@ requires-python = ">=3.8"
 description = "Snowflake CLI"
 readme = "README.md"
 dependencies = [
-  "jinja2==3.1.3",
-  "pluggy==1.4.0",
+  "jinja2==3.1.4",
+  "pluggy==1.5.0",
   "PyYAML==6.0.1",
   "rich==13.7.1",
   "requests==2.31.0",
   "requirements-parser==0.9.0",
   "setuptools==69.5.1",
-  "snowflake-connector-python[secure-local-storage]==3.8.1",
+  "snowflake-connector-python[secure-local-storage]==3.10.0",
   "tomlkit==0.12.3",
   "typer==0.12.3",
   "urllib3>=1.21.1,<2.3",
   "GitPython==3.1.43",
   "pip",
-  "pydantic==2.7.0"
+  "pydantic==2.7.1",
 ]
 classifiers = [
   "Development Status :: 3 - Alpha",
@@ -40,9 +40,9 @@ classifiers = [
 
 [project.optional-dependencies]
 development = [
-  "coverage==7.4.4",
+  "coverage==7.5.0",
   "pre-commit>=3.5.0",
-  "pytest==8.1.1",
+  "pytest==8.2.0",
   "pytest-randomly==3.15.0",
   "syrupy==4.6.1",
 ]
@@ -101,6 +101,9 @@ features = ["development"]
 
 [tool.hatch.envs.integration.scripts]
 test = ["pytest -m integration --snapshot-warn-unused"]
+test-experimental = [
+  "pytest -m integration_experimental --snapshot-warn-unused",
+]
 
 [[tool.hatch.envs.local.matrix]]
 python = ["3.8", "3.9", "3.10", "3.11", "3.12"]
@@ -125,12 +128,14 @@ select = [
 ]
 
 [tool.pytest.ini_options]
-addopts = "-m 'not integration and not performance and not e2e and not spcs and not loaded_modules'"
+addopts = "-m 'not integration and not performance and not e2e and not spcs and not loaded_modules and not integration_experimental'"
 markers = [
   "integration: mark test as integration test",
   "performance: mark test as performance test",
   "e2e: mark test to execute on Snowflake CLI installed in fresh virtual environment",
   "loaded_modules: checks loaded modules",
+  "patch_app_version: sets app version to 0.0.0-test_patched",
+  "integration_experimental: experimental integration test",
 ]
 
 

snowflake_cli_labs-2.3.0/scripts/main.py

@@ -0,0 +1,52 @@
+from pathlib import Path
+
+from typer import Argument, Typer
+
+EMPTY_RELEASE_NOTES = """\
+# Unreleased version
+## Backward incompatibility
+
+## Deprecations
+
+## New additions
+
+## Fixes and improvements
+
+"""
+
+REPO_ROOT = Path(__file__).parent.parent
+
+
+def main():
+    app = Typer(help="Snowflake CLI dev tool")
+
+    @app.command(hidden=True)
+    def _dummy():
+        """To force typer to require command"""
+        pass
+
+    @app.command()
+    def update_release_notes(
+        new_version: str = Argument(help="Version being released"),
+    ):
+        """
+        Updates release notes with version being release, prepares template
+        for next unreleased version.
+        """
+        release_notes = REPO_ROOT / "RELEASE-NOTES.md"
+        with release_notes.open("r") as fh:
+            current_content = fh.readlines()
+        current_content[0] = f"# v{new_version}\n"
+        new_content = [
+            *EMPTY_RELEASE_NOTES.splitlines(keepends=True),
+            "\n",
+            *current_content,
+        ]
+        with release_notes.open("w+") as fh:
+            fh.writelines(new_content)
+
+    app()
+
+
+if __name__ == "__main__":
+    main()

{snowflake_cli_labs-2.2.0rc0 → snowflake_cli_labs-2.3.0}/snyk/requirements.txt

@@ -1,19 +1,19 @@
-jinja2==3.1.3
-pluggy==1.4.0
+jinja2==3.1.4
+pluggy==1.5.0
 PyYAML==6.0.1
 rich==13.7.1
 requests==2.31.0
 requirements-parser==0.9.0
 setuptools==69.5.1
-snowflake-connector-python[secure-local-storage]==3.8.1
+snowflake-connector-python[secure-local-storage]==3.10.0
 tomlkit==0.12.3
 typer==0.12.3
 urllib3>=1.21.1,<2.3
 GitPython==3.1.43
 pip
-pydantic==2.7.0
-coverage==7.4.4
+pydantic==2.7.1
+coverage==7.5.0
 pre-commit>=3.5.0
-pytest==8.1.1
+pytest==8.2.0
 pytest-randomly==3.15.0
 syrupy==4.6.1

{snowflake_cli_labs-2.2.0rc0 → snowflake_cli_labs-2.3.0}/src/snowflake/cli/__about__.py

@@ -1,3 +1,3 @@
 from __future__ import annotations
 
-VERSION = "2.2.0rc0"
+VERSION = "2.3.0"

{snowflake_cli_labs-2.2.0rc0 → snowflake_cli_labs-2.3.0}/src/snowflake/cli/api/commands/flags.py

@@ -1,6 +1,7 @@
 from __future__ import annotations
 
 import tempfile
+from dataclasses import dataclass
 from enum import Enum
 from inspect import signature
 from pathlib import Path
@@ -532,3 +533,25 @@ def deprecated_flag_callback_enum(msg: str):
         return value.value
 
     return _warning_callback
+
+
+@dataclass
+class Variable:
+    key: str
+    value: str
+
+    def __init__(self, key: str, value: str):
+        self.key = key
+        self.value = value
+
+
+def parse_key_value_variables(variables: List[str]) -> List[Variable]:
+    """Util for parsing key=value input. Useful for commands accepting multiple input options."""
+    result = []
+    for p in variables:
+        if "=" not in p:
+            raise ClickException(f"Invalid variable: '{p}'")
+
+        key, value = p.split("=", 1)
+        result.append(Variable(key.strip(), value.strip()))
+    return result

{snowflake_cli_labs-2.2.0rc0 → snowflake_cli_labs-2.3.0}/src/snowflake/cli/api/config.py

@@ -17,10 +17,11 @@ from snowflake.cli.api.exceptions import (
 )
 from snowflake.cli.api.secure_path import SecurePath
 from snowflake.cli.api.secure_utils import file_permissions_are_strict
+from snowflake.cli.api.utils.types import try_cast_to_bool
 from snowflake.connector.compat import IS_WINDOWS
 from snowflake.connector.config_manager import CONFIG_MANAGER
 from snowflake.connector.constants import CONFIG_FILE, CONNECTIONS_FILE
-from snowflake.connector.errors import MissingConfigOptionError
+from snowflake.connector.errors import ConfigSourceError, MissingConfigOptionError
 from tomlkit import TOMLDocument, dump
 from tomlkit.container import Container
 from tomlkit.exceptions import NonExistentKey
@@ -141,7 +142,12 @@ def _read_config_file():
                 message="Bad owner or permissions.*",
                 module="snowflake.connector.config_manager",
             )
-        CONFIG_MANAGER.read_config()
+        try:
+            CONFIG_MANAGER.read_config()
+        except ConfigSourceError as exception:
+            raise ClickException(
+                f"Configuration file seems to be corrupted. {str(exception.__cause__)}"
+            )
 
 
 def _initialise_logs_section():
@@ -207,7 +213,13 @@ def get_default_connection_name() -> str:
 
 
 def get_default_connection_dict() -> dict:
-    return get_connection_dict(get_default_connection_name())
+    def_connection_name = get_default_connection_name()
+    if not connection_exists(def_connection_name):
+        raise MissingConfiguration(
+            f"Couldn't find connection for default connection `{def_connection_name}`. "
+            f"Specify connection name or configure default connection."
+        )
+    return get_connection_dict(def_connection_name)
 
 
 def get_config_section(*path) -> dict:
@@ -234,21 +246,12 @@ def get_config_value(*path, key: str, default: Optional[Any] = Empty) -> Any:
 
 def get_config_bool_value(*path, key: str, default: Optional[Any] = Empty) -> bool:
     value = get_config_value(*path, key=key, default=default)
-    # If we get bool then we can return
-    if isinstance(value, bool):
-        return value
-
-    # Now if value is not string then cast it to str. Simplifies logic for 1 and 0
-    if not isinstance(value, str):
-        value = str(value)
-
-    know_booleans_mapping = {"true": True, "false": False, "1": True, "0": False}
-
-    if value.lower() not in know_booleans_mapping:
+    try:
+        return try_cast_to_bool(value)
+    except ValueError:
         raise ClickException(
             f"Expected boolean value for {'.'.join((*path, key))} option."
         )
-    return know_booleans_mapping[value.lower()]
 
 
 def _initialise_config(config_file: Path) -> None:
@@ -307,3 +310,21 @@ def _check_default_config_files_permissions() -> None:
         raise ConfigFileTooWidePermissionsError(CONNECTIONS_FILE)
     if CONFIG_FILE.exists() and not file_permissions_are_strict(CONFIG_FILE):
         raise ConfigFileTooWidePermissionsError(CONFIG_FILE)
+
+
+from typing import Literal
+
+
+def get_feature_flags_section() -> Dict[str, bool | Literal["UNKNOWN"]]:
+    if not config_section_exists(*FEATURE_FLAGS_SECTION_PATH):
+        return {}
+
+    flags = get_config_section(*FEATURE_FLAGS_SECTION_PATH)
+
+    def _bool_or_unknown(value):
+        try:
+            return try_cast_to_bool(value)
+        except ValueError:
+            return "UNKNOWN"
+
+    return {k: _bool_or_unknown(v) for k, v in flags.items()}

{snowflake_cli_labs-2.2.0rc0 → snowflake_cli_labs-2.3.0}/src/snowflake/cli/api/feature_flags.py

@@ -34,3 +34,4 @@ class FeatureFlag(FeatureFlagMixin):
     ENABLE_STREAMLIT_EMBEDDED_STAGE = BooleanFlag(
         "ENABLE_STREAMLIT_EMBEDDED_STAGE", False
     )
+    ENABLE_NOTEBOOKS = BooleanFlag("ENABLE_NOTEBOOKS", False)

{snowflake_cli_labs-2.2.0rc0 → snowflake_cli_labs-2.3.0}/src/snowflake/cli/api/project/definition.py

@@ -1,5 +1,5 @@
 from pathlib import Path
-from typing import Dict, List
+from typing import List
 
 import yaml.loader
 from snowflake.cli.api.cli_global_context import cli_context
@@ -17,19 +17,6 @@ from yaml import load
 DEFAULT_USERNAME = "unknown_user"
 
 
-def merge_left(target: Dict, source: Dict) -> None:
-    """
-    Recursively merges key/value pairs from source into target.
-    Modifies the original dict-like "target".
-    """
-    for k, v in source.items():
-        if k in target and isinstance(target[k], dict):
-            # assumption: all inputs have been validated.
-            merge_left(target[k], v)
-        else:
-            target[k] = v
-
-
 def load_project_definition(paths: List[Path]) -> ProjectDefinition:
     """
     Loads project definition, optionally overriding values. Definition values
@@ -41,17 +28,16 @@ def load_project_definition(paths: List[Path]) -> ProjectDefinition:
 
     with spaths[0].open("r", read_file_limit_mb=DEFAULT_SIZE_LIMIT_MB) as base_yml:
         definition = load(base_yml.read(), Loader=yaml.loader.BaseLoader)
+        project = ProjectDefinition(**definition)
 
     for override_path in spaths[1:]:
         with override_path.open(
             "r", read_file_limit_mb=DEFAULT_SIZE_LIMIT_MB
         ) as override_yml:
             overrides = load(override_yml.read(), Loader=yaml.loader.BaseLoader)
-            merge_left(definition, overrides)
-
-        # TODO: how to show good error messages here?
+            project.update_from_dict(overrides)
 
-    return ProjectDefinition(**definition)
+    return project
 
 
 def generate_local_override_yml(
@@ -82,12 +68,8 @@ def generate_local_override_yml(
             },
             "package": {"name": package_identifier, "role": role},
         }
-    # TODO: this is an ugly workaround, because pydantics BaseModel.model_copy(update=) doesn't work properly
-    # After fixing UpdatableModel.update_from_dict it should be used here
-    target_definition = project.model_dump()
-    merge_left(target_definition, local)
 
-    return ProjectDefinition(**target_definition)
+    return project.update_from_dict(local)
 
 
 def default_app_package(project_name: str):

{snowflake_cli_labs-2.2.0rc0 → snowflake_cli_labs-2.3.0}/src/snowflake/cli/api/project/schemas/native_app/application.py

@@ -1,6 +1,6 @@
 from __future__ import annotations
 
-from typing import Literal, Optional
+from typing import Optional
 
 from pydantic import Field
 from snowflake.cli.api.project.schemas.updatable_model import (
@@ -26,6 +26,3 @@ class Application(UpdatableModel):
         title="Whether to enable debug mode when using a named stage to create an application object",
         default=True,
     )
-
-
-DistributionOptions = Literal["internal", "external", "INTERNAL", "EXTERNAL"]

{snowflake_cli_labs-2.2.0rc0 → snowflake_cli_labs-2.3.0}/src/snowflake/cli/api/project/schemas/native_app/package.py

@@ -1,14 +1,15 @@
 from __future__ import annotations
 
-from typing import List, Optional
+from typing import List, Literal, Optional
 
 from pydantic import Field, field_validator
-from snowflake.cli.api.project.schemas.native_app.application import DistributionOptions
 from snowflake.cli.api.project.schemas.updatable_model import (
     IdentifierField,
     UpdatableModel,
 )
 
+DistributionOptions = Literal["internal", "external", "INTERNAL", "EXTERNAL"]
+
 
 class Package(UpdatableModel):
     scripts: Optional[List[str]] = Field(

snowflake_cli_labs-2.3.0/src/snowflake/cli/api/project/schemas/native_app/path_mapping.py

@@ -0,0 +1,22 @@
+from __future__ import annotations
+
+from typing import Any, Dict, List, Optional, Union
+
+from pydantic import Field
+from snowflake.cli.api.project.schemas.updatable_model import UpdatableModel
+
+
+class AnnotationProcessor(UpdatableModel):
+    name: str = Field(
+        title="Name of a processor to invoke on a collection of artifacts."
+    )
+    properties: Optional[Dict[str, Any]] = Field(
+        title="A set of key-value pairs used to configure the output of the processor. Consult a specific processor's documentation for more details on the supported properties.",
+        default=None,
+    )
+
+
+class PathMapping(UpdatableModel):
+    src: str
+    dest: Optional[str] = None
+    processors: Optional[List[Union[str, AnnotationProcessor]]] = None

{snowflake_cli_labs-2.2.0rc0 → snowflake_cli_labs-2.3.0}/src/snowflake/cli/api/project/schemas/snowpark/callable.py

@@ -40,7 +40,7 @@ class Callable(UpdatableModel):
     )
     secrets: Optional[Dict[str, str]] = Field(
         title="Assigns the names of secrets to variables so that you can use the variables to reference the secrets",
-        default=[],
+        default={},
     )
     imports: Optional[List[str]] = Field(
         title="Stage and path to previously uploaded files you want to import",

snowflake_cli_labs-2.3.0/src/snowflake/cli/api/project/schemas/updatable_model.py

@@ -0,0 +1,39 @@
+from typing import Any, Dict
+
+from pydantic import BaseModel, ConfigDict, Field, ValidationError
+from snowflake.cli.api.project.errors import SchemaValidationError
+from snowflake.cli.api.project.util import IDENTIFIER_NO_LENGTH
+
+
+class UpdatableModel(BaseModel):
+    model_config = ConfigDict(validate_assignment=True, extra="forbid")
+
+    def __init__(self, *args, **kwargs):
+        try:
+            super().__init__(**kwargs)
+        except ValidationError as e:
+            raise SchemaValidationError(e)
+
+    def update_from_dict(self, update_values: Dict[str, Any]):
+        """
+        Takes a dictionary with values to override.
+        If the field type is subclass of a UpdatableModel, its update_from_dict() method is called with
+        the value to be set.
+        If not, we use simple setattr to set new value.
+        Values provided are validated against original restrictions, so it's impossible to overwrite string field with
+        integer value etc.
+        """
+        for field, value in update_values.items():
+            if field in self.model_fields.keys():
+                if (
+                    hasattr(getattr(self, field), "update_from_dict")
+                    and field in self.model_fields_set
+                ):
+                    getattr(self, field).update_from_dict(value)
+                else:
+                    setattr(self, field, value)
+        return self
+
+
+def IdentifierField(*args, **kwargs):  # noqa
+    return Field(max_length=254, pattern=IDENTIFIER_NO_LENGTH, *args, **kwargs)

{snowflake_cli_labs-2.2.0rc0 → snowflake_cli_labs-2.3.0}/src/snowflake/cli/api/sql_execution.py

@@ -64,7 +64,13 @@ class SqlExecutionMixin:
         return list(self._execute_string(dedent(queries), **kwargs))
 
     def use(self, object_type: ObjectType, name: str):
-        return self._execute_query(f"use {object_type.value.sf_name} {name}")
+        try:
+            self._execute_query(f"use {object_type.value.sf_name} {name}")
+        except ProgrammingError:
+            # Rewrite the error to make the message more useful.
+            raise ProgrammingError(
+                f"Could not use {object_type} {name}. Object does not exist, or operation cannot be performed."
+            )
 
     @contextmanager
     def use_role(self, new_role: str):
@@ -224,3 +230,11 @@ class SqlExecutionMixin:
             lambda row: row[name_col] == unquote_identifier(unqualified_name),
         )
         return show_obj_row
+
+    def qualified_name_for_url(
+        self, object_name: str, database: str | None = None, schema: str | None = None
+    ):
+        fqn = self.to_fully_qualified_name(
+            object_name, database=database, schema=schema
+        )
+        return ".".join(unquote_identifier(part) for part in fqn.split("."))