Skip to content

Commit 7d3174f

Browse files
make stuff missing from SEA None in SEA mapping
Signed-off-by: varun-edachali-dbx <varun.edachali@databricks.com>
1 parent 939c542 commit 7d3174f

File tree

1 file changed

+16
-30
lines changed

1 file changed

+16
-30
lines changed

src/databricks/sql/backend/sea/utils/metadata_mappings.py

Lines changed: 16 additions & 30 deletions
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@ class MetadataColumnMappings:
99
"TABLE_CAT", "catalog", SqlType.VARCHAR
1010
)
1111

12-
CATALOG_FULL_COLUMN = ResultColumn("TABLE_CATALOG", "catalogName", SqlType.VARCHAR)
12+
CATALOG_FULL_COLUMN = ResultColumn("TABLE_CATALOG", None, SqlType.VARCHAR)
1313
SCHEMA_COLUMN_FOR_GET_SCHEMA = ResultColumn(
1414
"TABLE_SCHEM", "databaseName", SqlType.VARCHAR
1515
)
@@ -19,22 +19,16 @@ class MetadataColumnMappings:
1919
TABLE_NAME_COLUMN = ResultColumn("TABLE_NAME", "tableName", SqlType.VARCHAR)
2020
TABLE_TYPE_COLUMN = ResultColumn("TABLE_TYPE", "tableType", SqlType.VARCHAR)
2121
REMARKS_COLUMN = ResultColumn("REMARKS", "remarks", SqlType.VARCHAR)
22-
TYPE_CATALOG_COLUMN = ResultColumn(
23-
"TYPE_CAT", "TYPE_CATALOG_COLUMN", SqlType.VARCHAR
24-
)
25-
TYPE_SCHEM_COLUMN = ResultColumn(
26-
"TYPE_SCHEM", "TYPE_SCHEMA_COLUMN", SqlType.VARCHAR
27-
)
28-
TYPE_NAME_COLUMN = ResultColumn("TYPE_NAME", "TYPE_NAME", SqlType.VARCHAR)
22+
TYPE_CATALOG_COLUMN = ResultColumn("TYPE_CAT", None, SqlType.VARCHAR)
23+
TYPE_SCHEM_COLUMN = ResultColumn("TYPE_SCHEM", None, SqlType.VARCHAR)
24+
TYPE_NAME_COLUMN = ResultColumn("TYPE_NAME", None, SqlType.VARCHAR)
2925
SELF_REFERENCING_COL_NAME_COLUMN = ResultColumn(
30-
"SELF_REFERENCING_COL_NAME", "SELF_REFERENCING_COLUMN_NAME", SqlType.VARCHAR
31-
)
32-
REF_GENERATION_COLUMN = ResultColumn(
33-
"REF_GENERATION", "REF_GENERATION_COLUMN", SqlType.VARCHAR
26+
"SELF_REFERENCING_COL_NAME", None, SqlType.VARCHAR
3427
)
28+
REF_GENERATION_COLUMN = ResultColumn("REF_GENERATION", None, SqlType.VARCHAR)
3529

3630
COL_NAME_COLUMN = ResultColumn("COLUMN_NAME", "col_name", SqlType.VARCHAR)
37-
DATA_TYPE_COLUMN = ResultColumn("DATA_TYPE", "dataType", SqlType.INT)
31+
DATA_TYPE_COLUMN = ResultColumn("DATA_TYPE", None, SqlType.INT)
3832
COLUMN_TYPE_COLUMN = ResultColumn("TYPE_NAME", "columnType", SqlType.VARCHAR)
3933
COLUMN_SIZE_COLUMN = ResultColumn("COLUMN_SIZE", "columnSize", SqlType.INT)
4034
BUFFER_LENGTH_COLUMN = ResultColumn("BUFFER_LENGTH", None, SqlType.INT)
@@ -51,25 +45,17 @@ class MetadataColumnMappings:
5145
SqlType.INT,
5246
)
5347

54-
NULLABLE_COLUMN = ResultColumn("NULLABLE", "Nullable", SqlType.INT)
55-
COLUMN_DEF_COLUMN = ResultColumn("COLUMN_DEF", "columnType", SqlType.VARCHAR)
56-
SQL_DATA_TYPE_COLUMN = ResultColumn("SQL_DATA_TYPE", "SQLDataType", SqlType.INT)
57-
SQL_DATETIME_SUB_COLUMN = ResultColumn(
58-
"SQL_DATETIME_SUB", "SQLDateTimeSub", SqlType.INT
59-
)
60-
CHAR_OCTET_LENGTH_COLUMN = ResultColumn(
61-
"CHAR_OCTET_LENGTH", "CharOctetLength", SqlType.INT
62-
)
48+
NULLABLE_COLUMN = ResultColumn("NULLABLE", None, SqlType.INT)
49+
COLUMN_DEF_COLUMN = ResultColumn("COLUMN_DEF", None, SqlType.VARCHAR)
50+
SQL_DATA_TYPE_COLUMN = ResultColumn("SQL_DATA_TYPE", None, SqlType.INT)
51+
SQL_DATETIME_SUB_COLUMN = ResultColumn("SQL_DATETIME_SUB", None, SqlType.INT)
52+
CHAR_OCTET_LENGTH_COLUMN = ResultColumn("CHAR_OCTET_LENGTH", None, SqlType.INT)
6353
IS_NULLABLE_COLUMN = ResultColumn("IS_NULLABLE", "isNullable", SqlType.VARCHAR)
6454

65-
SCOPE_CATALOG_COLUMN = ResultColumn(
66-
"SCOPE_CATALOG", "ScopeCatalog", SqlType.VARCHAR
67-
)
68-
SCOPE_SCHEMA_COLUMN = ResultColumn("SCOPE_SCHEMA", "ScopeSchema", SqlType.VARCHAR)
69-
SCOPE_TABLE_COLUMN = ResultColumn("SCOPE_TABLE", "ScopeTable", SqlType.VARCHAR)
70-
SOURCE_DATA_TYPE_COLUMN = ResultColumn(
71-
"SOURCE_DATA_TYPE", "SourceDataType", SqlType.INT
72-
)
55+
SCOPE_CATALOG_COLUMN = ResultColumn("SCOPE_CATALOG", None, SqlType.VARCHAR)
56+
SCOPE_SCHEMA_COLUMN = ResultColumn("SCOPE_SCHEMA", None, SqlType.VARCHAR)
57+
SCOPE_TABLE_COLUMN = ResultColumn("SCOPE_TABLE", None, SqlType.VARCHAR)
58+
SOURCE_DATA_TYPE_COLUMN = ResultColumn("SOURCE_DATA_TYPE", None, SqlType.INT)
7359

7460
IS_AUTO_INCREMENT_COLUMN = ResultColumn(
7561
"IS_AUTOINCREMENT", "isAutoIncrement", SqlType.VARCHAR

0 commit comments

Comments
 (0)