fix: allow snowflake object_construct with string keys to transpile t…

…o sqlglot dialect
tobymao · Mar 25, 2024 · e4dd052 · e4dd052
1 parent ec4648f
commit e4dd052
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/sqlglot/generator.py b/sqlglot/generator.py
@@ -3586,7 +3586,9 @@ def struct_sql(self, expression: exp.Struct) -> str:
         expression.set(
             "expressions",
             [
-                exp.alias_(e.expression, e.this) if isinstance(e, exp.PropertyEQ) else e
+                exp.alias_(e.expression, e.name if e.this.is_string else e.this)
+                if isinstance(e, exp.PropertyEQ)
+                else e
                 for e in expression.expressions
             ],
         )

diff --git a/tests/dialects/test_snowflake.py b/tests/dialects/test_snowflake.py
@@ -396,6 +396,7 @@ def test_snowflake(self):
             write={
                 "duckdb": "{'a': b, 'c': d}",
                 "snowflake": "OBJECT_CONSTRUCT('a', b, 'c', d)",
+                "": "STRUCT(b AS a, d AS c)",
             },
         )
         self.validate_identity("OBJECT_CONSTRUCT(a, b, c, d)")