Skip to content

Commit

Permalink
feat: [SNOW-1890085] implement dbt deploy command
Browse files Browse the repository at this point in the history
  • Loading branch information
sfc-gh-jwilkowski committed Feb 21, 2025
1 parent b996a9b commit 905d993
Show file tree
Hide file tree
Showing 2 changed files with 15 additions and 1 deletion.
15 changes: 14 additions & 1 deletion src/snowflake/cli/_plugins/dbt/manager.py
Original file line number Diff line number Diff line change
Expand Up @@ -19,12 +19,25 @@
from click import ClickException
from snowflake.cli._plugins.stage.manager import StageManager
from snowflake.cli.api.console import cli_console
from snowflake.cli.api.feature_flags import FeatureFlag
from snowflake.cli.api.identifiers import FQN
from snowflake.cli.api.sql_execution import SqlExecutionMixin
from snowflake.connector.cursor import SnowflakeCursor


class DBTManager(SqlExecutionMixin):
class StdoutExecutionMixin(SqlExecutionMixin):
def execute_query(self, query, **kwargs):
if FeatureFlag.ENABLE_DBT_POC.is_enabled():
from unittest.mock import MagicMock

cli_console.message(f"Sending query: {query}")
mock_cursor = MagicMock()
mock_cursor.description = []
return mock_cursor
return super().execute_query(query, **kwargs)


class DBTManager(StdoutExecutionMixin):
def list(self) -> SnowflakeCursor: # noqa: A003
query = "SHOW DBT"
return self.execute_query(query)
Expand Down
1 change: 1 addition & 0 deletions src/snowflake/cli/api/feature_flags.py
Original file line number Diff line number Diff line change
Expand Up @@ -68,3 +68,4 @@ class FeatureFlag(FeatureFlagMixin):
)
ENABLE_SNOWPARK_GLOB_SUPPORT = BooleanFlag("ENABLE_SNOWPARK_GLOB_SUPPORT", False)
ENABLE_SPCS_SERVICE_EVENTS = BooleanFlag("ENABLE_SPCS_SERVICE_EVENTS", False)
ENABLE_DBT_POC = BooleanFlag("ENABLE_DBT_POC", False)

0 comments on commit 905d993

Please sign in to comment.