snowflake-cli 3.1.0__py3-none-any.whl → 3.2.0__py3-none-any.whl

snowflake/cli/_plugins/nativeapp/sf_sql_facade.py

@@ -0,0 +1,525 @@
+# Copyright (c) 2024 Snowflake Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+from __future__ import annotations
+
+import logging
+from contextlib import contextmanager
+from textwrap import dedent
+from typing import Any, Dict, List
+
+from snowflake.cli._plugins.nativeapp.sf_facade_constants import UseObjectType
+from snowflake.cli._plugins.nativeapp.sf_facade_exceptions import (
+    CouldNotUseObjectError,
+    InsufficientPrivilegesError,
+    UnexpectedResultError,
+    UserScriptError,
+    handle_unclassified_error,
+)
+from snowflake.cli.api.errno import (
+    DOES_NOT_EXIST_OR_CANNOT_BE_PERFORMED,
+    INSUFFICIENT_PRIVILEGES,
+    NO_WAREHOUSE_SELECTED_IN_SESSION,
+)
+from snowflake.cli.api.identifiers import FQN
+from snowflake.cli.api.project.util import (
+    identifier_to_show_like_pattern,
+    is_valid_unquoted_identifier,
+    to_identifier,
+    to_quoted_identifier,
+    to_string_literal,
+)
+from snowflake.cli.api.sql_execution import BaseSqlExecutor, SqlExecutor
+from snowflake.connector import DictCursor, ProgrammingError
+
+
+class SnowflakeSQLFacade:
+    def __init__(self, sql_executor: SqlExecutor | None = None):
+        self._sql_executor = (
+            sql_executor if sql_executor is not None else BaseSqlExecutor()
+        )
+        self._log = logging.getLogger(__name__)
+
+    def _use_object(self, object_type: UseObjectType, name: str):
+        """
+        Call sql to use snowflake object with error handling
+        @param object_type: ObjectType, type of snowflake object to use
+        @param name: object name, has to be a valid snowflake identifier.
+        """
+        try:
+            self._sql_executor.execute_query(f"use {object_type} {name}")
+        except ProgrammingError as err:
+            if err.errno == DOES_NOT_EXIST_OR_CANNOT_BE_PERFORMED:
+                raise CouldNotUseObjectError(object_type, name) from err
+            else:
+                handle_unclassified_error(err, f"Failed to use {object_type} {name}.")
+        except Exception as err:
+            handle_unclassified_error(err, f"Failed to use {object_type} {name}.")
+
+    @contextmanager
+    def _use_object_optional(self, object_type: UseObjectType, name: str | None):
+        """
+        Call sql to use snowflake object with error handling
+        @param object_type: ObjectType, type of snowflake object to use
+        @param name: object name, will be cast to a valid snowflake identifier.
+        """
+        if name is None:
+            yield
+            return
+
+        try:
+            current_obj_result_row = self._sql_executor.execute_query(
+                f"select current_{object_type}()"
+            ).fetchone()
+        except Exception as err:
+            return handle_unclassified_error(
+                err, f"Failed to select current {object_type}."
+            )
+
+        try:
+            prev_obj = current_obj_result_row[0]
+        except IndexError:
+            prev_obj = None
+
+        if prev_obj is not None and _same_identifier(prev_obj, name):
+            yield
+            return
+
+        self._log.debug(f"Switching to {object_type}: {name}")
+        self._use_object(object_type, to_identifier(name))
+        try:
+            yield
+        finally:
+            if prev_obj is not None:
+                self._log.debug(f"Switching back to {object_type}: {prev_obj}")
+                self._use_object(object_type, prev_obj)
+
+    def _use_warehouse_optional(self, new_wh: str | None):
+        """
+        Switches to a different warehouse for a while, then switches back.
+        This is a no-op if the requested warehouse is already active or if no warehouse is passed in.
+        @param new_wh: Name of the warehouse to use. If not a valid Snowflake identifier, will be converted before use.
+        """
+        return self._use_object_optional(UseObjectType.WAREHOUSE, new_wh)
+
+    def _use_role_optional(self, new_role: str | None):
+        """
+        Switches to a different role for a while, then switches back.
+        This is a no-op if the requested role is already active or if no role is passed in.
+        @param new_role: Name of the role to use. If not a valid Snowflake identifier, will be converted before use.
+        """
+        return self._use_object_optional(UseObjectType.ROLE, new_role)
+
+    def _use_database_optional(self, database_name: str | None):
+        """
+        Switch to database `database_name`, then switches back.
+        This is a no-op if the requested database is already selected or if no database_name is passed in.
+        @param database_name: Name of the database to use. If not a valid Snowflake identifier, will be converted before use.
+        """
+        return self._use_object_optional(UseObjectType.DATABASE, database_name)
+
+    def _use_schema_optional(self, schema_name: str | None):
+        """
+        Switch to schema `schema_name`, then switches back.
+        This is a no-op if the requested schema is already selected or if no schema_name is passed in.
+        @param schema_name: Name of the schema to use. If not a valid Snowflake identifier, will be converted before use.
+        """
+        return self._use_object_optional(UseObjectType.SCHEMA, schema_name)
+
+    def execute_user_script(
+        self,
+        queries: str,
+        script_name: str,
+        role: str | None = None,
+        warehouse: str | None = None,
+        database: str | None = None,
+    ):
+        """
+        Runs the user-provided sql script.
+        @param queries: Queries to run in this script
+        @param script_name: Name of the file containing the script. Used to show logs to the user.
+        @param [Optional] role: Role to switch to while running this script. Current role will be used if no role is passed in.
+        @param [Optional] warehouse: Warehouse to use while running this script.
+        @param [Optional] database: Database to use while running this script.
+        """
+        with (
+            self._use_role_optional(role),
+            self._use_warehouse_optional(warehouse),
+            self._use_database_optional(database),
+        ):
+            try:
+                self._sql_executor.execute_queries(queries)
+            except ProgrammingError as err:
+                if err.errno == NO_WAREHOUSE_SELECTED_IN_SESSION:
+                    raise UserScriptError(
+                        script_name,
+                        f"{err.msg}. Please provide a warehouse in your project definition file, config.toml file, or via command line",
+                    ) from err
+                else:
+                    raise UserScriptError(script_name, err.msg) from err
+            except Exception as err:
+                handle_unclassified_error(err, f"Failed to run script {script_name}.")
+
+    def get_account_event_table(self, role: str | None = None) -> str | None:
+        """
+        Returns the name of the event table for the account.
+        If the account has no event table set up or the event table is set to NONE, returns None.
+        @param [Optional] role: Role to switch to while running this script. Current role will be used if no role is passed in.
+        """
+        query = "show parameters like 'event_table' in account"
+        with self._use_role_optional(role):
+            try:
+                results = self._sql_executor.execute_query(
+                    query, cursor_class=DictCursor
+                )
+            except Exception as err:
+                handle_unclassified_error(err, f"Failed to get event table.")
+        table = next((r["value"] for r in results if r["key"] == "EVENT_TABLE"), None)
+        if table is None or table == "NONE":
+            return None
+        return table
+
+    def create_version_in_package(
+        self,
+        package_name: str,
+        stage_fqn: str,
+        version: str,
+        label: str | None = None,
+        role: str | None = None,
+    ):
+        """
+        Creates a new version in an existing application package.
+        @param package_name: Name of the application package to alter.
+        @param stage_fqn: Stage fully qualified name.
+        @param version: Version name to create.
+        @param [Optional] role: Switch to this role while executing create version.
+        @param [Optional] label: Label for this version, visible to consumers.
+        """
+
+        # Make the version a valid identifier, adding quotes if necessary
+        version = to_identifier(version)
+
+        # Label must be a string literal
+        with_label_cause = (
+            f"\nlabel={to_string_literal(label)}" if label is not None else ""
+        )
+        add_version_query = dedent(
+            f"""\
+                alter application package {package_name}
+                    add version {version}
+                    using @{stage_fqn}{with_label_cause}
+            """
+        )
+        with self._use_role_optional(role):
+            try:
+                self._sql_executor.execute_query(add_version_query)
+            except Exception as err:
+                handle_unclassified_error(
+                    err,
+                    f"Failed to add version {version} to application package {package_name}.",
+                )
+
+    def add_patch_to_package_version(
+        self,
+        package_name: str,
+        stage_fqn: str,
+        version: str,
+        patch: int | None = None,
+        label: str | None = None,
+        role: str | None = None,
+    ) -> int:
+        """
+        Add a new patch, optionally a custom one, to an existing version in an application package.
+        @param package_name: Name of the application package to alter.
+        @param stage_fqn: Stage fully qualified name.
+        @param version: Version name to create.
+        @param [Optional] patch: Patch number to create.
+        @param [Optional] label: Label for this patch, visible to consumers.
+        @param [Optional] role: Switch to this role while executing create version.
+
+        @return patch number created for the version.
+        """
+
+        # Make the version a valid identifier, adding quotes if necessary
+        version = to_identifier(version)
+
+        # Label must be a string literal
+        with_label_clause = (
+            f"\nlabel={to_string_literal(label)}" if label is not None else ""
+        )
+        patch_query = f"{patch}" if patch else ""
+        add_patch_query = dedent(
+            f"""\
+                 alter application package {package_name}
+                     add patch {patch_query} for version {version}
+                     using @{stage_fqn}{with_label_clause}
+             """
+        )
+        with self._use_role_optional(role):
+            try:
+                result_cursor = self._sql_executor.execute_query(
+                    add_patch_query, cursor_class=DictCursor
+                ).fetchall()
+            except Exception as err:
+                handle_unclassified_error(
+                    err,
+                    f"Failed to create patch {patch_query} for version {version} in application package {package_name}.",
+                )
+            try:
+                show_row = result_cursor[0]
+            except IndexError as err:
+                raise UnexpectedResultError(
+                    f"Expected to receive the new patch but the result is empty"
+                ) from err
+            new_patch = show_row["patch"]
+
+        return new_patch
+
+    def get_event_definitions(
+        self, app_name: str, role: str | None = None
+    ) -> list[dict]:
+        """
+        Retrieves event definitions for the specified application.
+        @param app_name: Name of the application to get event definitions for.
+        @return: A list of dictionaries containing event definitions.
+        """
+        query = (
+            f"show telemetry event definitions in application {to_identifier(app_name)}"
+        )
+        with self._use_role_optional(role):
+            try:
+                results = self._sql_executor.execute_query(
+                    query, cursor_class=DictCursor
+                ).fetchall()
+            except Exception as err:
+                handle_unclassified_error(
+                    err,
+                    f"Failed to get event definitions for application {to_identifier(app_name)}.",
+                )
+        return [dict(row) for row in results]
+
+    def get_app_properties(
+        self, app_name: str, role: str | None = None
+    ) -> Dict[str, str]:
+        """
+        Retrieve the properties of the specified application.
+        @param app_name: Name of the application.
+        @return: A dictionary containing the properties of the application.
+        """
+
+        query = f"desc application {to_identifier(app_name)}"
+        with self._use_role_optional(role):
+            try:
+                results = self._sql_executor.execute_query(
+                    query, cursor_class=DictCursor
+                ).fetchall()
+            except Exception as err:
+                handle_unclassified_error(
+                    err, f"Failed to describe application {to_identifier(app_name)}."
+                )
+        return {row["property"]: row["value"] for row in results}
+
+    def share_telemetry_events(
+        self, app_name: str, event_names: List[str], role: str | None = None
+    ):
+        """
+        Shares the specified events from the specified application to the application package provider.
+        @param app_name: Name of the application to share events from.
+        @param events: List of event names to share.
+        """
+
+        self._log.info("sharing events %s", event_names)
+        query = f"alter application {to_identifier(app_name)} set shared telemetry events ({', '.join([to_string_literal(x) for x in event_names])})"
+
+        with self._use_role_optional(role):
+            try:
+                self._sql_executor.execute_query(query)
+            except Exception as err:
+                handle_unclassified_error(
+                    err,
+                    f"Failed to share telemetry events for application {to_identifier(app_name)}.",
+                )
+
+    def create_schema(
+        self, name: str, role: str | None = None, database: str | None = None
+    ):
+        """
+        Creates a schema.
+        @param name: Name of the schema to create. Can be a database-qualified name or just the schema name, in which case the current database or the database passed in will be used.
+        @param [Optional] role: Role to switch to while running this script. Current role will be used if no role is passed in.
+        @param [Optional] database: Database to use while running this query, unless the schema name is database-qualified.
+        """
+        fqn = FQN.from_string(name)
+        identifier = to_identifier(fqn.name)
+        database = fqn.prefix or database
+        with (
+            self._use_role_optional(role),
+            self._use_database_optional(database),
+        ):
+            try:
+                self._sql_executor.execute_query(
+                    f"create schema if not exists {identifier}"
+                )
+            except ProgrammingError as err:
+                if err.errno == INSUFFICIENT_PRIVILEGES:
+                    raise InsufficientPrivilegesError(
+                        f"Insufficient privileges to create schema {name}",
+                        role=role,
+                        database=database,
+                    ) from err
+                handle_unclassified_error(err, f"Failed to create schema {name}.")
+
+    def stage_exists(
+        self,
+        name: str,
+        role: str | None = None,
+        database: str | None = None,
+        schema: str | None = None,
+    ) -> bool:
+        """
+        Checks if a stage exists.
+        @param name: Name of the stage to check for. Can be a fully qualified name or just the stage name, in which case the current database and schema or the database and schema passed in will be used.
+        @param [Optional] role: Role to switch to while running this script. Current role will be used if no role is passed in.
+        @param [Optional] database: Database to use while running this script, unless the stage name is database-qualified.
+        @param [Optional] schema: Schema to use while running this script, unless the stage name is schema-qualified.
+        """
+        fqn = FQN.from_string(name)
+        identifier = to_identifier(fqn.name)
+        database = fqn.database or database
+        schema = fqn.schema or schema
+
+        pattern = identifier_to_show_like_pattern(identifier)
+        if schema and database:
+            in_schema_clause = f" in schema {database}.{schema}"
+        elif schema:
+            in_schema_clause = f" in schema {schema}"
+        elif database:
+            in_schema_clause = f" in database {database}"
+        else:
+            in_schema_clause = ""
+
+        try:
+            with self._use_role_optional(role):
+                try:
+                    results = self._sql_executor.execute_query(
+                        f"show stages like {pattern}{in_schema_clause}",
+                    )
+                except ProgrammingError as err:
+                    if err.errno == DOES_NOT_EXIST_OR_CANNOT_BE_PERFORMED:
+                        return False
+                    if err.errno == INSUFFICIENT_PRIVILEGES:
+                        raise InsufficientPrivilegesError(
+                            f"Insufficient privileges to check if stage {name} exists",
+                            role=role,
+                            database=database,
+                            schema=schema,
+                        ) from err
+                    handle_unclassified_error(
+                        err, f"Failed to check if stage {name} exists."
+                    )
+            return results.rowcount > 0
+        except CouldNotUseObjectError:
+            return False
+
+    def create_stage(
+        self,
+        name: str,
+        encryption_type: str = "SNOWFLAKE_SSE",
+        enable_directory: bool = True,
+        role: str | None = None,
+        database: str | None = None,
+        schema: str | None = None,
+    ):
+        """
+        Creates a stage.
+        @param name: Name of the stage to create. Can be a fully qualified name or just the stage name, in which case the current database and schema or the database and schema passed in will be used.
+        @param [Optional] encryption_type: Encryption type for the stage. Default is Snowflake SSE. Pass an empty string to disable encryption.
+        @param [Optional] enable_directory: Directory settings for the stage. Default is enabled.
+        @param [Optional] role: Role to switch to while running this script. Current role will be used if no role is passed in.
+        @param [Optional] database: Database to use while running this script, unless the stage name is database-qualified.
+        @param [Optional] schema: Schema to use while running this script, unless the stage name is schema-qualified.
+        """
+        fqn = FQN.from_string(name)
+        identifier = to_identifier(fqn.name)
+        database = fqn.database or database
+        schema = fqn.schema or schema
+
+        query = f"create stage if not exists {identifier}"
+        if encryption_type:
+            query += f" encryption = (type = '{encryption_type}')"
+        if enable_directory:
+            query += f" directory = (enable = {str(enable_directory)})"
+        with (
+            self._use_role_optional(role),
+            self._use_database_optional(database),
+            self._use_schema_optional(schema),
+        ):
+            try:
+                self._sql_executor.execute_query(query)
+            except ProgrammingError as err:
+                if err.errno == INSUFFICIENT_PRIVILEGES:
+                    raise InsufficientPrivilegesError(
+                        f"Insufficient privileges to create stage {name}",
+                        role=role,
+                        database=database,
+                        schema=schema,
+                    ) from err
+                handle_unclassified_error(err, f"Failed to create stage {name}.")
+
+    def show_release_directives(
+        self, package_name: str, role: str | None = None
+    ) -> list[dict[str, Any]]:
+        """
+        Show release directives for a package
+        @param package_name: Name of the package
+        @param [Optional] role: Role to switch to while running this script. Current role will be used if no role is passed in.
+        """
+        package_identifier = to_identifier(package_name)
+        with self._use_role_optional(role):
+            try:
+                cursor = self._sql_executor.execute_query(
+                    f"show release directives in application package {package_identifier}",
+                    cursor_class=DictCursor,
+                )
+            except ProgrammingError as err:
+                if err.errno == INSUFFICIENT_PRIVILEGES:
+                    raise InsufficientPrivilegesError(
+                        f"Insufficient privileges to show release directives for package {package_name}",
+                        role=role,
+                    ) from err
+                handle_unclassified_error(
+                    err,
+                    f"Failed to show release directives for package {package_name}.",
+                )
+            return cursor.fetchall()
+
+
+# TODO move this to src/snowflake/cli/api/project/util.py in a separate
+# PR since it's codeowned by the CLI team
+def _same_identifier(id1: str, id2: str) -> bool:
+    """
+    Returns whether two identifiers refer to the same object.
+
+    Two unquoted identifiers are considered the same if they are equal when both are converted to uppercase
+    Two quoted identifiers are considered the same if they are exactly equal
+    An unquoted identifier and a quoted identifier are considered the same
+      if the quoted identifier is equal to the unquoted identifier
+      when the unquoted identifier is converted to uppercase and quoted
+    """
+    # Canonicalize the identifiers by converting unquoted identifiers to uppercase and leaving quoted identifiers as is
+    canonical_id1 = id1.upper() if is_valid_unquoted_identifier(id1) else id1
+    canonical_id2 = id2.upper() if is_valid_unquoted_identifier(id2) else id2
+
+    # The canonical identifiers are equal if they are equal when both are quoted
+    # (if they are already quoted, this is a no-op)
+    return to_quoted_identifier(canonical_id1) == to_quoted_identifier(canonical_id2)

snowflake/cli/_plugins/nativeapp/v2_conversions/compat.py

@@ -16,7 +16,7 @@ from __future__ import annotations
 
 import inspect
 from functools import wraps
-from typing import Any, Dict, Optional, Type, TypeVar, Union
+from typing import Optional, Type, TypeVar
 
 import typer
 from click import ClickException
@@ -34,13 +34,10 @@ from snowflake.cli.api.project.definition_conversion import (
 )
 from snowflake.cli.api.project.schemas.entities.common import EntityModelBase
 from snowflake.cli.api.project.schemas.project_definition import (
-    DefinitionV11,
     DefinitionV20,
     ProjectDefinition,
     ProjectDefinitionV1,
 )
-from snowflake.cli.api.project.schemas.v1.native_app.path_mapping import PathMapping
-from snowflake.cli.api.utils.definition_rendering import render_definition_template
 
 APP_AND_PACKAGE_OPTIONS = [
     inspect.Parameter(
@@ -64,91 +61,6 @@ APP_AND_PACKAGE_OPTIONS = [
 ]
 
 
-def _convert_v2_artifact_to_v1_dict(
-    v2_artifact: Union[PathMapping, str]
-) -> Union[Dict, str]:
-    if isinstance(v2_artifact, PathMapping):
-        return {
-            "src": v2_artifact.src,
-            "dest": v2_artifact.dest,
-            "processors": v2_artifact.processors,
-        }
-    return v2_artifact
-
-
-def _pdf_v2_to_v1(
-    v2_definition: DefinitionV20,
-    package_entity_id: str = "",
-    app_entity_id: str = "",
-    app_required: bool = False,
-) -> DefinitionV11:
-    pdfv1: Dict[str, Any] = {"definition_version": "1.1", "native_app": {}}
-
-    app_definition, app_package_definition = _find_app_and_package_entities(
-        v2_definition=v2_definition,
-        package_entity_id=package_entity_id,
-        app_entity_id=app_entity_id,
-        app_required=app_required,
-    )
-
-    # NativeApp
-    if app_definition and app_definition.fqn.identifier:
-        pdfv1["native_app"]["name"] = app_definition.fqn.identifier
-    else:
-        pdfv1["native_app"]["name"] = app_package_definition.fqn.identifier.split(
-            "_pkg_"
-        )[0]
-    pdfv1["native_app"]["artifacts"] = [
-        _convert_v2_artifact_to_v1_dict(a) for a in app_package_definition.artifacts
-    ]
-    pdfv1["native_app"]["source_stage"] = app_package_definition.stage
-    pdfv1["native_app"]["bundle_root"] = app_package_definition.bundle_root
-    pdfv1["native_app"]["generated_root"] = app_package_definition.generated_root
-    pdfv1["native_app"]["deploy_root"] = app_package_definition.deploy_root
-    pdfv1["native_app"]["scratch_stage"] = app_package_definition.scratch_stage
-
-    # Package
-    pdfv1["native_app"]["package"] = {}
-    pdfv1["native_app"]["package"]["name"] = app_package_definition.fqn.identifier
-    if app_package_definition.distribution:
-        pdfv1["native_app"]["package"][
-            "distribution"
-        ] = app_package_definition.distribution
-    if app_package_definition.meta and app_package_definition.meta.post_deploy:
-        pdfv1["native_app"]["package"][
-            "post_deploy"
-        ] = app_package_definition.meta.post_deploy
-    if app_package_definition.meta:
-        if app_package_definition.meta.role:
-            pdfv1["native_app"]["package"]["role"] = app_package_definition.meta.role
-        if app_package_definition.meta.warehouse:
-            pdfv1["native_app"]["package"][
-                "warehouse"
-            ] = app_package_definition.meta.warehouse
-
-    # Application
-    if app_definition:
-        pdfv1["native_app"]["application"] = {}
-        pdfv1["native_app"]["application"]["name"] = app_definition.fqn.identifier
-        if app_definition.debug:
-            pdfv1["native_app"]["application"]["debug"] = app_definition.debug
-        if app_definition.meta:
-            if app_definition.meta.role:
-                pdfv1["native_app"]["application"]["role"] = app_definition.meta.role
-            if app_definition.meta.warehouse:
-                pdfv1["native_app"]["application"][
-                    "warehouse"
-                ] = app_definition.meta.warehouse
-            if app_definition.meta.post_deploy:
-                pdfv1["native_app"]["application"][
-                    "post_deploy"
-                ] = app_definition.meta.post_deploy
-
-    result = render_definition_template(pdfv1, {})
-    # Override the definition object in global context
-    return result.project_definition
-
-
 def _find_app_and_package_entities(
     v2_definition: DefinitionV20,
     package_entity_id: str,

snowflake/cli/_plugins/nativeapp/version/commands.py

@@ -18,7 +18,6 @@ import logging
 from typing import Optional
 
 import typer
-from click import MissingParameter
 from snowflake.cli._plugins.nativeapp.common_flags import ForceOption, InteractiveOption
 from snowflake.cli._plugins.nativeapp.v2_conversions.compat import (
     force_project_definition_v2,
@@ -54,6 +53,11 @@ def create(
         help=f"""The patch number you want to create for an existing version.
         Defaults to undefined if it is not set, which means the Snowflake CLI either uses the patch specified in the `manifest.yml` file or automatically generates a new patch number.""",
     ),
+    label: Optional[str] = typer.Option(
+        None,
+        "--label",
+        help="A label for the version that is displayed to consumers. If unset, the version label specified in `manifest.yml` file is used.",
+    ),
     skip_git_check: Optional[bool] = typer.Option(
         False,
         "--skip-git-check",
@@ -67,8 +71,6 @@ def create(
     """
     Adds a new patch to the provided version defined in your application package. If the version does not exist, creates a version with patch 0.
     """
-    if version is None and patch is not None:
-        raise MissingParameter("Cannot provide a patch without version!")
 
     cli_context = get_cli_context()
     ws = WorkspaceManager(
@@ -81,6 +83,7 @@ def create(
         EntityActions.VERSION_CREATE,
         version=version,
         patch=patch,
+        label=label,
         force=force,
         interactive=interactive,
         skip_git_check=skip_git_check,

snowflake/cli/_plugins/notebook/manager.py

@@ -26,7 +26,7 @@ from snowflake.cli.api.sql_execution import SqlExecutionMixin
 class NotebookManager(SqlExecutionMixin):
     def execute(self, notebook_name: FQN):
         query = f"EXECUTE NOTEBOOK {notebook_name.sql_identifier}()"
-        return self._execute_query(query=query)
+        return self.execute_query(query=query)
 
     def get_url(self, notebook_name: FQN):
         fqn = notebook_name.using_connection(self._conn)
@@ -64,7 +64,7 @@ class NotebookManager(SqlExecutionMixin):
             ALTER NOTEBOOK {notebook_fqn.identifier} ADD LIVE VERSION FROM LAST;
             """
         )
-        self._execute_queries(queries=queries)
+        self.execute_queries(queries=queries)
 
         return make_snowsight_url(
             self._conn, f"/#/notebooks/{notebook_fqn.url_identifier}"