antlr changes to include table properties

databrickslabs · Dec 24, 2024 · 8b7bb16 · 8b7bb16
1 parent 21da02a
commit 8b7bb16
Showing 1 changed file with 2 additions and 5 deletions.
diff --git 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: