diff --git a/.changes/unreleased/Features-20240812-161625.yaml b/.changes/unreleased/Features-20240812-161625.yaml new file mode 100644 index 00000000..51d21829 --- /dev/null +++ b/.changes/unreleased/Features-20240812-161625.yaml @@ -0,0 +1,7 @@ +kind: Features +body: Added update_type and rowcount to dbt commands SQL status in CLI output +time: 2024-08-12T16:16:25.330055+02:00 +custom: + Author: damian3031 + Issue: "428" + PR: "429" diff --git a/dbt/adapters/trino/connections.py b/dbt/adapters/trino/connections.py index 28b0e432..f2d49ad9 100644 --- a/dbt/adapters/trino/connections.py +++ b/dbt/adapters/trino/connections.py @@ -493,12 +493,20 @@ def open(cls, connection): @classmethod def get_response(cls, cursor) -> TrinoAdapterResponse: - message = "SUCCESS" + code = cursor._cursor.update_type + if code is None: + code = "SUCCESS" + + rows_affected = cursor._cursor.rowcount + if rows_affected == -1: + message = f"{code}" + else: + message = f"{code} ({rows_affected:_} rows)" return TrinoAdapterResponse( _message=message, query=cursor._cursor.query, query_id=cursor._cursor.query_id, - rows_affected=cursor._cursor.rowcount, + rows_affected=rows_affected, ) # type: ignore def cancel(self, connection):