snowflake-cli-labs 2.8.0rc1__py3-none-any.whl → 3.0.0rc1__py3-none-any.whl

snowflake/cli/api/commands/snow_typer.py

@@ -20,6 +20,7 @@ from functools import wraps
 from typing import Any, Callable, Dict, List, Optional, Tuple
 
 import typer
+from click import ClickException
 from snowflake.cli.api.commands.decorators import (
     global_options,
     global_options_with_connection,
@@ -33,6 +34,7 @@ from snowflake.cli.api.commands.typer_pre_execute import run_pre_execute_command
 from snowflake.cli.api.exceptions import CommandReturnTypeError
 from snowflake.cli.api.output.types import CommandResult
 from snowflake.cli.api.sanitizers import sanitize_for_terminal
+from snowflake.cli.api.sql_execution import SqlExecutionMixin
 
 log = logging.getLogger(__name__)
 
@@ -46,6 +48,7 @@ class SnowTyper(typer.Typer):
             pretty_exceptions_show_locals=False,
             no_args_is_help=True,
             add_completion=True,
+            rich_markup_mode="markdown",
         )
 
     @staticmethod
@@ -70,6 +73,7 @@ class SnowTyper(typer.Typer):
         requires_global_options: bool = True,
         requires_connection: bool = False,
         is_enabled: Callable[[], bool] | None = None,
+        require_warehouse: bool = False,
         **kwargs,
     ):
         """
@@ -96,7 +100,7 @@ class SnowTyper(typer.Typer):
             def command_callable_decorator(*args, **kw):
                 """Wrapper around command callable. This is what happens at "runtime"."""
                 execution = ExecutionMetadata()
-                self.pre_execute(execution)
+                self.pre_execute(execution, require_warehouse=require_warehouse)
                 try:
                     result = command_callable(*args, **kw)
                     self.process_result(result)
@@ -115,22 +119,26 @@ class SnowTyper(typer.Typer):
         return custom_command
 
     @staticmethod
-    def pre_execute(execution: ExecutionMetadata):
+    def pre_execute(execution: ExecutionMetadata, require_warehouse: bool = False):
         """
         Callback executed before running any command callable (after context execution).
         Pay attention to make this method safe to use if performed operations are not necessary
         for executing the command in proper way.
         """
-        from snowflake.cli.app.telemetry import log_command_usage
+        from snowflake.cli._app.telemetry import log_command_usage
 
         log.debug("Executing command pre execution callback")
         run_pre_execute_commands()
         log_command_usage(execution)
+        if require_warehouse and not SqlExecutionMixin().session_has_warehouse():
+            raise ClickException(
+                "The command requires warehouse. No warehouse found in current connection."
+            )
 
     @staticmethod
     def process_result(result):
         """Command result processor"""
-        from snowflake.cli.app.printing import print_result
+        from snowflake.cli._app.printing import print_result
 
         # Because we still have commands like "logs" that do not return anything.
         # We should improve it in future.
@@ -145,7 +153,7 @@ class SnowTyper(typer.Typer):
         """
         Callback executed on command execution error.
         """
-        from snowflake.cli.app.telemetry import log_command_execution_error
+        from snowflake.cli._app.telemetry import log_command_execution_error
 
         log.debug("Executing command exception callback")
         log_command_execution_error(exception, execution)
@@ -156,7 +164,7 @@ class SnowTyper(typer.Typer):
         Callback executed after running any command callable. Pay attention to make this method safe to
         use if performed operations are not necessary for executing the command in proper way.
         """
-        from snowflake.cli.app.telemetry import flush_telemetry, log_command_result
+        from snowflake.cli._app.telemetry import flush_telemetry, log_command_result
 
         log.debug("Executing command post execution callback")
         log_command_result(execution)

snowflake/cli/api/commands/typer_pre_execute.py

@@ -14,13 +14,13 @@
 
 from typing import Callable
 
-from snowflake.cli.api.cli_global_context import cli_context_manager
+from snowflake.cli.api.cli_global_context import get_cli_context_manager
 
 
 def register_pre_execute_command(command: Callable[[], None]) -> None:
-    cli_context_manager.add_typer_pre_execute_commands(command)
+    get_cli_context_manager().add_typer_pre_execute_commands(command)
 
 
 def run_pre_execute_commands() -> None:
-    for command in cli_context_manager.typer_pre_execute_commands:
+    for command in get_cli_context_manager().typer_pre_execute_commands:
         command()

snowflake/cli/api/commands/utils.py

@@ -0,0 +1,18 @@
+from typing import List, Optional
+
+from click import ClickException
+from snowflake.cli.api.commands.common import Variable
+
+
+def parse_key_value_variables(variables: Optional[List[str]]) -> List[Variable]:
+    """Util for parsing key=value input. Useful for commands accepting multiple input options."""
+    result: List[Variable] = []
+    if not variables:
+        return 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/api/config.py

@@ -30,7 +30,10 @@ from snowflake.cli.api.exceptions import (
     UnsupportedConfigSectionTypeError,
 )
 from snowflake.cli.api.secure_path import SecurePath
-from snowflake.cli.api.secure_utils import file_permissions_are_strict
+from snowflake.cli.api.secure_utils import (
+    file_permissions_are_strict,
+    windows_get_not_whitelisted_users_with_access,
+)
 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
@@ -77,7 +80,7 @@ class ConnectionConfig:
     warehouse: Optional[str] = None
     role: Optional[str] = None
     authenticator: Optional[str] = None
-    private_key_path: Optional[str] = None
+    private_key_file: Optional[str] = None
     token_file_path: Optional[str] = None
 
     _other_settings: dict = field(default_factory=lambda: {})
@@ -115,7 +118,7 @@ def config_init(config_file: Optional[Path]):
     Initializes the app configuration. Config provided via cli flag takes precedence.
     If config file does not exist we create an empty one.
     """
-    from snowflake.cli.app.loggers import create_initial_loggers
+    from snowflake.cli._app.loggers import create_initial_loggers
 
     if config_file:
         CONFIG_MANAGER.file_path = config_file
@@ -160,6 +163,18 @@ def _read_config_file():
                 message="Bad owner or permissions.*",
                 module="snowflake.connector.config_manager",
             )
+
+            if not file_permissions_are_strict(CONFIG_MANAGER.file_path):
+                users = ", ".join(
+                    windows_get_not_whitelisted_users_with_access(
+                        CONFIG_MANAGER.file_path
+                    )
+                )
+                warnings.warn(
+                    f"Unauthorized users ({users}) have access to configuration file {CONFIG_MANAGER.file_path}.\n"
+                    f'Run `icacls "{CONFIG_MANAGER.file_path}" /deny <USER_ID>:F` on those users to restrict permissions.'
+                )
+
         try:
             CONFIG_MANAGER.read_config()
         except ConfigSourceError as exception:
@@ -322,8 +337,6 @@ def _dump_config(conf_file_cache: Dict):
 
 
 def _check_default_config_files_permissions() -> None:
-    if IS_WINDOWS:
-        return
     if CONNECTIONS_FILE.exists() and not file_permissions_are_strict(CONNECTIONS_FILE):
         raise ConfigFileTooWidePermissionsError(CONNECTIONS_FILE)
     if CONFIG_FILE.exists() and not file_permissions_are_strict(CONFIG_FILE):

snowflake/cli/api/console/abc.py

@@ -20,7 +20,10 @@ 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
+from snowflake.cli.api.cli_global_context import (
+    _CliGlobalContextAccess,
+    get_cli_context,
+)
 
 
 class AbstractConsole(ABC):
@@ -39,7 +42,7 @@ class AbstractConsole(ABC):
 
     def __init__(self):
         super().__init__()
-        self._cli_context = cli_context
+        self._cli_context = get_cli_context()
         self._in_phase = False
 
     @property

snowflake/cli/api/constants.py

@@ -77,3 +77,14 @@ VALID_SCOPES = ["database", "schema", "compute-pool"]
 DEFAULT_SIZE_LIMIT_MB = 128
 
 SF_REST_API_URL_PREFIX = "/api/v2"
+
+PROJECT_TEMPLATE_VARIABLE_OPENING = "<%"
+PROJECT_TEMPLATE_VARIABLE_CLOSING = "%>"
+
+INIT_TEMPLATE_VARIABLE_OPENING = "<!"
+INIT_TEMPLATE_VARIABLE_CLOSING = "!>"
+
+SNOWPARK_SHARED_MIXIN = "snowpark_shared"
+
+DEFAULT_ENV_FILE = "environment.yml"
+DEFAULT_PAGES_DIR = "pages"

snowflake/cli/api/entities/application_entity.py

@@ -0,0 +1,12 @@
+from snowflake.cli.api.entities.common import EntityBase
+from snowflake.cli.api.project.schemas.entities.application_entity_model import (
+    ApplicationEntityModel,
+)
+
+
+class ApplicationEntity(EntityBase[ApplicationEntityModel]):
+    """
+    A Native App application object, created from an application package.
+    """
+
+    pass