fakesnow 0.9.42__py3-none-any.whl → 0.9.43__py3-none-any.whl

fakesnow/cursor.py

@@ -10,7 +10,6 @@ from types import TracebackType
 from typing import TYPE_CHECKING, Any, cast
 
 import duckdb
-import pandas as pd
 import pyarrow  # needed by fetch_arrow_table()
 import snowflake.connector.converter
 import snowflake.connector.errors
@@ -194,7 +193,7 @@ class FakeSnowflakeCursor:
 
     def _put_files(self, put_stage_data: stage.UploadCommandDict) -> None:
         results = stage.upload_files(put_stage_data)
-        _df = pd.DataFrame.from_records(results)
+        _df = pyarrow.Table.from_pylist(results)
         self._duck_conn.execute("select * from _df")
         self._arrow_table = self._duck_conn.fetch_arrow_table()
         self._rowcount = self._arrow_table.num_rows

fakesnow/expr.py

@@ -30,3 +30,14 @@ def key_command(expression: exp.Expression) -> str:
         key = expression.key.upper()
 
     return key
+
+
+def normalise_ident(name: str) -> str:
+    """
+    Strip double quotes if present else return uppercased.
+    Snowflake treats quoted identifiers as case-sensitive and un-quoted identifiers as case-insensitive
+    """
+    if name.startswith('"') and name.endswith('"'):
+        return name[1:-1]  # Strip quotes
+
+    return name.upper()

fakesnow/server.py

@@ -18,6 +18,7 @@ from starlette.routing import Route
 
 from fakesnow.arrow import to_ipc, to_sf
 from fakesnow.converter import from_binding
+from fakesnow.expr import normalise_ident
 from fakesnow.fakes import FakeSnowflakeConnection
 from fakesnow.instance import FakeSnow
 from fakesnow.rowtype import describe_as_rowtype
@@ -39,8 +40,8 @@ class ServerError(Exception):
 
 
 async def login_request(request: Request) -> JSONResponse:
-    database = request.query_params.get("databaseName")
-    schema = request.query_params.get("schemaName")
+    database = (d := request.query_params.get("databaseName")) and normalise_ident(d)
+    schema = (s := request.query_params.get("schemaName")) and normalise_ident(s)
     body = await request.body()
     if request.headers.get("Content-Encoding") == "gzip":
         body = gzip.decompress(body)
@@ -64,7 +65,10 @@ async def login_request(request: Request) -> JSONResponse:
                     {"name": "AUTOCOMMIT", "value": True},
                     {"name": "CLIENT_SESSION_KEEP_ALIVE_HEARTBEAT_FREQUENCY", "value": 3600},
                 ],
-                "sessionInfo": {},
+                "sessionInfo": {
+                    "databaseName": database,
+                    "schemaName": schema,
+                },
             },
             "success": True,
         }
@@ -147,6 +151,8 @@ async def query_request(request: Request) -> JSONResponse:
                     "total": cur._rowcount,  # noqa: SLF001
                     "queryId": cur.sfqid,
                     "queryResultFormat": "arrow",
+                    "finalDatabaseName": conn.database,
+                    "finalSchemaName": conn.schema,
                 },
                 "success": True,
             }

fakesnow/transforms/stage.py

@@ -13,6 +13,7 @@ import sqlglot
 from snowflake.connector.file_util import SnowflakeFileUtil
 from sqlglot import exp
 
+from fakesnow.expr import normalise_ident
 from fakesnow.params import MutableParams
 
 # TODO: clean up temp files on exit
@@ -198,17 +199,6 @@ def put_stage(
     return transformed
 
 
-def normalise_ident(name: str) -> str:
-    """
-    Strip double quotes if present else return uppercased.
-    Snowflake treats quoted identifiers as case-sensitive and un-quoted identifiers as case-insensitive
-    """
-    if name.startswith('"') and name.endswith('"'):
-        return name[1:-1]  # Strip quotes
-
-    return name.upper()
-
-
 def parts_from_var(var: str, current_database: str | None, current_schema: str | None) -> tuple[str, str, str]:
     parts = var.split(".")
     if len(parts) == 3:

{fakesnow-0.9.42.dist-info → fakesnow-0.9.43.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: fakesnow
-Version: 0.9.42
+Version: 0.9.43
 Summary: Fake Snowflake Connector for Python. Run, mock and test Snowflake DB locally.
 License:                                  Apache License
                                    Version 2.0, January 2004
@@ -420,7 +420,7 @@ Fully supported:
 - Multiple databases
 - [Parameter binding](https://docs.snowflake.com/en/user-guide/python-connector-example#binding-data) in queries
 - Table comments
-- Pandas integration including [write_pandas(..)](https://docs.snowflake.com/en/user-guide/python-connector-api#write_pandas) (not available via the server yet)
+- Pandas integration including [write_pandas(..)](https://docs.snowflake.com/en/user-guide/python-connector-api#write_pandas)
 - Result batch retrieval via [get_result_batches()](https://docs.snowflake.com/en/user-guide/python-connector-api#get_result_batches)
 - HTTP server for non-Python connectors
 
@@ -431,7 +431,8 @@ Partially supported:
 - Semi-structured data operations
 - Tags
 - User management
-- `COPY INTO` from S3 sources, see [COPY INTO](#copy-into)
+- Stages and PUT
+- `COPY INTO` from S3 sources and stages, see [COPY INTO](#copy-into)
 
 Not yet implemented:
 
@@ -447,7 +448,7 @@ For more detail see the [test suite](tests/).
 
 ## COPY INTO
 
-`COPY INTO` can be used from S3 sources. By default the standard AWS credential chain will be used. If you are getting an HTTP 403 or need to provide alternative S3 credentials you can use the duckdb [CREATE SECRET](https://duckdb.org/docs/stable/extensions/httpfs/s3api) statement. For an example of creating a secret to use a moto S3 endpoint see `s3_client` in [conftest.py](tests/conftest.py#L80)
+`COPY INTO` can be used from S3 sources and stages. By default the standard AWS credential chain will be used. If you are getting an HTTP 403 or need to provide alternative S3 credentials you can use the duckdb [CREATE SECRET](https://duckdb.org/docs/stable/extensions/httpfs/s3api) statement. For an example of creating a secret to use a moto S3 endpoint see `s3_client` in [conftest.py](tests/conftest.py#L80)
 
 ## Contributing
 

{fakesnow-0.9.42.dist-info → fakesnow-0.9.43.dist-info}/RECORD

@@ -6,8 +6,8 @@ fakesnow/cli.py,sha256=9qfI-Ssr6mo8UmIlXkUAOz2z2YPBgDsrEVaZv9FjGFs,2201
 fakesnow/conn.py,sha256=diCwcjaCBrlCn9PyjbScfIQTNQjqiPTkQanUTqcvblE,6009
 fakesnow/converter.py,sha256=wPOfsFXIUJNJSx5oFNAxh13udxmAVIIHsLK8BiGkXGA,1635
 fakesnow/copy_into.py,sha256=utlV03RWHdWblIlKs92Q__9BWHi_gvqT76RV6tEMWTo,16006
-fakesnow/cursor.py,sha256=T0aI_AC8U7_mRMccGUIOs8WFby4-8O3ETHpcxJLtM38,25432
-fakesnow/expr.py,sha256=CAxuYIUkwI339DQIBzvFF0F-m1tcVGKEPA5rDTzmH9A,892
+fakesnow/cursor.py,sha256=A1jWLGkXzHcYbM0wFrbh2PiewzFFS7ZlfOm-x3AGHgg,25412
+fakesnow/expr.py,sha256=tLcHC-mx2O6-vCt77RVlRQa3CovqUdu4vc5WOSmuQdk,1231
 fakesnow/fakes.py,sha256=JQTiUkkwPeQrJ8FDWhPFPK6pGwd_aR2oiOrNzCWznlM,187
 fakesnow/fixtures.py,sha256=2rj0MTZlaZc4PNWhaqC5IiiLa7E9G0QZT3g45YawsL0,633
 fakesnow/info_schema.py,sha256=lqEYD5aWK2MamjALbj6ct7pz_1yyAq3tAk51kLa8NKk,9872
@@ -18,17 +18,17 @@ fakesnow/pandas_tools.py,sha256=wI203UQHC8JvDzxE_VjE1NeV4rThek2P-u52oTg2foo,3481
 fakesnow/params.py,sha256=Hp7tBiDycdOh8LuRoISsOZycD7DufDAMKyF_hqEjh6M,929
 fakesnow/py.typed,sha256=B-DLSjYBi7pkKjwxCSdpVj2J02wgfJr-E7B1wOUyxYU,80
 fakesnow/rowtype.py,sha256=QUp8EaXD5LT0Xv8BXk5ze4WseEn52xoJ6R05pJjs5mM,2729
-fakesnow/server.py,sha256=anlieEI2WO81rDv7h5-M10CetYL_HJFXWZO9wScr0Dw,7196
+fakesnow/server.py,sha256=NzhpVM_3L9QeG84bhn0_It6ABM9ifs1CYWPgKhWaOeE,7511
 fakesnow/variables.py,sha256=BGnD4LAdVByfJ2GXL6qpGBaTF8ZJRjt3pdJsd9sIAcw,3134
 fakesnow/transforms/__init__.py,sha256=3dpcXjeLkfc4e9rO0G8Mnv9zZgmwsKSwQ7UJeyLYnOg,2837
 fakesnow/transforms/merge.py,sha256=H2yYyzGsxjpggS6PY91rAUtYCFuOkBm8uGi0EO0r6T0,8375
 fakesnow/transforms/show.py,sha256=NIJDkf4-ns4VTw0DQphEn2DnwxQjXyTU2wS9VSUh6Wc,19919
-fakesnow/transforms/stage.py,sha256=yBC0jMmUYhmHY5Of08lPxgH9R6arNUWAjL9M4ArDnjY,10344
+fakesnow/transforms/stage.py,sha256=VRE122Twha1Dd7QC2rdAAWWG93OxIiJZv2DfvGSsMZ8,10047
 fakesnow/transforms/transforms.py,sha256=ykzay7J28MsifhRKhm5qJJd__hMcLNVZTT14MSdWcJs,49102
-fakesnow-0.9.42.dist-info/licenses/LICENSE,sha256=kW-7NWIyaRMQiDpryfSmF2DObDZHGR1cJZ39s6B1Svg,11344
+fakesnow-0.9.43.dist-info/licenses/LICENSE,sha256=kW-7NWIyaRMQiDpryfSmF2DObDZHGR1cJZ39s6B1Svg,11344
 tools/decode.py,sha256=kC5kUvLQxdCkMRsnH6BqCajlKxKeN77w6rwCKsY6gqU,1781
-fakesnow-0.9.42.dist-info/METADATA,sha256=tYGXY_3aBbYYkLyFlajE1ebKbXT4xhyain0JLC9vfb0,20680
-fakesnow-0.9.42.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-fakesnow-0.9.42.dist-info/entry_points.txt,sha256=2riAUgu928ZIHawtO8EsfrMEJhi-EH-z_Vq7Q44xKPM,47
-fakesnow-0.9.42.dist-info/top_level.txt,sha256=Yos7YveA3f03xVYuURqnBsfMV2DePXfu_yGcsj3pPzI,30
-fakesnow-0.9.42.dist-info/RECORD,,
+fakesnow-0.9.43.dist-info/METADATA,sha256=tUAcWIfezqI1Lu64-sz8VToB4bFIyfaEFyOPCJpDREY,20684
+fakesnow-0.9.43.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+fakesnow-0.9.43.dist-info/entry_points.txt,sha256=2riAUgu928ZIHawtO8EsfrMEJhi-EH-z_Vq7Q44xKPM,47
+fakesnow-0.9.43.dist-info/top_level.txt,sha256=Yos7YveA3f03xVYuURqnBsfMV2DePXfu_yGcsj3pPzI,30
+fakesnow-0.9.43.dist-info/RECORD,,