snowflake-cli-labs 2.8.1__py3-none-any.whl → 2.8.2__py3-none-any.whl

snowflake/cli/api/console/abc.py

@@ -1,89 +0,0 @@
-# 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
-
-from abc import ABC, abstractmethod
-from contextlib import contextmanager
-from typing import Callable, Iterator, Optional
-
-from rich import print as rich_print
-from rich.text import Text
-from snowflake.cli.api.cli_global_context import _CliGlobalContextAccess, cli_context
-
-
-class AbstractConsole(ABC):
-    """Interface for cli console implementation.
-
-    Each console should have three methods implemented:
-    - `step` - for more detailed information on steps
-    - `warning` - for displaying messages in a style that makes it
-      visually stand out from other output
-    - `phase` a context manager for organising steps into logical group
-    """
-
-    _print_fn: Callable[[str], None]
-    _cli_context: _CliGlobalContextAccess
-    _in_phase: bool
-
-    def __init__(self):
-        super().__init__()
-        self._cli_context = cli_context
-        self._in_phase = False
-
-    @property
-    def is_silent(self) -> bool:
-        """Returns information whether intermediate output is muted."""
-        return self._cli_context.silent
-
-    @property
-    def in_phase(self) -> bool:
-        """Indicated whether output should be grouped."""
-        return self._in_phase
-
-    def _print(self, text: Text):
-        if self.is_silent:
-            return
-        rich_print(text)
-
-    @contextmanager
-    @abstractmethod
-    def phase(
-        self,
-        enter_message: str,
-        exit_message: Optional[str] = None,
-    ) -> Iterator[Callable[[str], None]]:
-        """A context manager for organising steps into logical group."""
-
-    @contextmanager
-    @abstractmethod
-    def indented(self):
-        """
-        A context manager for temporarily indenting messages and warnings. Phases and steps cannot be used in indented blocks,
-        but multiple indented blocks can be nested (use sparingly).
-        """
-
-    @abstractmethod
-    def step(self, message: str):
-        """Displays a message to output."""
-
-    @abstractmethod
-    def message(self, _message: str):
-        """Displays an informational message to output."""
-
-    @abstractmethod
-    def warning(self, message: str):
-        """Displays message in a style that makes it visually stand out from other output.
-
-        Intended for displaying messages related to important messages."""

snowflake/cli/api/console/console.py

@@ -1,134 +0,0 @@
-# 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
-
-from contextlib import contextmanager
-from typing import Optional
-
-from rich.style import Style
-from rich.text import Text
-from snowflake.cli.api.console.abc import AbstractConsole
-from snowflake.cli.api.console.enum import Output
-
-PHASE_STYLE: Style = Style(bold=True)
-STEP_STYLE: Style = Style(italic=True)
-INFO_STYLE: Style = Style()
-IMPORTANT_STYLE: Style = Style(bold=True, italic=True)
-INDENTATION_LEVEL: int = 2
-
-
-class CliConsoleNestingProhibitedError(RuntimeError):
-    """CliConsole phase nesting not allowed."""
-
-
-class CliConsole(AbstractConsole):
-    """An utility for displaying intermediate output.
-
-    Provides following methods for handling displaying messages:
-    - `step` - for more detailed information on steps
-    - `warning` - for displaying messages in a style that makes it
-      visually stand out from other output
-    - `phase` a context manager for organising steps into logical group
-    """
-
-    _indentation_level: int = INDENTATION_LEVEL
-    _extra_indent: int = 0
-    _styles: dict = {
-        "default": "",
-        Output.PHASE: PHASE_STYLE,
-        Output.STEP: STEP_STYLE,
-        Output.INFO: None,
-        Output.IMPORTANT: IMPORTANT_STYLE,
-    }
-
-    def _format_message(self, message: str, output: Output) -> Text:
-        """Wraps message in rich Text object and applies formatting."""
-        style = self._styles.get(output, "default")
-        if style is not None:
-            text = Text(message, style=style)
-        else:
-            text = Text.from_markup(message)
-
-        current_indent = self._extra_indent
-        if self.in_phase and output in {Output.STEP, Output.INFO, Output.IMPORTANT}:
-            current_indent += 1
-        text.pad_left(current_indent * self._indentation_level)
-        return text
-
-    @contextmanager
-    def phase(self, enter_message: str, exit_message: Optional[str] = None):
-        """A context manager for organising steps into logical group."""
-        if self.in_phase:
-            raise CliConsoleNestingProhibitedError("Only one phase allowed at a time.")
-        if self._extra_indent > 0:
-            raise CliConsoleNestingProhibitedError(
-                "Phase cannot be used in an indented block."
-            )
-
-        self._print(self._format_message(enter_message, Output.PHASE))
-        self._in_phase = True
-
-        try:
-            yield self.step
-        finally:
-            self._in_phase = False
-            if exit_message:
-                self._print(self._format_message(exit_message, Output.PHASE))
-
-    @contextmanager
-    def indented(self):
-        """
-        A context manager for temporarily indenting messages and warnings. Phases and steps cannot be used in indented blocks,
-        but multiple indented blocks can be nested (use sparingly).
-        """
-        self._extra_indent += 1
-        try:
-            yield
-        finally:
-            self._extra_indent -= 1
-
-    def step(self, message: str):
-        """Displays a message to output.
-
-        If called within a phase, the output will be indented.
-        """
-        if self._extra_indent > 0:
-            raise CliConsoleNestingProhibitedError(
-                "Step cannot be used in an indented block."
-            )
-        text = self._format_message(message, Output.STEP)
-        self._print(text)
-
-    def message(self, _message: str):
-        """Displays an informational message to output.
-
-        If called within a phase, the output will be indented."""
-        text = self._format_message(_message, Output.INFO)
-        self._print(text)
-
-    def warning(self, message: str):
-        """Displays message in a style that makes it visually stand out from other output.
-
-        This should be used to display important messages to the console."""
-        text = self._format_message(message, Output.IMPORTANT)
-        self._print(text)
-
-
-def get_cli_console() -> AbstractConsole:
-    console = CliConsole()
-    return console
-
-
-cli_console = get_cli_console()

snowflake/cli/api/console/enum.py

@@ -1,17 +0,0 @@
-# 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 enum import Enum
-
-Output = Enum("Output", ("PHASE", "STEP", "INFO", "IMPORTANT"))

snowflake/cli/api/constants.py

@@ -1,79 +0,0 @@
-# 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
-
-from dataclasses import dataclass
-from enum import Enum
-from pathlib import Path
-
-TEMPLATES_PATH = Path(__file__).parent.parent / "templates"
-DEPLOYMENT_STAGE = "deployments"
-PYTHON_3_12 = (3, 12)
-
-
-@dataclass(frozen=True)
-class ObjectNames:
-    cli_name: str
-    sf_name: str
-    sf_plural_name: str
-
-    def __str__(self):
-        return self.sf_name
-
-
-class ObjectType(Enum):
-    COMPUTE_POOL = ObjectNames("compute-pool", "compute pool", "compute pools")
-    DATABASE = ObjectNames("database", "database", "databases")
-    FUNCTION = ObjectNames("function", "function", "functions")
-    INTEGRATION = ObjectNames("integration", "integration", "integrations")
-    EXTERNAL_ACCESS_INTEGRATION = ObjectNames(
-        "external-access-integration",
-        "external access integration",
-        "external access integrations",
-    )
-    # JOB = ObjectNames("job", "job", "jobs")
-    NETWORK_RULE = ObjectNames("network-rule", "network rule", "network rules")
-    PROCEDURE = ObjectNames("procedure", "procedure", "procedures")
-    ROLE = ObjectNames("role", "role", "roles")
-    SCHEMA = ObjectNames("schema", "schema", "schemas")
-    SERVICE = ObjectNames("service", "service", "services")
-    SECRET = ObjectNames("secret", "secret", "secrets")
-    STAGE = ObjectNames("stage", "stage", "stages")
-    STREAM = ObjectNames("stream", "stream", "streams")
-    STREAMLIT = ObjectNames("streamlit", "streamlit", "streamlits")
-    TABLE = ObjectNames("table", "table", "tables")
-    TASK = ObjectNames("task", "task", "tasks")
-    USER = ObjectNames("user", "user", "users")
-    WAREHOUSE = ObjectNames("warehouse", "warehouse", "warehouses")
-    VIEW = ObjectNames("view", "view", "views")
-    IMAGE_REPOSITORY = ObjectNames(
-        "image-repository", "image repository", "image repositories"
-    )
-    GIT_REPOSITORY = ObjectNames("git-repository", "git repository", "git repositories")
-
-    def __str__(self):
-        """This makes using this Enum easier in formatted string"""
-        return self.value.cli_name
-
-
-OBJECT_TO_NAMES = {o.value.cli_name: o.value for o in ObjectType}
-SUPPORTED_OBJECTS = sorted(OBJECT_TO_NAMES.keys())
-
-# Scope names here must replace spaces with '-'. For example 'compute pool' is 'compute-pool'.
-VALID_SCOPES = ["database", "schema", "compute-pool"]
-
-DEFAULT_SIZE_LIMIT_MB = 128
-
-SF_REST_API_URL_PREFIX = "/api/v2"

snowflake/cli/api/errno.py

@@ -1,27 +0,0 @@
-# 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.
-
-# General errors
-NO_WAREHOUSE_SELECTED_IN_SESSION = 606
-
-DOES_NOT_EXIST_OR_NOT_AUTHORIZED = 2003
-DOES_NOT_EXIST_OR_CANNOT_BE_PERFORMED = 2043
-
-# Native Apps
-CANNOT_UPGRADE_FROM_LOOSE_FILES_TO_VERSION = 93044
-CANNOT_UPGRADE_FROM_VERSION_TO_LOOSE_FILES = 93045
-ONLY_SUPPORTED_ON_DEV_MODE_APPLICATIONS = 93046
-NOT_SUPPORTED_ON_DEV_MODE_APPLICATIONS = 93055
-APPLICATION_NO_LONGER_AVAILABLE = 93079
-APPLICATION_OWNS_EXTERNAL_OBJECTS = 93128

snowflake/cli/api/exceptions.py

@@ -1,164 +0,0 @@
-# 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
-
-from pathlib import Path
-from typing import Optional
-
-from click.exceptions import ClickException
-from snowflake.cli.api.constants import ObjectType
-
-
-class EnvironmentVariableNotFoundError(ClickException):
-    def __init__(self, env_variable_name: str):
-        super().__init__(f"Environment variable {env_variable_name} not found")
-
-
-class MissingConfiguration(ClickException):
-    pass
-
-
-class CycleDetectedError(ClickException):
-    pass
-
-
-class InvalidTemplate(ClickException):
-    pass
-
-
-class InvalidConnectionConfiguration(ClickException):
-    def format_message(self):
-        return f"Invalid connection configuration. {self.message}"
-
-
-class InvalidLogsConfiguration(ClickException):
-    def format_message(self):
-        return f"Invalid logs configuration. {self.message}"
-
-
-class InvalidPluginConfiguration(ClickException):
-    def format_message(self):
-        return f"Invalid plugin configuration. {self.message}"
-
-
-class SnowflakeConnectionError(ClickException):
-    def __init__(self, snowflake_err: Exception):
-        super().__init__(f"Could not connect to Snowflake. Reason: {snowflake_err}")
-
-
-class UnsupportedConfigSectionTypeError(Exception):
-    def __init__(self, section_type: type):
-        super().__init__(f"Unsupported configuration section type {section_type}")
-
-
-class OutputDataTypeError(ClickException):
-    def __init__(self, got_type: type, expected_type: type):
-        super().__init__(f"Got {got_type} type but expected {expected_type}")
-
-
-class CommandReturnTypeError(ClickException):
-    def __init__(self, got_type: type):
-        super().__init__(f"Commands have to return OutputData type, but got {got_type}")
-
-
-class SnowflakeSQLExecutionError(ClickException):
-    """
-    Could not successfully execute the Snowflake SQL statements.
-    """
-
-    def __init__(self, queries: Optional[str] = None):
-        super().__init__(
-            f"""
-                {self.__doc__}
-                {queries if queries else ""}
-            """
-        )
-
-
-class ObjectAlreadyExistsError(ClickException):
-    def __init__(
-        self,
-        object_type: ObjectType,
-        name: str,
-        replace_available: bool = False,
-    ):
-        msg = f"{str(object_type).capitalize()} {name} already exists."
-        if replace_available:
-            msg += " Use --replace flag to update objects."
-        super().__init__(msg)
-
-
-class NoProjectDefinitionError(ClickException):
-    def __init__(self, project_type: str, project_file: str):
-        super().__init__(
-            f"No {project_type} project definition found in {project_file}"
-        )
-
-
-class InvalidSchemaError(ClickException):
-    def __init__(self, schema: str):
-        super().__init__(f"Invalid schema {schema}")
-
-
-class SecretsWithoutExternalAccessIntegrationError(ClickException):
-    def __init__(self, object_name: str):
-        super().__init__(
-            f"{object_name} defined with secrets but without external integration."
-        )
-
-
-class FileTooLargeError(ClickException):
-    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):
-    def __init__(self, path: Path):
-        super().__init__(f"Directory '{path}' is not empty")
-
-
-class ConfigFileTooWidePermissionsError(ClickException):
-    def __init__(self, path: Path):
-        super().__init__(
-            f'Configuration file {path} has too wide permissions, run `chmod 0600 "{path}"`'
-        )
-
-
-class DatabaseNotProvidedError(ClickException):
-    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):
-    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):
-    def __init__(self, name: str):
-        super().__init__(f"Specified name '{name}' is not valid name.")
-
-
-class FQNInconsistencyError(ClickException):
-    def __init__(self, part: str, name: str):
-        super().__init__(
-            f"{part.capitalize()} provided but name '{name}' is fully qualified name."
-        )

snowflake/cli/api/feature_flags.py

@@ -1,55 +0,0 @@
-# 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 enum import Enum, unique
-from typing import NamedTuple
-
-from snowflake.cli.api.config import (
-    FEATURE_FLAGS_SECTION_PATH,
-    get_config_bool_value,
-    get_env_variable_name,
-)
-
-
-class BooleanFlag(NamedTuple):
-    name: str
-    default: bool = False
-
-
-@unique
-class FeatureFlagMixin(Enum):
-    def is_enabled(self) -> bool:
-        return get_config_bool_value(
-            *FEATURE_FLAGS_SECTION_PATH,
-            key=self.value.name.lower(),
-            default=self.value.default,
-        )
-
-    def is_disabled(self):
-        return not self.is_enabled()
-
-    def env_variable(self):
-        return get_env_variable_name(*FEATURE_FLAGS_SECTION_PATH, key=self.value.name)
-
-
-@unique
-class FeatureFlag(FeatureFlagMixin):
-    ENABLE_STREAMLIT_EMBEDDED_STAGE = BooleanFlag(
-        "ENABLE_STREAMLIT_EMBEDDED_STAGE", False
-    )
-    ENABLE_STREAMLIT_NO_CHECKOUTS = BooleanFlag("ENABLE_STREAMLIT_NO_CHECKOUTS", False)
-    ENABLE_STREAMLIT_VERSIONED_STAGE = BooleanFlag(
-        "ENABLE_STREAMLIT_VERSIONED_STAGE", False
-    )
-    ENABLE_PROJECT_DEFINITION_V2 = BooleanFlag("ENABLE_PROJECT_DEFINITION_V2", False)