Skip to content

Commit 905d993

Browse files
feat: [SNOW-1890085] implement dbt deploy command
1 parent b996a9b commit 905d993

File tree

2 files changed

+15
-1
lines changed

2 files changed

+15
-1
lines changed

src/snowflake/cli/_plugins/dbt/manager.py

Lines changed: 14 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,12 +19,25 @@
1919
from click import ClickException
2020
from snowflake.cli._plugins.stage.manager import StageManager
2121
from snowflake.cli.api.console import cli_console
22+
from snowflake.cli.api.feature_flags import FeatureFlag
2223
from snowflake.cli.api.identifiers import FQN
2324
from snowflake.cli.api.sql_execution import SqlExecutionMixin
2425
from snowflake.connector.cursor import SnowflakeCursor
2526

2627

27-
class DBTManager(SqlExecutionMixin):
28+
class StdoutExecutionMixin(SqlExecutionMixin):
29+
def execute_query(self, query, **kwargs):
30+
if FeatureFlag.ENABLE_DBT_POC.is_enabled():
31+
from unittest.mock import MagicMock
32+
33+
cli_console.message(f"Sending query: {query}")
34+
mock_cursor = MagicMock()
35+
mock_cursor.description = []
36+
return mock_cursor
37+
return super().execute_query(query, **kwargs)
38+
39+
40+
class DBTManager(StdoutExecutionMixin):
2841
def list(self) -> SnowflakeCursor: # noqa: A003
2942
query = "SHOW DBT"
3043
return self.execute_query(query)

src/snowflake/cli/api/feature_flags.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -68,3 +68,4 @@ class FeatureFlag(FeatureFlagMixin):
6868
)
6969
ENABLE_SNOWPARK_GLOB_SUPPORT = BooleanFlag("ENABLE_SNOWPARK_GLOB_SUPPORT", False)
7070
ENABLE_SPCS_SERVICE_EVENTS = BooleanFlag("ENABLE_SPCS_SERVICE_EVENTS", False)
71+
ENABLE_DBT_POC = BooleanFlag("ENABLE_DBT_POC", False)

0 commit comments

Comments
 (0)