fakesnow 0.9.8__tar.gz → 0.9.9__tar.gz

{fakesnow-0.9.8 → fakesnow-0.9.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: fakesnow
-Version: 0.9.8
+Version: 0.9.9
 Summary: Fake Snowflake Connector for Python. Run, mock and test Snowflake DB locally.
 License:                                  Apache License
                                    Version 2.0, January 2004
@@ -213,7 +213,7 @@ License-File: LICENSE
 Requires-Dist: duckdb~=0.10.0
 Requires-Dist: pyarrow
 Requires-Dist: snowflake-connector-python
-Requires-Dist: sqlglot~=23.3.0
+Requires-Dist: sqlglot~=23.12.2
 Provides-Extra: dev
 Requires-Dist: build~=1.0; extra == "dev"
 Requires-Dist: pandas-stubs; extra == "dev"

{fakesnow-0.9.8 → fakesnow-0.9.9}/fakesnow/transforms.py

@@ -309,7 +309,7 @@ def extract_comment_on_table(expression: exp.Expression) -> exp.Expression:
         if props := cast(exp.Properties, expression.args.get("properties")):
             other_props = []
             for p in props.expressions:
-                if isinstance(p, exp.SchemaCommentProperty) and (isinstance(p.this, (exp.Literal, exp.Identifier))):
+                if isinstance(p, exp.SchemaCommentProperty) and (isinstance(p.this, (exp.Literal, exp.Var))):
                     comment = p.this.this
                 else:
                     other_props.append(p)
@@ -360,10 +360,19 @@ def extract_text_length(expression: exp.Expression) -> exp.Expression:
 
     if isinstance(expression, (exp.Create, exp.AlterTable)):
         text_lengths = []
-        for dt in expression.find_all(exp.DataType):
-            if dt.this in (exp.DataType.Type.VARCHAR, exp.DataType.Type.TEXT):
-                col_name = dt.parent and dt.parent.this and dt.parent.this.this
-                if dt_size := dt.find(exp.DataTypeParam):
+
+        # exp.Select is for a ctas, exp.Schema is a plain definition
+        if cols := expression.find(exp.Select, exp.Schema):
+            expressions = cols.expressions
+        else:
+            # alter table
+            expressions = expression.args.get("actions") or []
+        for e in expressions:
+            if dts := [
+                dt for dt in e.find_all(exp.DataType) if dt.this in (exp.DataType.Type.VARCHAR, exp.DataType.Type.TEXT)
+            ]:
+                col_name = e.alias if isinstance(e, exp.Alias) else e.name
+                if len(dts) == 1 and (dt_size := dts[0].find(exp.DataTypeParam)):
                     size = (
                         isinstance(dt_size.this, exp.Literal)
                         and isinstance(dt_size.this.this, str)

{fakesnow-0.9.8 → fakesnow-0.9.9}/fakesnow.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: fakesnow
-Version: 0.9.8
+Version: 0.9.9
 Summary: Fake Snowflake Connector for Python. Run, mock and test Snowflake DB locally.
 License:                                  Apache License
                                    Version 2.0, January 2004
@@ -213,7 +213,7 @@ License-File: LICENSE
 Requires-Dist: duckdb~=0.10.0
 Requires-Dist: pyarrow
 Requires-Dist: snowflake-connector-python
-Requires-Dist: sqlglot~=23.3.0
+Requires-Dist: sqlglot~=23.12.2
 Provides-Extra: dev
 Requires-Dist: build~=1.0; extra == "dev"
 Requires-Dist: pandas-stubs; extra == "dev"

{fakesnow-0.9.8 → fakesnow-0.9.9}/fakesnow.egg-info/requires.txt

@@ -1,7 +1,7 @@
 duckdb~=0.10.0
 pyarrow
 snowflake-connector-python
-sqlglot~=23.3.0
+sqlglot~=23.12.2
 
 [dev]
 build~=1.0

{fakesnow-0.9.8 → fakesnow-0.9.9}/pyproject.toml

@@ -1,7 +1,7 @@
 [project]
 name = "fakesnow"
 description = "Fake Snowflake Connector for Python. Run, mock and test Snowflake DB locally."
-version = "0.9.8"
+version = "0.9.9"
 readme = "README.md"
 license = { file = "LICENSE" }
 classifiers = ["License :: OSI Approved :: MIT License"]
@@ -11,7 +11,7 @@ dependencies = [
     "duckdb~=0.10.0",
     "pyarrow",
     "snowflake-connector-python",
-    "sqlglot~=23.3.0",
+    "sqlglot~=23.12.2",
 ]
 
 [project.urls]

{fakesnow-0.9.8 → fakesnow-0.9.9}/tests/test_fakes.py

@@ -1284,6 +1284,13 @@ def test_show_primary_keys(dcur: snowflake.connector.cursor.SnowflakeCursor):
     assert result3 == []
 
 
+def test_sqlglot_regression(cur: snowflake.connector.cursor.SnowflakeCursor):
+    assert cur.execute(
+        """with SOURCE_TABLE AS (SELECT '2024-01-01' AS start_date)
+            SELECT date(a.start_date) from SOURCE_TABLE AS a"""
+    ).fetchone() == (datetime.date(2024, 1, 1),)
+
+
 def test_sqlstate(cur: snowflake.connector.cursor.SnowflakeCursor):
     cur.execute("select 'hello world'")
     # sqlstate is None on success

{fakesnow-0.9.8 → fakesnow-0.9.9}/tests/test_transforms.py

@@ -335,6 +335,25 @@ def test_extract_text_length() -> None:
     assert e.sql() == sql
     assert e.args["text_lengths"] == [("t1", 16777216), ("t2", 10), ("t3", 20)]
 
+    sql = "ALTER TABLE t1 ALTER COLUMN c4 SET DATA TYPE VARCHAR(50)"
+    e = sqlglot.parse_one(sql).transform(extract_text_length)
+    assert e.sql() == sql
+    assert e.args["text_lengths"] == [("c4", 50)]
+
+    # test column name is correct with alias
+    sql = """CREATE TABLE table1 AS (
+                SELECT CAST(C1 AS TEXT) AS K, CAST(C2 AS TEXT(10)) AS V
+                FROM (VALUES (1, 2)) AS T(C1, C2))"""
+    e = sqlglot.parse_one(sql).transform(extract_text_length)
+    assert e.args["text_lengths"] == [("K", 16777216), ("V", 10)]
+
+    # test ctas column name is correct for combined field
+    sql = """CREATE TABLE SOME_TABLE AS (
+                SELECT CAST(C1 AS TEXT) || '-' || CAST(C1 AS TEXT) AS K
+                FROM VALUES (1), (2) AS T (C1))"""
+    e = sqlglot.parse_one(sql).transform(extract_text_length)
+    assert e.args["text_lengths"] == [("K", 16777216)]
+
 
 def test_flatten() -> None:
     assert (