fakesnow 0.9.25__py3-none-any.whl → 0.9.26__py3-none-any.whl

fakesnow/cursor.py

@@ -162,6 +162,7 @@ class FakeSnowflakeCursor:
             .transform(transforms.extract_comment_on_columns)
             .transform(transforms.information_schema_fs_columns_snowflake)
             .transform(transforms.information_schema_fs_tables_ext)
+            .transform(transforms.information_schema_fs_views)
             .transform(transforms.drop_schema_cascade)
             .transform(transforms.tag)
             .transform(transforms.semi_structured_types)
@@ -294,6 +295,9 @@ class FakeSnowflakeCursor:
             (affected_count,) = self._duck_conn.fetchall()[0]
             result_sql = SQL_DELETED_ROWS.substitute(count=affected_count)
 
+        elif cmd == "TRUNCATETABLE":
+            result_sql = SQL_SUCCESS
+
         elif cmd in ("DESCRIBE TABLE", "DESCRIBE VIEW"):
             # DESCRIBE TABLE/VIEW has already been run above to detect and error if the table exists
             # We now rerun DESCRIBE TABLE/VIEW but transformed with columns to match Snowflake

fakesnow/info_schema.py

@@ -102,7 +102,7 @@ where catalog_name not in ('memory', 'system', 'temp', '_fs_global')
 # replicates https://docs.snowflake.com/sql-reference/info-schema/views
 SQL_CREATE_INFORMATION_SCHEMA_VIEWS_VIEW = Template(
     """
-create view if not exists ${catalog}.information_schema.views AS
+create view if not exists ${catalog}.information_schema._fs_views AS
 select
     database_name as table_catalog,
     schema_name as table_schema,

fakesnow/transforms.py

@@ -37,7 +37,7 @@ def alias_in_join(expression: exp.Expression) -> exp.Expression:
 def alter_table_strip_cluster_by(expression: exp.Expression) -> exp.Expression:
     """Turn alter table cluster by into a no-op"""
     if (
-        isinstance(expression, exp.AlterTable)
+        isinstance(expression, exp.Alter)
         and (actions := expression.args.get("actions"))
         and len(actions) == 1
         and (isinstance(actions[0], exp.Cluster))
@@ -356,7 +356,7 @@ def extract_comment_on_columns(expression: exp.Expression) -> exp.Expression:
         exp.Expression: The transformed expression, with any comment stored in the new 'table_comment' arg.
     """
 
-    if isinstance(expression, exp.AlterTable) and (actions := expression.args.get("actions")):
+    if isinstance(expression, exp.Alter) and (actions := expression.args.get("actions")):
         new_actions: list[exp.Expression] = []
         col_comments: list[tuple[str, str]] = []
         for a in actions:
@@ -410,7 +410,7 @@ def extract_comment_on_table(expression: exp.Expression) -> exp.Expression:
         new.args["table_comment"] = (table, cexp.this)
         return new
     elif (
-        isinstance(expression, exp.AlterTable)
+        isinstance(expression, exp.Alter)
         and (sexp := expression.find(exp.AlterSet))
         and (scp := sexp.find(exp.SchemaCommentProperty))
         and isinstance(scp.this, exp.Literal)
@@ -436,7 +436,7 @@ def extract_text_length(expression: exp.Expression) -> exp.Expression:
         exp.Expression: The original expression, with any text lengths stored in the new 'text_lengths' arg.
     """
 
-    if isinstance(expression, (exp.Create, exp.AlterTable)):
+    if isinstance(expression, (exp.Create, exp.Alter)):
         text_lengths = []
 
         # exp.Select is for a ctas, exp.Schema is a plain definition
@@ -471,7 +471,6 @@ def flatten(expression: exp.Expression) -> exp.Expression:
 
     See https://docs.snowflake.com/en/sql-reference/functions/flatten
 
-    TODO: return index.
     TODO: support objects.
     """
     if (
@@ -483,20 +482,34 @@ def flatten(expression: exp.Expression) -> exp.Expression:
     ):
         explode_expression = expression.this.this.expression
 
-        return exp.Lateral(
-            this=exp.Unnest(
+        value = exp.Cast(
+            this=explode_expression,
+            to=exp.DataType(
+                this=exp.DataType.Type.ARRAY,
+                expressions=[exp.DataType(this=exp.DataType.Type.JSON, nested=False, prefix=False)],
+                nested=True,
+            ),
+        )
+
+        return exp.Subquery(
+            this=exp.Select(
                 expressions=[
-                    exp.Cast(
-                        this=explode_expression,
-                        to=exp.DataType(
-                            this=exp.DataType.Type.ARRAY,
-                            expressions=[exp.DataType(this=exp.DataType.Type.JSON, nested=False, prefix=False)],
-                            nested=True,
+                    exp.Unnest(
+                        expressions=[value],
+                        alias=exp.Identifier(this="VALUE", quoted=False),
+                    ),
+                    exp.Alias(
+                        this=exp.Sub(
+                            this=exp.Anonymous(
+                                this="generate_subscripts", expressions=[value, exp.Literal(this="1", is_string=False)]
+                            ),
+                            expression=exp.Literal(this="1", is_string=False),
                         ),
-                    )
+                        alias=exp.Identifier(this="INDEX", quoted=False),
+                    ),
                 ],
             ),
-            alias=exp.TableAlias(this=alias.this, columns=[exp.Identifier(this="VALUE", quoted=False)]),
+            alias=exp.TableAlias(this=alias.this),
         )
 
     return expression
@@ -622,6 +635,20 @@ def information_schema_fs_tables_ext(expression: exp.Expression) -> exp.Expressi
     return expression
 
 
+def information_schema_fs_views(expression: exp.Expression) -> exp.Expression:
+    """Use information_schema._fs_views to return Snowflake's version instead of duckdb's."""
+
+    if (
+        isinstance(expression, exp.Select)
+        and (tbl_exp := expression.find(exp.Table))
+        and tbl_exp.name.upper() == "VIEWS"
+        and tbl_exp.db.upper() == "INFORMATION_SCHEMA"
+    ):
+        tbl_exp.set("this", exp.Identifier(this="_FS_VIEWS", quoted=False))
+
+    return expression
+
+
 def integer_precision(expression: exp.Expression) -> exp.Expression:
     """Convert integers to bigint.
 
@@ -707,8 +734,8 @@ def random(expression: exp.Expression) -> exp.Expression:
         new_rand = exp.Cast(
             this=exp.Paren(
                 this=exp.Mul(
-                    this=exp.Paren(this=exp.Sub(this=exp.Rand(), expression=exp.Literal(this=0.5, is_string=False))),
-                    expression=exp.Literal(this=9223372036854775807, is_string=False),
+                    this=exp.Paren(this=exp.Sub(this=exp.Rand(), expression=exp.Literal(this="0.5", is_string=False))),
+                    expression=exp.Literal(this="9223372036854775807", is_string=False),
                 )
             ),
             to=exp.DataType(this=exp.DataType.Type.BIGINT, nested=False, prefix=False),
@@ -809,31 +836,24 @@ def regex_substr(expression: exp.Expression) -> exp.Expression:
         pattern.args["this"] = pattern.this.replace("\\\\", "\\")
 
         # number of characters from the beginning of the string where the function starts searching for matches
-        try:
-            position = expression.args["position"]
-        except KeyError:
-            position = exp.Literal(this="1", is_string=False)
+        position = expression.args["position"] or exp.Literal(this="1", is_string=False)
 
         # which occurrence of the pattern to match
-        try:
-            occurrence = int(expression.args["occurrence"].this)
-        except KeyError:
-            occurrence = 1
+        occurrence = expression.args["occurrence"]
+        occurrence = int(occurrence.this) if occurrence else 1
 
         # the duckdb dialect increments bracket (ie: index) expressions by 1 because duckdb is 1-indexed,
         # so we need to compensate by subtracting 1
         occurrence = exp.Literal(this=str(occurrence - 1), is_string=False)
 
-        try:
-            regex_parameters_value = str(expression.args["parameters"].this)
+        if parameters := expression.args["parameters"]:
             # 'e' parameter doesn't make sense for duckdb
-            regex_parameters = exp.Literal(this=regex_parameters_value.replace("e", ""), is_string=True)
-        except KeyError:
+            regex_parameters = exp.Literal(this=parameters.this.replace("e", ""), is_string=True)
+        else:
             regex_parameters = exp.Literal(is_string=True)
 
-        try:
-            group_num = expression.args["group"]
-        except KeyError:
+        group_num = expression.args["group"]
+        if not group_num:
             if isinstance(regex_parameters.this, str) and "e" in regex_parameters.this:
                 group_num = exp.Literal(this="1", is_string=False)
             else:
@@ -1023,7 +1043,7 @@ def tag(expression: exp.Expression) -> exp.Expression:
         exp.Expression: The transformed expression.
     """
 
-    if isinstance(expression, exp.AlterTable) and (actions := expression.args.get("actions")):
+    if isinstance(expression, exp.Alter) and (actions := expression.args.get("actions")):
         for a in actions:
             if isinstance(a, exp.AlterSet) and a.args.get("tag"):
                 return SUCCESS_NOP

{fakesnow-0.9.25.dist-info → fakesnow-0.9.26.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: fakesnow
-Version: 0.9.25
+Version: 0.9.26
 Summary: Fake Snowflake Connector for Python. Run, mock and test Snowflake DB locally.
 License: Apache License
                                    Version 2.0, January 2004
@@ -210,22 +210,22 @@ Classifier: License :: OSI Approved :: MIT License
 Requires-Python: >=3.9
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: duckdb~=1.0.0
+Requires-Dist: duckdb~=1.1.3
 Requires-Dist: pyarrow
 Requires-Dist: snowflake-connector-python
-Requires-Dist: sqlglot~=25.9.0
+Requires-Dist: sqlglot~=25.24.1
 Provides-Extra: dev
 Requires-Dist: build~=1.0; extra == "dev"
 Requires-Dist: dirty-equals; 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: pre-commit~=4.0; extra == "dev"
 Requires-Dist: pyarrow-stubs==10.0.1.9; extra == "dev"
 Requires-Dist: pytest~=8.0; extra == "dev"
 Requires-Dist: pytest-asyncio; extra == "dev"
-Requires-Dist: ruff~=0.6.3; extra == "dev"
+Requires-Dist: ruff~=0.7.2; extra == "dev"
 Requires-Dist: twine~=5.0; extra == "dev"
-Requires-Dist: snowflake-sqlalchemy~=1.5.0; extra == "dev"
+Requires-Dist: snowflake-sqlalchemy~=1.6.1; extra == "dev"
 Provides-Extra: notebook
 Requires-Dist: duckdb-engine; extra == "notebook"
 Requires-Dist: ipykernel; extra == "notebook"

{fakesnow-0.9.25.dist-info → fakesnow-0.9.26.dist-info}/RECORD

@@ -4,23 +4,23 @@ fakesnow/arrow.py,sha256=EGAYeuCnRuvmWBEGqw2YOcgQR4zcCsZBu85kSRl70dQ,4698
 fakesnow/checks.py,sha256=N8sXldhS3u1gG32qvZ4VFlsKgavRKrQrxLiQU8am1lw,2691
 fakesnow/cli.py,sha256=9qfI-Ssr6mo8UmIlXkUAOz2z2YPBgDsrEVaZv9FjGFs,2201
 fakesnow/conn.py,sha256=Gy_Z7BZRm5yMjV3x6hR4iegDQFdG9aJBjqWdc3iWYFU,5353
-fakesnow/cursor.py,sha256=JbLSTzIN5Hu6ECn1kQ8-hC8V-ENeEWTID4JxHbGpEIo,19948
+fakesnow/cursor.py,sha256=8wWtRCxzrM1yiHmH2C-9CT0b98nTzr23ygeaEAkumRE,20086
 fakesnow/expr.py,sha256=CAxuYIUkwI339DQIBzvFF0F-m1tcVGKEPA5rDTzmH9A,892
 fakesnow/fakes.py,sha256=JQTiUkkwPeQrJ8FDWhPFPK6pGwd_aR2oiOrNzCWznlM,187
 fakesnow/fixtures.py,sha256=G-NkVeruSQAJ7fvSS2fR2oysUn0Yra1pohHlOvacKEk,455
-fakesnow/info_schema.py,sha256=DObVOrhzppAFHsdtj4YI9oRISn9SkJUG6ONjVleQQ_Y,6303
+fakesnow/info_schema.py,sha256=nsDceFtjiSXrvkksKziVvqrefskaSyOmAspBwMAsaDg,6307
 fakesnow/instance.py,sha256=3cJvPRuFy19dMKXbtBLl6imzO48pEw8uTYhZyFDuwhk,3133
 fakesnow/macros.py,sha256=pX1YJDnQOkFJSHYUjQ6ErEkYIKvFI6Ncz_au0vv1csA,265
 fakesnow/pandas_tools.py,sha256=WjyjTV8QUCQQaCGboaEOvx2uo4BkknpWYjtLwkeCY6U,3468
 fakesnow/py.typed,sha256=B-DLSjYBi7pkKjwxCSdpVj2J02wgfJr-E7B1wOUyxYU,80
 fakesnow/server.py,sha256=SO5xKZ4rvySsuKDsoSPSCZcFuIX_K7d1XJYhRRJ-7Bk,4150
-fakesnow/transforms.py,sha256=hPQ9L1TvsDjFiGbs8mGUnIFyPUR7JxFU8FZRKFj5ZD0,54568
+fakesnow/transforms.py,sha256=VFLA5Fc1i4FuiVdvUuDrK-kA2caqiT8Gw9btMDPJhRA,55367
 fakesnow/transforms_merge.py,sha256=7rq-UPjfFNRrFsqR8xx3otwP6-k4eslLVLhfuqSXq1A,8314
 fakesnow/types.py,sha256=9Tt83Z7ctc9_v6SYyayXYz4MEI4RZo4zq_uqdj4g3Dk,2681
 fakesnow/variables.py,sha256=WXyPnkeNwD08gy52yF66CVe2twiYC50tztNfgXV4q1k,3032
-fakesnow-0.9.25.dist-info/LICENSE,sha256=kW-7NWIyaRMQiDpryfSmF2DObDZHGR1cJZ39s6B1Svg,11344
-fakesnow-0.9.25.dist-info/METADATA,sha256=1RktYqC8KfU4ekh8IGApjHSLb5oe3KMgBLjyKJKXHRc,18074
-fakesnow-0.9.25.dist-info/WHEEL,sha256=eOLhNAGa2EW3wWl_TU484h7q1UNgy0JXjjoqKoxAAQc,92
-fakesnow-0.9.25.dist-info/entry_points.txt,sha256=2riAUgu928ZIHawtO8EsfrMEJhi-EH-z_Vq7Q44xKPM,47
-fakesnow-0.9.25.dist-info/top_level.txt,sha256=500evXI1IFX9so82cizGIEMHAb_dJNPaZvd2H9dcKTA,24
-fakesnow-0.9.25.dist-info/RECORD,,
+fakesnow-0.9.26.dist-info/LICENSE,sha256=kW-7NWIyaRMQiDpryfSmF2DObDZHGR1cJZ39s6B1Svg,11344
+fakesnow-0.9.26.dist-info/METADATA,sha256=92zIwzq7FP-BrfhUcKbdbqYs0eqN9TCKvT_NVdEKZTI,18075
+fakesnow-0.9.26.dist-info/WHEEL,sha256=bFJAMchF8aTQGUgMZzHJyDDMPTO3ToJ7x23SLJa1SVo,92
+fakesnow-0.9.26.dist-info/entry_points.txt,sha256=2riAUgu928ZIHawtO8EsfrMEJhi-EH-z_Vq7Q44xKPM,47
+fakesnow-0.9.26.dist-info/top_level.txt,sha256=500evXI1IFX9so82cizGIEMHAb_dJNPaZvd2H9dcKTA,24
+fakesnow-0.9.26.dist-info/RECORD,,

{fakesnow-0.9.25.dist-info → fakesnow-0.9.26.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: bdist_wheel (0.44.0)
+Generator: bdist_wheel (0.45.0)
 Root-Is-Purelib: true
 Tag: py3-none-any