From 8b7bb16f091ab7bd8456f692c36a897d36f6dcf5 Mon Sep 17 00:00:00 2001 From: Sriram Mohanty Date: Tue, 24 Dec 2024 19:18:08 +0530 Subject: [PATCH] antlr changes to include table properties --- .../remorph/transpiler/sqlglot/generator/databricks.py | 7 ++----- 1 file changed, 2 insertions(+), 5 deletions(-) diff --git a/src/databricks/labs/remorph/transpiler/sqlglot/generator/databricks.py b/src/databricks/labs/remorph/transpiler/sqlglot/generator/databricks.py index 5659787532..d43999e0fa 100644 --- a/src/databricks/labs/remorph/transpiler/sqlglot/generator/databricks.py +++ b/src/databricks/labs/remorph/transpiler/sqlglot/generator/databricks.py @@ -477,11 +477,8 @@ def format_time(self, expression: exp.Expression, inverse_time_mapping=None, inv def create_sql(self, expression: exp.Create) -> str: expression.args["indexes"] = None # Removing indexes from create statement create_sql = super().create_sql(expression) - for from_ in expression.find_all( - exp.DefaultColumnConstraint - ): # Generate table properties if DDL contain default values - create_sql = create_sql + " TBLPROPERTIES ('delta.feature.allowColumnDefaults' = 'supported')" - break + if expression.find(exp.DefaultColumnConstraint): + create_sql += " TBLPROPERTIES ('delta.feature.allowColumnDefaults' = 'supported')" return create_sql def join_sql(self, expression: exp.Join) -> str: