feat: [SNOW-1890085] wip: capture success code from server

Author: sfc-gh-jwilkowski

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

@@ -18,6 +18,7 @@
 from typing import Optional
 
 import typer
+from click import ClickException
 from rich.progress import Progress, SpinnerColumn, TextColumn
 from snowflake.cli._plugins.dbt.constants import DBT_COMMANDS
 from snowflake.cli._plugins.dbt.manager import DBTManager
@@ -33,7 +34,6 @@
     CommandResult,
     MessageResult,
     QueryResult,
-    SingleQueryResult,
 )
 from snowflake.cli.api.secure_path import SecurePath
 
@@ -148,5 +148,16 @@ def _dbt_execute(
             transient=True,
         ) as progress:
             progress.add_task(description=f"Executing 'dbt {dbt_command}'", total=None)
+
             result = dbt_manager.execute(*execute_args)
-            return SingleQueryResult(result)
+            columns = [column.name for column in result.description]
+            success_column_index = columns.index("SUCCESS")
+            stdout_column_index = columns.index("STDOUT")
+            is_success, output = [
+                (row[success_column_index], row[stdout_column_index]) for row in result
+            ][-1]
+
+            if is_success is True:
+                return MessageResult(output)
+            else:
+                raise ClickException(output)

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

@@ -69,7 +69,9 @@ def deploy(
 
             return self.execute_query(query)
 
-    def execute(self, dbt_command: str, name: str, run_async: bool, *dbt_cli_args):
+    def execute(
+        self, dbt_command: str, name: str, run_async: bool, *dbt_cli_args
+    ) -> SnowflakeCursor:
         if dbt_cli_args:
             dbt_command = " ".join([dbt_command, *dbt_cli_args]).strip()
         query = f"EXECUTE DBT PROJECT {name} args='{dbt_command}'"