fakesnow 0.9.8__tar.gz → 0.9.10__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: fakesnow
-Version: 0.9.8
+Version: 0.9.10
 Summary: Fake Snowflake Connector for Python. Run, mock and test Snowflake DB locally.
 License:                                  Apache License
                                    Version 2.0, January 2004
@@ -213,14 +213,14 @@ 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"
 Requires-Dist: snowflake-connector-python[pandas,secure-local-storage]; extra == "dev"
 Requires-Dist: pre-commit~=3.4; extra == "dev"
 Requires-Dist: pytest~=8.0; extra == "dev"
-Requires-Dist: ruff~=0.3.2; extra == "dev"
+Requires-Dist: ruff~=0.4.2; extra == "dev"
 Requires-Dist: twine~=5.0; extra == "dev"
 Requires-Dist: snowflake-sqlalchemy~=1.5.0; extra == "dev"
 Provides-Extra: notebook

{fakesnow-0.9.8 → fakesnow-0.9.10}/fakesnow/fakes.py

@@ -193,6 +193,7 @@ class FakeSnowflakeCursor:
             .transform(transforms.show_users)
             .transform(transforms.create_user)
             .transform(transforms.sha256)
+            .transform(transforms.create_clone)
         )
 
     def _execute(

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

@@ -55,6 +55,27 @@ def array_agg_within_group(expression: exp.Expression) -> exp.Expression:
     return expression
 
 
+def create_clone(expression: exp.Expression) -> exp.Expression:
+    """Transform create table clone to create table as select."""
+
+    if (
+        isinstance(expression, exp.Create)
+        and str(expression.args.get("kind")).upper() == "TABLE"
+        and (clone := expression.find(exp.Clone))
+    ):
+        return exp.Create(
+            this=expression.this,
+            kind="TABLE",
+            expression=exp.Select(
+                expressions=[
+                    exp.Star(),
+                ],
+                **{"from": exp.From(this=clone.this)},
+            ),
+        )
+    return expression
+
+
 # TODO: move this into a Dialect as a transpilation
 def create_database(expression: exp.Expression, db_path: Path | None = None) -> exp.Expression:
     """Transform create database to attach database.
@@ -309,7 +330,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 +381,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.10}/fakesnow.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: fakesnow
-Version: 0.9.8
+Version: 0.9.10
 Summary: Fake Snowflake Connector for Python. Run, mock and test Snowflake DB locally.
 License:                                  Apache License
                                    Version 2.0, January 2004
@@ -213,14 +213,14 @@ 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"
 Requires-Dist: snowflake-connector-python[pandas,secure-local-storage]; extra == "dev"
 Requires-Dist: pre-commit~=3.4; extra == "dev"
 Requires-Dist: pytest~=8.0; extra == "dev"
-Requires-Dist: ruff~=0.3.2; extra == "dev"
+Requires-Dist: ruff~=0.4.2; extra == "dev"
 Requires-Dist: twine~=5.0; extra == "dev"
 Requires-Dist: snowflake-sqlalchemy~=1.5.0; extra == "dev"
 Provides-Extra: notebook

{fakesnow-0.9.8 → fakesnow-0.9.10}/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
@@ -9,7 +9,7 @@ pandas-stubs
 snowflake-connector-python[pandas,secure-local-storage]
 pre-commit~=3.4
 pytest~=8.0
-ruff~=0.3.2
+ruff~=0.4.2
 twine~=5.0
 snowflake-sqlalchemy~=1.5.0
 

{fakesnow-0.9.8 → fakesnow-0.9.10}/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.10"
 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]
@@ -29,7 +29,7 @@ dev = [
     "snowflake-connector-python[pandas, secure-local-storage]",
     "pre-commit~=3.4",
     "pytest~=8.0",
-    "ruff~=0.3.2",
+    "ruff~=0.4.2",
     "twine~=5.0",
     "snowflake-sqlalchemy~=1.5.0",
 ]

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

@@ -110,6 +110,16 @@ def test_binding_qmark(_fakesnow: None):
         cur.execute("select * from customers where id = ?", (1,))
 
 
+def test_clone(cur: snowflake.connector.cursor.SnowflakeCursor):
+    cur.execute("create table customers (ID int, FIRST_NAME varchar, ACTIVE boolean)")
+    cur.execute("insert into customers values (1, 'Jenny', True)")
+
+    cur.execute("create table customers2 clone db1.schema1.customers")
+    cur.execute("select * from customers2")
+    # TODO check tags are copied too
+    assert cur.fetchall() == [(1, "Jenny", True)]
+
+
 def test_close_conn(conn: snowflake.connector.SnowflakeConnection, cur: snowflake.connector.cursor.SnowflakeCursor):
     conn.close()
     with pytest.raises(snowflake.connector.errors.DatabaseError) as excinfo:
@@ -1284,6 +1294,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.10}/tests/test_transforms.py

@@ -9,6 +9,7 @@ from fakesnow.transforms import (
     _get_to_number_args,
     array_agg_within_group,
     array_size,
+    create_clone,
     create_database,
     dateadd_date_cast,
     dateadd_string_literal_timestamp_cast,
@@ -84,6 +85,13 @@ def test_array_agg_within_group() -> None:
     )
 
 
+def test_create_clone() -> None:
+    assert (
+        sqlglot.parse_one("create table customers2 clone db1.schema1.customer").transform(create_clone).sql()
+        == "CREATE TABLE customers2 AS SELECT * FROM db1.schema1.customer"
+    )
+
+
 def test_create_database() -> None:
     e = sqlglot.parse_one("create database foobar").transform(create_database)
     assert e.sql() == "ATTACH DATABASE ':memory:' AS foobar"
@@ -335,6 +343,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 (