snowflake-cli 3.2.2__py3-none-any.whl → 3.4.1__py3-none-any.whl

snowflake/cli/_plugins/nativeapp/entities/application_package_child_interface.py

@@ -0,0 +1,43 @@
+from abc import ABC, abstractmethod
+from pathlib import Path
+from typing import Optional
+
+
+class ApplicationPackageChildInterface(ABC):
+    @abstractmethod
+    def bundle(self, bundle_root=Path, *args, **kwargs) -> None:
+        """
+        Bundles the entity artifacts into the provided root directory. Must not have any side-effects, such as deploying the artifacts into a stage, etc.
+        @param bundle_root: The directory where the bundle contents should be put.
+        """
+        pass
+
+    @abstractmethod
+    def get_deploy_sql(
+        self,
+        artifacts_dir: Path,
+        schema: Optional[str],
+        *args,
+        **kwargs,
+    ) -> str:
+        """
+        Returns the SQL that would create the entity object. Must not execute the SQL or have any other side-effects.
+        @param artifacts_dir: Path to the child entity artifacts directory relative to the deploy root.
+        @param [Optional] schema: Schema to use when creating the object.
+        """
+        pass
+
+    @abstractmethod
+    def get_usage_grant_sql(
+        self,
+        app_role: str,
+        schema: Optional[str],
+        *args,
+        **kwargs,
+    ) -> str:
+        """
+        Returns the SQL that would grant the required USAGE privilege to the provided application role on the entity object. Must not execute the SQL or have any other side-effects.
+        @param app_role: The application role to grant the privileges to.
+        @param [Optional] schema: The schema where the object was created.
+        """
+        pass

snowflake/cli/_plugins/nativeapp/feature_flags.py

@@ -18,7 +18,11 @@ from snowflake.cli.api.feature_flags import BooleanFlag, FeatureFlagMixin
 
 
 @unique
-class FeatureFlag(FeatureFlagMixin):
+class FeatureFlag(
+    FeatureFlagMixin
+):  # TODO move this to snowflake.cli.api.feature_flags
     ENABLE_NATIVE_APP_PYTHON_SETUP = BooleanFlag(
         "ENABLE_NATIVE_APP_PYTHON_SETUP", False
     )
+    ENABLE_NATIVE_APP_CHILDREN = BooleanFlag("ENABLE_NATIVE_APP_CHILDREN", False)
+    ENABLE_RELEASE_CHANNELS = BooleanFlag("ENABLE_RELEASE_CHANNELS", None)

snowflake/cli/_plugins/nativeapp/release_channel/__init__.py

@@ -0,0 +1,13 @@
+# 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.

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

@@ -0,0 +1,246 @@
+# 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 typing import Optional
+
+import typer
+from snowflake.cli._plugins.nativeapp.v2_conversions.compat import (
+    force_project_definition_v2,
+)
+from snowflake.cli._plugins.workspace.manager import WorkspaceManager
+from snowflake.cli.api.cli_global_context import get_cli_context
+from snowflake.cli.api.commands.decorators import with_project_definition
+from snowflake.cli.api.commands.snow_typer import SnowTyperFactory
+from snowflake.cli.api.entities.utils import EntityActions
+from snowflake.cli.api.output.formats import OutputFormat
+from snowflake.cli.api.output.types import (
+    CollectionResult,
+    CommandResult,
+    MessageResult,
+)
+
+app = SnowTyperFactory(
+    name="release-channel",
+    help="Manages release channels of an application package",
+)
+
+log = logging.getLogger(__name__)
+
+
+@app.command("list", requires_connection=True)
+@with_project_definition()
+@force_project_definition_v2()
+def release_channel_list(
+    channel: Optional[str] = typer.Argument(
+        default=None,
+        show_default=False,
+        help="The release channel to list. If not provided, all release channels are listed.",
+    ),
+    **options,
+) -> CommandResult:
+    """
+    Lists the release channels available for an application package.
+    """
+
+    cli_context = get_cli_context()
+    ws = WorkspaceManager(
+        project_definition=cli_context.project_definition,
+        project_root=cli_context.project_root,
+    )
+    package_id = options["package_entity_id"]
+    channels = ws.perform_action(
+        package_id,
+        EntityActions.RELEASE_CHANNEL_LIST,
+        release_channel=channel,
+    )
+
+    if cli_context.output_format == OutputFormat.JSON:
+        return CollectionResult(channels)
+
+
+@app.command("add-accounts", requires_connection=True)
+@with_project_definition()
+@force_project_definition_v2()
+def release_channel_add_accounts(
+    channel: str = typer.Argument(
+        show_default=False,
+        help="The release channel to add accounts to.",
+    ),
+    target_accounts: str = typer.Option(
+        show_default=False,
+        help="The accounts to add to the release channel. Format must be `org1.account1,org2.account2`.",
+    ),
+    **options,
+) -> CommandResult:
+    """
+    Adds accounts to a release channel.
+    """
+
+    cli_context = get_cli_context()
+    ws = WorkspaceManager(
+        project_definition=cli_context.project_definition,
+        project_root=cli_context.project_root,
+    )
+    package_id = options["package_entity_id"]
+    ws.perform_action(
+        package_id,
+        EntityActions.RELEASE_CHANNEL_ADD_ACCOUNTS,
+        release_channel=channel,
+        target_accounts=target_accounts.split(","),
+    )
+
+    return MessageResult("Successfully added accounts to the release channel.")
+
+
+@app.command("remove-accounts", requires_connection=True)
+@with_project_definition()
+@force_project_definition_v2()
+def release_channel_remove_accounts(
+    channel: str = typer.Argument(
+        show_default=False,
+        help="The release channel to remove accounts from.",
+    ),
+    target_accounts: str = typer.Option(
+        show_default=False,
+        help="The accounts to remove from the release channel. Format must be `org1.account1,org2.account2`.",
+    ),
+    **options,
+) -> CommandResult:
+    """
+    Removes accounts from a release channel.
+    """
+
+    cli_context = get_cli_context()
+    ws = WorkspaceManager(
+        project_definition=cli_context.project_definition,
+        project_root=cli_context.project_root,
+    )
+    package_id = options["package_entity_id"]
+    ws.perform_action(
+        package_id,
+        EntityActions.RELEASE_CHANNEL_REMOVE_ACCOUNTS,
+        release_channel=channel,
+        target_accounts=target_accounts.split(","),
+    )
+
+    return MessageResult("Successfully removed accounts from the release channel.")
+
+
+@with_project_definition()
+@app.command("set-accounts", requires_connection=True)
+@force_project_definition_v2()
+def release_channel_set_accounts(
+    channel: str = typer.Argument(
+        show_default=False,
+        help="The release channel to set accounts for.",
+    ),
+    target_accounts: str = typer.Option(
+        show_default=False,
+        help="The accounts to set for the release channel. Format must be `org1.account1,org2.account2`.",
+    ),
+    **options,
+) -> CommandResult:
+    """
+    Sets accounts for a release channel.
+    """
+
+    cli_context = get_cli_context()
+    ws = WorkspaceManager(
+        project_definition=cli_context.project_definition,
+        project_root=cli_context.project_root,
+    )
+    package_id = options["package_entity_id"]
+    ws.perform_action(
+        package_id,
+        EntityActions.RELEASE_CHANNEL_SET_ACCOUNTS,
+        release_channel=channel,
+        target_accounts=target_accounts.split(","),
+    )
+
+    return MessageResult("Successfully set accounts for the release channel.")
+
+
+@app.command("add-version", requires_connection=True)
+@with_project_definition()
+@force_project_definition_v2()
+def release_channel_add_version(
+    channel: str = typer.Argument(
+        show_default=False,
+        help="The release channel to add a version to.",
+    ),
+    version: str = typer.Option(
+        show_default=False,
+        help="The version to add to the release channel.",
+    ),
+    **options,
+) -> CommandResult:
+    """
+    Adds a version to a release channel.
+    """
+
+    cli_context = get_cli_context()
+    ws = WorkspaceManager(
+        project_definition=cli_context.project_definition,
+        project_root=cli_context.project_root,
+    )
+    package_id = options["package_entity_id"]
+    ws.perform_action(
+        package_id,
+        EntityActions.RELEASE_CHANNEL_ADD_VERSION,
+        release_channel=channel,
+        version=version,
+    )
+
+    return MessageResult(
+        f"Successfully added version {version} to the release channel."
+    )
+
+
+@app.command("remove-version", requires_connection=True)
+@with_project_definition()
+@force_project_definition_v2()
+def release_channel_remove_version(
+    channel: str = typer.Argument(
+        show_default=False,
+        help="The release channel to remove a version from.",
+    ),
+    version: str = typer.Option(
+        show_default=False,
+        help="The version to remove from the release channel.",
+    ),
+    **options,
+) -> CommandResult:
+    """
+    Removes a version from a release channel.
+    """
+
+    cli_context = get_cli_context()
+    ws = WorkspaceManager(
+        project_definition=cli_context.project_definition,
+        project_root=cli_context.project_root,
+    )
+    package_id = options["package_entity_id"]
+    ws.perform_action(
+        package_id,
+        EntityActions.RELEASE_CHANNEL_REMOVE_VERSION,
+        release_channel=channel,
+        version=version,
+    )
+
+    return MessageResult(
+        f"Successfully removed version {version} from the release channel."
+    )

snowflake/cli/_plugins/nativeapp/release_directive/__init__.py

@@ -0,0 +1,13 @@
+# 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.

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

@@ -0,0 +1,243 @@
+# 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 typing import Optional
+
+import typer
+from snowflake.cli._plugins.nativeapp.constants import DEFAULT_CHANNEL
+from snowflake.cli._plugins.nativeapp.v2_conversions.compat import (
+    force_project_definition_v2,
+)
+from snowflake.cli._plugins.workspace.manager import WorkspaceManager
+from snowflake.cli.api.cli_global_context import get_cli_context
+from snowflake.cli.api.commands.decorators import with_project_definition
+from snowflake.cli.api.commands.flags import like_option
+from snowflake.cli.api.commands.snow_typer import SnowTyperFactory
+from snowflake.cli.api.entities.utils import EntityActions
+from snowflake.cli.api.output.types import (
+    CollectionResult,
+    CommandResult,
+    MessageResult,
+)
+
+app = SnowTyperFactory(
+    name="release-directive",
+    help="Manages release directives of an application package",
+)
+
+log = logging.getLogger(__name__)
+
+
+@app.command("list", requires_connection=True)
+@with_project_definition()
+@force_project_definition_v2()
+def release_directive_list(
+    like: str = like_option(
+        help_example="`snow app release-directive list --like='my%'` lists all release directives starting with 'my'",
+    ),
+    channel: Optional[str] = typer.Option(
+        default=None,
+        show_default=False,
+        help="The release channel to use when listing release directives. If not provided, release directives from all release channels are listed.",
+    ),
+    **options,
+) -> CommandResult:
+    """
+    Lists release directives in an application package.
+    If no release channel is specified, release directives for all channels are listed.
+    If a release channel is specified, only release directives for that channel are listed.
+
+    If `--like` is provided, only release directives matching the SQL pattern are listed.
+    """
+
+    cli_context = get_cli_context()
+    ws = WorkspaceManager(
+        project_definition=cli_context.project_definition,
+        project_root=cli_context.project_root,
+    )
+    package_id = options["package_entity_id"]
+    result = ws.perform_action(
+        package_id,
+        EntityActions.RELEASE_DIRECTIVE_LIST,
+        release_channel=channel,
+        like=like,
+    )
+
+    return CollectionResult(result)
+
+
+@app.command("set", requires_connection=True)
+@with_project_definition()
+@force_project_definition_v2()
+def release_directive_set(
+    directive: str = typer.Argument(
+        show_default=False,
+        help="Name of the release directive to set",
+    ),
+    channel: str = typer.Option(
+        DEFAULT_CHANNEL,
+        help="Name of the release channel to use",
+    ),
+    target_accounts: Optional[str] = typer.Option(
+        None,
+        show_default=False,
+        help="List of the accounts to apply the release directive to. Format must be `org1.account1,org2.account2`",
+    ),
+    version: str = typer.Option(
+        show_default=False,
+        help="Version of the application package to use",
+    ),
+    patch: int = typer.Option(
+        show_default=False,
+        help="Patch number to use for the selected version",
+    ),
+    **options,
+) -> CommandResult:
+    """
+    Sets a release directive.
+
+    target_accounts cannot be specified for default release directives.
+    target_accounts field is required when creating a new non-default release directive.
+    """
+
+    cli_context = get_cli_context()
+    ws = WorkspaceManager(
+        project_definition=cli_context.project_definition,
+        project_root=cli_context.project_root,
+    )
+    package_id = options["package_entity_id"]
+    ws.perform_action(
+        package_id,
+        EntityActions.RELEASE_DIRECTIVE_SET,
+        release_directive=directive,
+        version=version,
+        patch=patch,
+        target_accounts=None if target_accounts is None else target_accounts.split(","),
+        release_channel=channel,
+    )
+    return MessageResult("Successfully set release directive.")
+
+
+@app.command("unset", requires_connection=True)
+@with_project_definition()
+@force_project_definition_v2()
+def release_directive_unset(
+    directive: str = typer.Argument(
+        show_default=False,
+        help="Name of the release directive",
+    ),
+    channel: str = typer.Option(
+        DEFAULT_CHANNEL,
+        help="Name of the release channel to use",
+    ),
+    **options,
+) -> CommandResult:
+    """
+    Unsets a release directive.
+    """
+
+    cli_context = get_cli_context()
+    ws = WorkspaceManager(
+        project_definition=cli_context.project_definition,
+        project_root=cli_context.project_root,
+    )
+    package_id = options["package_entity_id"]
+    ws.perform_action(
+        package_id,
+        EntityActions.RELEASE_DIRECTIVE_UNSET,
+        release_directive=directive,
+        release_channel=channel,
+    )
+    return MessageResult(f"Successfully unset release directive {directive}.")
+
+
+@app.command("add-accounts", requires_connection=True)
+@with_project_definition()
+@force_project_definition_v2()
+def release_directive_add_accounts(
+    directive: str = typer.Argument(
+        show_default=False,
+        help="Name of the release directive",
+    ),
+    channel: str = typer.Option(
+        DEFAULT_CHANNEL,
+        help="Name of the release channel to use",
+    ),
+    target_accounts: str = typer.Option(
+        show_default=False,
+        help="List of the accounts to add to the release directive. Format must be `org1.account1,org2.account2`",
+    ),
+    **options,
+) -> CommandResult:
+    """
+    Adds accounts to a release directive.
+    """
+
+    cli_context = get_cli_context()
+    ws = WorkspaceManager(
+        project_definition=cli_context.project_definition,
+        project_root=cli_context.project_root,
+    )
+    package_id = options["package_entity_id"]
+    ws.perform_action(
+        package_id,
+        EntityActions.RELEASE_DIRECTIVE_ADD_ACCOUNTS,
+        release_directive=directive,
+        target_accounts=target_accounts.split(","),
+        release_channel=channel,
+    )
+
+    return MessageResult("Successfully added accounts to the release directive.")
+
+
+@app.command("remove-accounts", requires_connection=True)
+@with_project_definition()
+@force_project_definition_v2()
+def release_directive_remove_accounts(
+    directive: str = typer.Argument(
+        show_default=False,
+        help="Name of the release directive",
+    ),
+    channel: str = typer.Option(
+        DEFAULT_CHANNEL,
+        help="Name of the release channel to use",
+    ),
+    target_accounts: str = typer.Option(
+        show_default=False,
+        help="List of the accounts to remove from the release directive. Format must be `org1.account1,org2.account2`",
+    ),
+    **options,
+) -> CommandResult:
+    """
+    Removes accounts from a release directive.
+    """
+
+    cli_context = get_cli_context()
+    ws = WorkspaceManager(
+        project_definition=cli_context.project_definition,
+        project_root=cli_context.project_root,
+    )
+    package_id = options["package_entity_id"]
+    ws.perform_action(
+        package_id,
+        EntityActions.RELEASE_DIRECTIVE_REMOVE_ACCOUNTS,
+        release_directive=directive,
+        target_accounts=target_accounts.split(","),
+        release_channel=channel,
+    )
+
+    return MessageResult("Successfully removed accounts from the release directive.")

snowflake/cli/_plugins/nativeapp/same_account_install_method.py

@@ -1,3 +1,4 @@
+from dataclasses import dataclass
 from typing import Optional
 
 from snowflake.cli._plugins.nativeapp.constants import (
@@ -8,25 +9,15 @@ from snowflake.cli._plugins.nativeapp.exceptions import (
     ApplicationCreatedExternallyError,
 )
 from snowflake.cli._plugins.stage.manager import StageManager
+from snowflake.cli.api.project.util import to_identifier
 
 
+@dataclass
 class SameAccountInstallMethod:
     _requires_created_by_cli: bool
-    _from_release_directive: bool
-    version: Optional[str]
-    patch: Optional[int]
-
-    def __init__(
-        self,
-        requires_created_by_cli: bool,
-        version: Optional[str] = None,
-        patch: Optional[int] = None,
-        from_release_directive: bool = False,
-    ):
-        self._requires_created_by_cli = requires_created_by_cli
-        self.version = version
-        self.patch = patch
-        self._from_release_directive = from_release_directive
+    version: Optional[str] = None
+    patch: Optional[int] = None
+    _from_release_directive: bool = False
 
     @classmethod
     def unversioned_dev(cls):
@@ -39,7 +30,7 @@ class SameAccountInstallMethod:
 
     @classmethod
     def release_directive(cls):
-        return cls(False, from_release_directive=True)
+        return cls(False, _from_release_directive=True)
 
     @property
     def is_dev_mode(self) -> bool:
@@ -53,8 +44,9 @@ class SameAccountInstallMethod:
             return ""
 
         if self.version:
+            version_clause = f"version {to_identifier(self.version)}"
             patch_clause = f"patch {self.patch}" if self.patch else ""
-            return f"using version {self.version} {patch_clause}"
+            return f"using {version_clause} {patch_clause}"
 
         stage_name = StageManager.quote_stage_name(stage_fqn)
         return f"using {stage_name}"