Fix: make sqlglot generator overriding more robust (#3650)

georgesittas · web-flow · commit fea4507a1d1d · 2025-01-17T18:41:54.000+02:00
diff --git a/sqlmesh/core/dialect.py b/sqlmesh/core/dialect.py
@@ -1014,7 +1014,7 @@ def extend_sqlglot() -> None:
                     exp.Whens: _whens_sql,
                 }
             )
-
+        if MacroDef not in generator.WITH_SEPARATED_COMMENTS:
             generator.WITH_SEPARATED_COMMENTS = (
                 *generator.WITH_SEPARATED_COMMENTS,
                 Model,

Original file line number	Diff line number	Diff line change
`@@ -1014,7 +1014,7 @@ def extend_sqlglot() -> None:`
`1014`	`1014`	`exp.Whens: _whens_sql,`
`1015`	`1015`	`}`
`1016`	`1016`	`)`
`1017`		`-`
	`1017`	`+ if MacroDef not in generator.WITH_SEPARATED_COMMENTS:`
`1018`	`1018`	`generator.WITH_SEPARATED_COMMENTS = (`
`1019`	`1019`	`*generator.WITH_SEPARATED_COMMENTS,`
`1020`	`1020`	`Model,`