fakesnow 0.4.1__py3-none-any.whl → 0.5.1__py3-none-any.whl

fakesnow/fakes.py

@@ -46,6 +46,7 @@ class FakeSnowflakeCursor:
         self._use_dict_result = use_dict_result
         self._last_sql = None
         self._last_params = None
+        self._sqlstate = None
 
     def __enter__(self) -> Self:
         return self
@@ -76,7 +77,7 @@ class FakeSnowflakeCursor:
         # use a cursor to avoid destroying an unfetched result on the main connection
         with self._duck_conn.cursor() as cur:
             assert self._conn.database, "Not implemented when database is None"
-            assert self._conn.schema, "Not implemented when database is None"
+            assert self._conn.schema, "Not implemented when schema is None"
 
             # match database and schema used on the main connection
             cur.execute(f"SET SCHEMA = '{self._conn.database}.{self._conn.schema}'")
@@ -91,6 +92,20 @@ class FakeSnowflakeCursor:
         params: Sequence[Any] | dict[Any, Any] | None = None,
         *args: Any,
         **kwargs: Any,
+    ) -> FakeSnowflakeCursor:
+        try:
+            self._sqlstate = None
+            return self._execute(command, params, *args, **kwargs)
+        except snowflake.connector.errors.ProgrammingError as e:
+            self._sqlstate = e.sqlstate
+            raise e
+
+    def _execute(
+        self,
+        command: str | exp.Expression,
+        params: Sequence[Any] | dict[Any, Any] | None = None,
+        *args: Any,
+        **kwargs: Any,
     ) -> FakeSnowflakeCursor:
         self._arrow_table = None
 
@@ -108,13 +123,13 @@ class FakeSnowflakeCursor:
                 msg=f"Cannot perform {cmd}. This session does not have a current database. Call 'USE DATABASE', or use a qualified name.",  # noqa: E501
                 errno=90105,
                 sqlstate="22000",
-            ) from None
+            )
         elif no_schema and not self._conn.schema_set:
             raise snowflake.connector.errors.ProgrammingError(
                 msg=f"Cannot perform {cmd}. This session does not have a current schema. Call 'USE SCHEMA', or use a qualified name.",  # noqa: E501
                 errno=90106,
                 sqlstate="22000",
-            ) from None
+            )
 
         transformed = (
             expression.transform(transforms.upper_case_unquoted_identifiers)
@@ -247,13 +262,21 @@ class FakeSnowflakeCursor:
         # TODO: return number of rows updated/inserted (using returning)
         return None
 
+    @property
+    def sfqid(self) -> str | None:
+        return "fakesnow"
+
+    @property
+    def sqlstate(self) -> str | None:
+        return self._sqlstate
+
     @staticmethod
     def _describe_as_result_metadata(describe_results: list) -> list[ResultMetadata]:
         # fmt: off
         def as_result_metadata(column_name: str, column_type: str, _: str) -> ResultMetadata:
             # see https://docs.snowflake.com/en/user-guide/python-connector-api.html#type-codes
             # and https://arrow.apache.org/docs/python/api/datatypes.html#type-checking
-            if column_type == "BIGINT":
+            if column_type in {"BIGINT", "INTEGER"}:
                 return ResultMetadata(
                     name=column_name, type_code=0, display_size=None, internal_size=None, precision=38, scale=0, is_nullable=True               # noqa: E501
                 )
@@ -288,6 +311,14 @@ class FakeSnowflakeCursor:
                 return ResultMetadata(
                     name=column_name, type_code=8, display_size=None, internal_size=None, precision=0, scale=9, is_nullable=True                # noqa: E501
                 )
+            elif column_type == "BLOB":
+                return ResultMetadata(
+                    name=column_name, type_code=11, display_size=None, internal_size=8388608, precision=None, scale=None, is_nullable=True      # noqa: E501
+                )
+            elif column_type == "TIME":
+                return ResultMetadata(
+                    name=column_name, type_code=12, display_size=None, internal_size=None, precision=0, scale=9, is_nullable=True               # noqa: E501
+                )
             else:
                 # TODO handle more types
                 raise NotImplementedError(f"for column type {column_type}")

fakesnow/info_schema.py

@@ -37,11 +37,18 @@ create view ${catalog}.information_schema.columns_snowflake AS
 select table_catalog, table_schema, table_name, column_name, ordinal_position, column_default, is_nullable,
 case when starts_with(data_type, 'DECIMAL') or data_type='BIGINT' then 'NUMBER'
      when data_type='VARCHAR' then 'TEXT'
+     when data_type='DOUBLE' then 'FLOAT'
+     when data_type='BLOB' then 'BINARY'
+     when data_type='TIMESTAMP' then 'TIMESTAMP_NTZ'
      else data_type end as data_type,
 ext_character_maximum_length as character_maximum_length, ext_character_octet_length as character_octet_length,
-case when data_type='BIGINT' then 38 else numeric_precision end as numeric_precision,
-case when data_type='BIGINT' then 10 else numeric_precision_radix end as numeric_precision_radix,
-numeric_scale,
+case when data_type='BIGINT' then 38
+     when data_type='DOUBLE' then NULL
+    else numeric_precision end as numeric_precision,
+case when data_type='BIGINT' then 10
+    when data_type='DOUBLE' then NULL
+    else numeric_precision_radix end as numeric_precision_radix,
+case when data_type='DOUBLE' then NULL else numeric_scale end as numeric_scale,
 collation_name, is_identity, identity_generation, identity_cycle
 from ${catalog}.information_schema.columns
 left join ${catalog}.information_schema.columns_ext ext

fakesnow/transforms.py

@@ -181,10 +181,7 @@ def float_to_double(expression: exp.Expression) -> exp.Expression:
     """
 
     if isinstance(expression, exp.DataType) and expression.this == exp.DataType.Type.FLOAT:
-        # TODO don't copy!
-        new = expression.copy()
-        new.args["this"] = exp.DataType.Type.DOUBLE
-        return new
+        expression.args["this"] = exp.DataType.Type.DOUBLE
 
     return expression
 

{fakesnow-0.4.1.dist-info → fakesnow-0.5.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: fakesnow
-Version: 0.4.1
+Version: 0.5.1
 Summary: Fake Snowflake Connector for Python. Run Snowflake DB locally.
 License: MIT License
         
@@ -124,12 +124,13 @@ def _fakesnow_session() -> Iterator[None]:
 
 ## Implementation coverage
 
-- [x] multiple databases
-- [x] cursors
+- [x] cursors and standard SQL
 - [x] [get_result_batches()](https://docs.snowflake.com/en/user-guide/python-connector-api#get_result_batches)
-- [x] [write_pandas(..)](https://docs.snowflake.com/en/user-guide/python-connector-api#write_pandas)
-- [x] table comments
+- [x] information schema
+- [x] multiple databases
 - [x] [qmark binding](https://docs.snowflake.com/en/user-guide/python-connector-example#binding-data)
+- [x] table comments
+- [x] [write_pandas(..)](https://docs.snowflake.com/en/user-guide/python-connector-api#write_pandas)
 - [ ] [access control](https://docs.snowflake.com/en/user-guide/security-access-control-overview)
 - [ ] standalone/out of process api/support for faking non-python connectors
 - [ ] [stored procedures](https://docs.snowflake.com/en/sql-reference/stored-procedures)
@@ -137,8 +138,9 @@ def _fakesnow_session() -> Iterator[None]:
 Partial support
 
 - [x] date functions
-- [x] tags
+- [x] regex functions
 - [x] semi-structured data
+- [x] tags
 
 For more detail see [tests/test_fakes.py](tests/test_fakes.py)
 

fakesnow-0.5.1.dist-info/RECORD

@@ -0,0 +1,13 @@
+fakesnow/__init__.py,sha256=ogZl2Z61FIiU2fg8WiM1FhXYLpG-Phvg855nmuMhdqQ,2992
+fakesnow/checks.py,sha256=1qVLR0ZB3z3UPij3Hm8hqlkcNLH2QJnwe8OqkoFCwv8,2356
+fakesnow/expr.py,sha256=CAxuYIUkwI339DQIBzvFF0F-m1tcVGKEPA5rDTzmH9A,892
+fakesnow/fakes.py,sha256=v6t0sLy9KsiiUW-KyRggamt7K7ADEw8uSXdQt6HmQN4,20760
+fakesnow/fixtures.py,sha256=LANb4LuiUjKbTZRHmgnAi50xC1rs1xF8SHLoBikB88c,509
+fakesnow/info_schema.py,sha256=0eJgDJ0sbEUVNcfGBb5hdCR2SiKJYmO0MA-ZnnsUhbo,3781
+fakesnow/py.typed,sha256=B-DLSjYBi7pkKjwxCSdpVj2J02wgfJr-E7B1wOUyxYU,80
+fakesnow/transforms.py,sha256=PTK3RA1SEiCT9urw97tLyhzXZuWZx5hCUWcvPO2DIwI,22305
+fakesnow-0.5.1.dist-info/LICENSE,sha256=BL6v_VTnU7xdsocviIQJMFr3stX_-uRfTyByo3gRu4M,1071
+fakesnow-0.5.1.dist-info/METADATA,sha256=H2Lo5Okt1MPsbpBW2CfeDYX9Yti-d8sTUOWFNAaHW8k,5225
+fakesnow-0.5.1.dist-info/WHEEL,sha256=AtBG6SXL3KF_v0NxLf0ehyVOh0cold-JbJYXNGorC6Q,92
+fakesnow-0.5.1.dist-info/top_level.txt,sha256=x8S-sMmvfgNm2_1w0zlIF5YlDs2hR7eNQdVA6TgmPZE,14
+fakesnow-0.5.1.dist-info/RECORD,,

{fakesnow-0.4.1.dist-info → fakesnow-0.5.1.dist-info}/WHEEL

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

fakesnow-0.4.1.dist-info/RECORD

@@ -1,13 +0,0 @@
-fakesnow/__init__.py,sha256=ogZl2Z61FIiU2fg8WiM1FhXYLpG-Phvg855nmuMhdqQ,2992
-fakesnow/checks.py,sha256=1qVLR0ZB3z3UPij3Hm8hqlkcNLH2QJnwe8OqkoFCwv8,2356
-fakesnow/expr.py,sha256=CAxuYIUkwI339DQIBzvFF0F-m1tcVGKEPA5rDTzmH9A,892
-fakesnow/fakes.py,sha256=6PYHwykRDXaHM1sET-A-ZGTrDrQ0DPyjsxWgpPW-mP0,19626
-fakesnow/fixtures.py,sha256=LANb4LuiUjKbTZRHmgnAi50xC1rs1xF8SHLoBikB88c,509
-fakesnow/info_schema.py,sha256=4GnYuAWsX9qfQKQyTd-GkI24DHAxHCmGCst6ohcQAXY,3495
-fakesnow/py.typed,sha256=B-DLSjYBi7pkKjwxCSdpVj2J02wgfJr-E7B1wOUyxYU,80
-fakesnow/transforms.py,sha256=VqkxxtDu8RbetcDMBm4nMCGyzuXwpU8-0WqXkmKSEhU,22376
-fakesnow-0.4.1.dist-info/LICENSE,sha256=BL6v_VTnU7xdsocviIQJMFr3stX_-uRfTyByo3gRu4M,1071
-fakesnow-0.4.1.dist-info/METADATA,sha256=5rRxOoTmT7yoJ75KJf211cXrey4LWrQ1wWvoyxCzhNo,5161
-fakesnow-0.4.1.dist-info/WHEEL,sha256=pkctZYzUS4AYVn6dJ-7367OJZivF2e8RA9b_ZBjif18,92
-fakesnow-0.4.1.dist-info/top_level.txt,sha256=x8S-sMmvfgNm2_1w0zlIF5YlDs2hR7eNQdVA6TgmPZE,14
-fakesnow-0.4.1.dist-info/RECORD,,