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

fakesnow/arrow.py

@@ -5,7 +5,7 @@ from typing import cast
 import pyarrow as pa
 import pyarrow.compute as pc
 
-from fakesnow.types import ColumnInfo
+from fakesnow.rowtype import ColumnInfo
 
 
 def to_sf_schema(schema: pa.Schema, rowtype: list[ColumnInfo]) -> pa.Schema:
@@ -27,6 +27,10 @@ def to_sf_schema(schema: pa.Schema, rowtype: list[ColumnInfo]) -> pa.Schema:
             field = field.with_type(pa.struct(fields))
         elif isinstance(field.type, pa.Time64Type):
             field = field.with_type(pa.int64())
+        elif pa.types.is_uint64(field.type):
+            # snowflake-python-connector expects signed ints
+            # see https://github.com/snowflakedb/snowflake-connector-python/blob/5d7064c7f3f756792c1f6252bf5c9d807e4307e8/src/snowflake/connector/nanoarrow_cpp/ArrowIterator/CArrowChunkIterator.cpp#L187
+            field = field.with_type(pa.int64())
 
         return field.with_metadata(
             {

fakesnow/conn.py

@@ -60,7 +60,7 @@ class FakeSnowflakeConnection:
                 where upper(catalog_name) = '{self.database}'"""
             ).fetchone()
         ):
-            db_file = f"{self.db_path/self.database}.db" if self.db_path else ":memory:"
+            db_file = f"{self.db_path / self.database}.db" if self.db_path else ":memory:"
             duck_conn.execute(f"ATTACH DATABASE '{db_file}' AS {self.database}")
             duck_conn.execute(info_schema.creation_sql(self.database))
             duck_conn.execute(macros.creation_sql(self.database))
@@ -114,6 +114,10 @@ class FakeSnowflakeConnection:
     ) -> None:
         pass
 
+    def autocommit(self, _mode: bool) -> None:
+        # autcommit is always on in duckdb
+        pass
+
     def close(self, retry: bool = True) -> None:
         self._duck_conn.close()
         self._is_closed = True

fakesnow/cursor.py

@@ -3,13 +3,14 @@ from __future__ import annotations
 import os
 import re
 import sys
+import uuid
 from collections.abc import Iterator, Sequence
 from string import Template
 from types import TracebackType
 from typing import TYPE_CHECKING, Any, cast
 
 import duckdb
-import pyarrow
+import pyarrow  # needed by fetch_arrow_table()
 import snowflake.connector.converter
 import snowflake.connector.errors
 import sqlglot
@@ -23,7 +24,7 @@ import fakesnow.checks as checks
 import fakesnow.expr as expr
 import fakesnow.info_schema as info_schema
 import fakesnow.transforms as transforms
-from fakesnow.types import describe_as_result_metadata
+from fakesnow.rowtype import describe_as_result_metadata
 
 if TYPE_CHECKING:
     # don't require pandas at import time
@@ -71,6 +72,7 @@ class FakeSnowflakeCursor:
         self._arrow_table = None
         self._arrow_table_fetch_index = None
         self._rowcount = None
+        self._sfqid = None
         self._converter = snowflake.connector.converter.SnowflakeConverter()
 
     def __enter__(self) -> Self:
@@ -222,6 +224,7 @@ class FakeSnowflakeCursor:
         self._arrow_table = None
         self._arrow_table_fetch_index = None
         self._rowcount = None
+        self._sfqid = None
 
         cmd = expr.key_command(transformed)
 
@@ -353,6 +356,7 @@ class FakeSnowflakeCursor:
 
         self._arrow_table = self._duck_conn.fetch_arrow_table()
         self._rowcount = affected_count or self._arrow_table.num_rows
+        self._sfqid = str(uuid.uuid4())
 
         self._last_sql = result_sql or sql
         self._last_params = params
@@ -423,7 +427,7 @@ class FakeSnowflakeCursor:
 
     @property
     def sfqid(self) -> str | None:
-        return "fakesnow"
+        return self._sfqid
 
     @property
     def sqlstate(self) -> str | None:

fakesnow/info_schema.py

@@ -146,7 +146,7 @@ def insert_table_comment_sql(catalog: str, schema: str, table: str, comment: str
 
 def insert_text_lengths_sql(catalog: str, schema: str, table: str, text_lengths: list[tuple[str, int]]) -> str:
     values = ", ".join(
-        f"('{catalog}', '{schema}', '{table}', '{col_name}', {size}, {min(size*4,16777216)})"
+        f"('{catalog}', '{schema}', '{table}', '{col_name}', {size}, {min(size * 4, 16777216)})"
         for (col_name, size) in text_lengths
     )
 

fakesnow/pandas_tools.py

@@ -4,13 +4,13 @@ import json
 from collections.abc import Sequence
 from typing import TYPE_CHECKING, Any, Literal, Optional
 
-import numpy as np
 from duckdb import DuckDBPyConnection
 
 from fakesnow.conn import FakeSnowflakeConnection
 
 if TYPE_CHECKING:
-    # don't require pandas at import time
+    # don't require pandas or numpy at import time
+    import numpy as np
     import pandas as pd
 
 

fakesnow/{types.py → rowtype.py}

@@ -25,12 +25,14 @@ duckdb_to_sf_type = {
     "DATE": "date",
     "DECIMAL": "fixed",
     "DOUBLE": "real",
+    "HUGEINT": "fixed",
     "INTEGER": "fixed",
     "JSON": "variant",
     "TIME": "time",
     "TIMESTAMP WITH TIME ZONE": "timestamp_tz",
     "TIMESTAMP_NS": "timestamp_ntz",
     "TIMESTAMP": "timestamp_ntz",
+    "UBIGINT": "fixed",
     "VARCHAR": "text",
 }
 

fakesnow/server.py

@@ -17,7 +17,7 @@ from starlette.routing import Route
 from fakesnow.arrow import to_ipc, to_sf
 from fakesnow.fakes import FakeSnowflakeConnection
 from fakesnow.instance import FakeSnow
-from fakesnow.types import describe_as_rowtype
+from fakesnow.rowtype import describe_as_rowtype
 
 shared_fs = FakeSnow()
 sessions: dict[str, FakeSnowflakeConnection] = {}
@@ -87,7 +87,8 @@ async def query_request(request: Request) -> JSONResponse:
                 "data": {
                     "rowtype": rowtype,
                     "rowsetBase64": rowset_b64,
-                    "total": 1,
+                    "total": cur._rowcount,  # noqa: SLF001
+                    "queryId": cur.sfqid,
                     "queryResultFormat": "arrow",
                 },
                 "success": True,

fakesnow/transforms.py

@@ -28,6 +28,8 @@ def alias_in_join(expression: exp.Expression) -> exp.Expression:
                 and (col := on.this)
                 and (isinstance(col, exp.Column))
                 and (alias := aliases.get(col.this))
+                # don't rewrite col with table identifier
+                and not col.table
             ):
                 col.args["this"] = alias.this
 
@@ -131,7 +133,7 @@ def create_database(expression: exp.Expression, db_path: Path | None = None) ->
         ident = expression.find(exp.Identifier)
         assert ident, f"No identifier in {expression.sql}"
         db_name = ident.this
-        db_file = f"{db_path/db_name}.db" if db_path else ":memory:"
+        db_file = f"{db_path / db_name}.db" if db_path else ":memory:"
 
         if_not_exists = "IF NOT EXISTS " if expression.args.get("exists") else ""
 
@@ -658,8 +660,7 @@ def integer_precision(expression: exp.Expression) -> exp.Expression:
     if (
         isinstance(expression, exp.DataType)
         and (expression.this == exp.DataType.Type.DECIMAL and not expression.expressions)
-        or expression.this in (exp.DataType.Type.INT, exp.DataType.Type.SMALLINT, exp.DataType.Type.TINYINT)
-    ):
+    ) or expression.this in (exp.DataType.Type.INT, exp.DataType.Type.SMALLINT, exp.DataType.Type.TINYINT):
         return exp.DataType(
             this=exp.DataType.Type.BIGINT,
             nested=False,

fakesnow/transforms_merge.py

@@ -48,7 +48,7 @@ def _create_merge_candidates(merge_expr: exp.Merge) -> exp.Expression:
     )
 
     # Iterate through the WHEN clauses to build up the CASE WHEN clauses
-    for w_idx, w in enumerate(merge_expr.expressions):
+    for w_idx, w in enumerate(merge_expr.args["whens"]):
         assert isinstance(w, exp.When), f"Expected When expression, got {w}"
 
         predicate = join_expr.copy()
@@ -83,9 +83,9 @@ def _create_merge_candidates(merge_expr: exp.Merge) -> exp.Expression:
     sql = f"""
     CREATE OR REPLACE TEMPORARY TABLE merge_candidates AS
     SELECT
-        {', '.join(sorted(values))},
+        {", ".join(sorted(values))},
         CASE
-            {' '.join(case_when_clauses)}
+            {" ".join(case_when_clauses)}
             ELSE NULL
         END AS MERGE_OP
     FROM {target_tbl}
@@ -109,7 +109,7 @@ def _mutations(merge_expr: exp.Merge) -> list[exp.Expression]:
     statements: list[exp.Expression] = []
 
     # Iterate through the WHEN clauses to generate delete/update/insert statements
-    for w_idx, w in enumerate(merge_expr.expressions):
+    for w_idx, w in enumerate(merge_expr.args["whens"]):
         assert isinstance(w, exp.When), f"Expected When expression, got {w}"
 
         matched = w.args.get("matched")
@@ -173,7 +173,7 @@ def _counts(merge_expr: exp.Merge) -> exp.Expression:
     operations = {"inserted": [], "updated": [], "deleted": []}
 
     # Iterate through the WHEN clauses to categorize operations
-    for w_idx, w in enumerate(merge_expr.expressions):
+    for w_idx, w in enumerate(merge_expr.args["whens"]):
         assert isinstance(w, exp.When), f"Expected When expression, got {w}"
 
         matched = w.args.get("matched")
@@ -191,12 +191,12 @@ def _counts(merge_expr: exp.Merge) -> exp.Expression:
             operations["inserted"].append(w_idx)
 
     count_statements = [
-        f"""COUNT_IF(merge_op in ({','.join(map(str, indices))})) as \"number of rows {op}\""""
+        f"""COUNT_IF(merge_op in ({",".join(map(str, indices))})) as \"number of rows {op}\""""
         for op, indices in operations.items()
         if indices
     ]
     sql = f"""
-    SELECT {', '.join(count_statements)}
+    SELECT {", ".join(count_statements)}
     FROM merge_candidates
     """
 

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

@@ -1,8 +1,8 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.2
 Name: fakesnow
-Version: 0.9.26
+Version: 0.9.28
 Summary: Fake Snowflake Connector for Python. Run, mock and test Snowflake DB locally.
-License: Apache License
+License:                                  Apache License
                                    Version 2.0, January 2004
                                 http://www.apache.org/licenses/
         
@@ -213,7 +213,7 @@ License-File: LICENSE
 Requires-Dist: duckdb~=1.1.3
 Requires-Dist: pyarrow
 Requires-Dist: snowflake-connector-python
-Requires-Dist: sqlglot~=25.24.1
+Requires-Dist: sqlglot~=26.3.9
 Provides-Extra: dev
 Requires-Dist: build~=1.0; extra == "dev"
 Requires-Dist: dirty-equals; extra == "dev"
@@ -223,9 +223,9 @@ 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.7.2; extra == "dev"
-Requires-Dist: twine~=5.0; extra == "dev"
-Requires-Dist: snowflake-sqlalchemy~=1.6.1; extra == "dev"
+Requires-Dist: ruff~=0.9.4; extra == "dev"
+Requires-Dist: twine~=6.0; extra == "dev"
+Requires-Dist: snowflake-sqlalchemy~=1.7.0; extra == "dev"
 Provides-Extra: notebook
 Requires-Dist: duckdb-engine; extra == "notebook"
 Requires-Dist: ipykernel; extra == "notebook"

fakesnow-0.9.28.dist-info/RECORD

@@ -0,0 +1,26 @@
+fakesnow/__init__.py,sha256=qUfgucQYPdELrJaxczalhJgWAWQ6cfTCUAHx6nUqRaI,3528
+fakesnow/__main__.py,sha256=GDrGyNTvBFuqn_UfDjKs7b3LPtU6gDv1KwosVDrukIM,76
+fakesnow/arrow.py,sha256=MwatkdZX5AFADzXvxhBFmcRJVxbW4D39VoqLyhpTbl0,5057
+fakesnow/checks.py,sha256=N8sXldhS3u1gG32qvZ4VFlsKgavRKrQrxLiQU8am1lw,2691
+fakesnow/cli.py,sha256=9qfI-Ssr6mo8UmIlXkUAOz2z2YPBgDsrEVaZv9FjGFs,2201
+fakesnow/conn.py,sha256=GJ7Y2dBW2jcOCIZ0gTYS0F8OmeoD7aW6lTWHpm02hbE,5459
+fakesnow/cursor.py,sha256=uYf3zshauWXnKdUoVEE3YxMbc-SoVLHCUUkqwtMW8ns,20228
+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=FyDcajHU0BK0Yx6JTIkWFoM0PPFm6f6Pf0B2NHLNR0M,6310
+fakesnow/instance.py,sha256=3cJvPRuFy19dMKXbtBLl6imzO48pEw8uTYhZyFDuwhk,3133
+fakesnow/macros.py,sha256=pX1YJDnQOkFJSHYUjQ6ErEkYIKvFI6Ncz_au0vv1csA,265
+fakesnow/pandas_tools.py,sha256=wI203UQHC8JvDzxE_VjE1NeV4rThek2P-u52oTg2foo,3481
+fakesnow/py.typed,sha256=B-DLSjYBi7pkKjwxCSdpVj2J02wgfJr-E7B1wOUyxYU,80
+fakesnow/rowtype.py,sha256=QUp8EaXD5LT0Xv8BXk5ze4WseEn52xoJ6R05pJjs5mM,2729
+fakesnow/server.py,sha256=lrXk8iXioSl-qWweXLH7l6aPrcV4Bym9rjB6x_C5Fg8,4222
+fakesnow/transforms.py,sha256=pSv3pQlD1Y8tzXQ1rft2g5wYLcHrDRoy5EkFWgqkmec,55453
+fakesnow/transforms_merge.py,sha256=Pg7_rwbAT_vr1U4ocBofUSyqaK8_e3qdIz_2SDm2S3s,8320
+fakesnow/variables.py,sha256=WXyPnkeNwD08gy52yF66CVe2twiYC50tztNfgXV4q1k,3032
+fakesnow-0.9.28.dist-info/LICENSE,sha256=kW-7NWIyaRMQiDpryfSmF2DObDZHGR1cJZ39s6B1Svg,11344
+fakesnow-0.9.28.dist-info/METADATA,sha256=t0B6J7rG5uyS2430rgPWieQgACfyUU9dIL5cUdYllDg,18107
+fakesnow-0.9.28.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+fakesnow-0.9.28.dist-info/entry_points.txt,sha256=2riAUgu928ZIHawtO8EsfrMEJhi-EH-z_Vq7Q44xKPM,47
+fakesnow-0.9.28.dist-info/top_level.txt,sha256=500evXI1IFX9so82cizGIEMHAb_dJNPaZvd2H9dcKTA,24
+fakesnow-0.9.28.dist-info/RECORD,,

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

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

fakesnow-0.9.26.dist-info/RECORD

@@ -1,26 +0,0 @@
-fakesnow/__init__.py,sha256=qUfgucQYPdELrJaxczalhJgWAWQ6cfTCUAHx6nUqRaI,3528
-fakesnow/__main__.py,sha256=GDrGyNTvBFuqn_UfDjKs7b3LPtU6gDv1KwosVDrukIM,76
-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=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=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=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.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,,