fakesnow 0.9.32__py3-none-any.whl → 0.9.34__py3-none-any.whl

fakesnow/converter.py

@@ -0,0 +1,60 @@
+from __future__ import annotations
+
+import binascii
+import datetime
+from datetime import date, time, timezone
+
+# convert bindings provided as strings to the server into python types
+
+
+def from_binding(binding: dict[str, str]) -> int | bytes | bool | date | time | datetime.datetime | str:
+    typ = binding["type"]
+    value = binding["value"]
+    if typ == "FIXED":
+        return int(value)
+    elif typ == "BINARY":
+        return from_binary(value)
+    # TODO: not strictly needed
+    elif typ == "BOOLEAN":
+        return value.lower() == "true"
+    elif typ == "DATE":
+        return from_date(value)
+    elif typ == "TIME":
+        return from_time(value)
+    elif typ == "TIMESTAMP_NTZ":
+        return from_datetime(value)
+    else:
+        # For other types, return str
+        return value
+
+
+def from_binary(s: str) -> bytes:
+    return binascii.unhexlify(s)
+
+
+def from_boolean(s: str) -> bool:
+    return s.lower() == "true"
+
+
+def from_date(s: str) -> date:
+    milliseconds = int(s)
+    seconds = milliseconds / 1000
+    return datetime.datetime.fromtimestamp(seconds, timezone.utc).date()
+
+
+def from_time(s: str) -> time:
+    nanoseconds = int(s)
+    microseconds = nanoseconds / 1000
+    return (
+        datetime.datetime.fromtimestamp(microseconds / 1_000_000, timezone.utc)
+        .replace(microsecond=int(microseconds % 1_000_000))
+        .time()
+    )
+
+
+def from_datetime(s: str) -> datetime.datetime:
+    nanoseconds = int(s)
+    microseconds = nanoseconds / 1000
+    return datetime.datetime.fromtimestamp(microseconds / 1_000_000, timezone.utc).replace(
+        microsecond=int(microseconds % 1_000_000)
+    )

fakesnow/cursor.py

@@ -139,7 +139,12 @@ class FakeSnowflakeCursor:
                 print(f"{command};{params=}" if params else f"{command};", file=sys.stderr)
 
             command = self._inline_variables(command)
-            command, params = self._rewrite_with_params(command, params)
+            if kwargs.get("binding_params"):
+                # params have come via the server
+                params = kwargs["binding_params"]
+            else:
+                command, params = self._rewrite_with_params(command, params)
+
             if self._conn.nop_regexes and any(re.match(p, command, re.IGNORECASE) for p in self._conn.nop_regexes):
                 transformed = transforms.SUCCESS_NOP
                 self._execute(transformed, params)
@@ -384,7 +389,7 @@ class FakeSnowflakeCursor:
         self._sfqid = str(uuid.uuid4())
 
         self._last_sql = result_sql or sql
-        self._last_params = params
+        self._last_params = None if result_sql else params
 
     def _log_sql(self, sql: str, params: Sequence[Any] | dict[Any, Any] | None = None) -> None:
         if (fs_debug := os.environ.get("FAKESNOW_DEBUG")) and fs_debug != "snowflake":

fakesnow/server.py

@@ -16,6 +16,7 @@ from starlette.responses import JSONResponse
 from starlette.routing import Route
 
 from fakesnow.arrow import to_ipc, to_sf
+from fakesnow.converter import from_binding
 from fakesnow.fakes import FakeSnowflakeConnection
 from fakesnow.instance import FakeSnow
 from fakesnow.rowtype import describe_as_rowtype
@@ -77,9 +78,16 @@ async def query_request(request: Request) -> JSONResponse:
 
         sql_text = body_json["sqlText"]
 
+        params = None
+
+        if bindings := body_json.get("bindings"):
+            # Convert parameters like {'1': {'type': 'FIXED', 'value': '1'}, ...} to tuple (1, ...)
+            params = tuple(from_binding(bindings[str(pos)]) for pos in range(1, len(bindings) + 1))
+            logger.debug(f"Bindings: {params}")
+
         try:
             # only a single sql statement is sent at a time by the python snowflake connector
-            cur = await run_in_threadpool(conn.cursor().execute, sql_text)
+            cur = await run_in_threadpool(conn.cursor().execute, sql_text, binding_params=params)
             rowtype = describe_as_rowtype(cur._describe_last_sql())  # noqa: SLF001
 
         except snowflake.connector.errors.ProgrammingError as e:

fakesnow/transforms/__init__.py

@@ -59,9 +59,13 @@ def alter_table_strip_cluster_by(expression: exp.Expression) -> exp.Expression:
 
 def array_size(expression: exp.Expression) -> exp.Expression:
     if isinstance(expression, exp.ArraySize):
-        # case is used to convert 0 to null, because null is returned by duckdb when no case matches
+        # return null if not json array
         jal = exp.Anonymous(this="json_array_length", expressions=[expression.this])
-        return exp.Case(ifs=[exp.If(this=jal, true=jal)])
+        is_json_array = exp.EQ(
+            this=exp.Anonymous(this="json_type", expressions=[expression.this]),
+            expression=exp.Literal(this="ARRAY", is_string=True),
+        )
+        return exp.Case(ifs=[exp.If(this=is_json_array, true=jal)])
 
     return expression
 

{fakesnow-0.9.32.dist-info → fakesnow-0.9.34.dist-info}/METADATA

@@ -1,6 +1,6 @@
-Metadata-Version: 2.2
+Metadata-Version: 2.4
 Name: fakesnow
-Version: 0.9.32
+Version: 0.9.34
 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~=1.2.0
 Requires-Dist: pyarrow
 Requires-Dist: snowflake-connector-python
-Requires-Dist: sqlglot~=26.10.1
+Requires-Dist: sqlglot~=26.12.1
 Provides-Extra: dev
 Requires-Dist: build~=1.0; extra == "dev"
 Requires-Dist: dirty-equals; extra == "dev"
@@ -233,6 +233,7 @@ Requires-Dist: jupysql; extra == "notebook"
 Provides-Extra: server
 Requires-Dist: starlette; extra == "server"
 Requires-Dist: uvicorn; extra == "server"
+Dynamic: license-file
 
 # fakesnow ❄️
 

{fakesnow-0.9.32.dist-info → fakesnow-0.9.34.dist-info}/RECORD

@@ -4,7 +4,8 @@ fakesnow/arrow.py,sha256=XjTpFyLrD9jULWOtPgpr0RyNMmO6a5yi82y6ivi2CCI,4884
 fakesnow/checks.py,sha256=be-xo0oMoAUVhlMDCu1_Rkoh_L8p_p8qo9P6reJSHIQ,2874
 fakesnow/cli.py,sha256=9qfI-Ssr6mo8UmIlXkUAOz2z2YPBgDsrEVaZv9FjGFs,2201
 fakesnow/conn.py,sha256=2WClMmUgfQkQA2hFQjfMP3R-85TbTbZh_8Y1tCdcerA,6053
-fakesnow/cursor.py,sha256=Nvr8TQmmTFs6i0sJwfgCocrEF9td0D0SdDG41quIudI,21621
+fakesnow/converter.py,sha256=7YlASaMomzchMZoorTH3KtVmgBakaHrF5fAl5VP747I,1635
+fakesnow/cursor.py,sha256=mK4nC1iucON1MohicTugJqUOfRsx5c8ToUJgnCfUSbs,21813
 fakesnow/expr.py,sha256=CAxuYIUkwI339DQIBzvFF0F-m1tcVGKEPA5rDTzmH9A,892
 fakesnow/fakes.py,sha256=JQTiUkkwPeQrJ8FDWhPFPK6pGwd_aR2oiOrNzCWznlM,187
 fakesnow/fixtures.py,sha256=G-NkVeruSQAJ7fvSS2fR2oysUn0Yra1pohHlOvacKEk,455
@@ -14,15 +15,15 @@ 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=oLnWJgcxwhPw4sNebJF3B9uxk28A-a-AI8Pyl_lz2_E,5986
+fakesnow/server.py,sha256=4DgZUTd-G_usjSqy6NdUqd2fWUw2a-wHSSeJt3cdneA,6375
 fakesnow/variables.py,sha256=WXyPnkeNwD08gy52yF66CVe2twiYC50tztNfgXV4q1k,3032
-fakesnow/transforms/__init__.py,sha256=gD8wPo9QprwHkTOEMQ0-IsXSNfUruU0kBJPjO0po-J4,49377
+fakesnow/transforms/__init__.py,sha256=xFrpw28DaHvMt6LGaRMsPqTo8PWogg10JgEu3oa6jdA,49515
 fakesnow/transforms/merge.py,sha256=Pg7_rwbAT_vr1U4ocBofUSyqaK8_e3qdIz_2SDm2S3s,8320
 fakesnow/transforms/show.py,sha256=2qfK3Fi0RLylqTnkwSVgv5JIorXYb1y0fnf5oErRZ2o,16839
+fakesnow-0.9.34.dist-info/licenses/LICENSE,sha256=kW-7NWIyaRMQiDpryfSmF2DObDZHGR1cJZ39s6B1Svg,11344
 tools/decode.py,sha256=kC5kUvLQxdCkMRsnH6BqCajlKxKeN77w6rwCKsY6gqU,1781
-fakesnow-0.9.32.dist-info/LICENSE,sha256=kW-7NWIyaRMQiDpryfSmF2DObDZHGR1cJZ39s6B1Svg,11344
-fakesnow-0.9.32.dist-info/METADATA,sha256=HTGvul9rrgbrcSK79gmq4KndGAga0be6WzaKSUc_3c4,18106
-fakesnow-0.9.32.dist-info/WHEEL,sha256=beeZ86-EfXScwlR_HKu4SllMC9wUEj_8Z_4FJ3egI2w,91
-fakesnow-0.9.32.dist-info/entry_points.txt,sha256=2riAUgu928ZIHawtO8EsfrMEJhi-EH-z_Vq7Q44xKPM,47
-fakesnow-0.9.32.dist-info/top_level.txt,sha256=Yos7YveA3f03xVYuURqnBsfMV2DePXfu_yGcsj3pPzI,30
-fakesnow-0.9.32.dist-info/RECORD,,
+fakesnow-0.9.34.dist-info/METADATA,sha256=Hqkb8CT1-QTNzQqRMVhEHWGwx3gJORv9YHy7wGSoBgQ,18128
+fakesnow-0.9.34.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
+fakesnow-0.9.34.dist-info/entry_points.txt,sha256=2riAUgu928ZIHawtO8EsfrMEJhi-EH-z_Vq7Q44xKPM,47
+fakesnow-0.9.34.dist-info/top_level.txt,sha256=Yos7YveA3f03xVYuURqnBsfMV2DePXfu_yGcsj3pPzI,30
+fakesnow-0.9.34.dist-info/RECORD,,

{fakesnow-0.9.32.dist-info → fakesnow-0.9.34.dist-info}/WHEEL

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