add column comment to get_columns method

cockroachdb · Sep 9, 2024 · ef4ff2d · ef4ff2d
1 parent 726b05d
commit ef4ff2d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/sqlalchemy_cockroachdb/base.py b/sqlalchemy_cockroachdb/base.py
@@ -185,7 +185,7 @@ def get_columns(self, conn, table_name, schema=None, **kw):
                 "SELECT column_name, data_type, is_nullable::bool, column_default,"
                 "numeric_precision, numeric_scale, character_maximum_length, "
                 "NULL AS is_generated, NULL AS generation_expression, is_hidden::bool,"
-                "comment "
+                "column_comment AS comment "
                 "FROM information_schema.columns "
                 "WHERE table_schema = :table_schema AND table_name = :table_name "
             )
@@ -201,7 +201,7 @@ def get_columns(self, conn, table_name, schema=None, **kw):
                 "numeric_precision, numeric_scale, character_maximum_length, "
                 "CASE is_generated WHEN 'ALWAYS' THEN true WHEN 'NEVER' THEN false "
                 "ELSE is_generated::bool END AS is_generated, "
-                "generation_expression, is_hidden::bool, crdb_sql_type, comment "
+                "generation_expression, is_hidden::bool, crdb_sql_type, column_comment AS comment "
                 "FROM information_schema.columns "
                 "WHERE table_schema = :table_schema AND table_name = :table_name "
             )