fakesnow 0.9.30__py3-none-any.whl → 0.9.31__py3-none-any.whl

fakesnow/arrow.py

@@ -62,7 +62,7 @@ def to_ipc(table: pa.Table) -> pa.Buffer:
 
 
 def to_sf(table: pa.Table, rowtype: list[ColumnInfo]) -> pa.Table:
-    def to_sf_col(col: pa.Array) -> pa.Array:
+    def to_sf_col(col: pa.ChunkedArray) -> pa.Array | pa.ChunkedArray:
         if pa.types.is_timestamp(col.type):
             return timestamp_to_sf_struct(col)
         elif pa.types.is_time(col.type):
@@ -83,7 +83,7 @@ def timestamp_to_sf_struct(ts: pa.Array | pa.ChunkedArray) -> pa.Array:
 
     # Round to seconds, ie: strip subseconds
     tsa_without_us = pc.floor_temporal(ts, unit="second")  # type: ignore https://github.com/zen-xu/pyarrow-stubs/issues/45
-    epoch = pc.divide(tsa_without_us.cast(pa.int64()), 1_000_000)  # type: ignore https://github.com/zen-xu/pyarrow-stubs/issues/44
+    epoch = pc.divide(tsa_without_us.cast(pa.int64()), 1_000_000)
 
     # Calculate fractional part as nanoseconds
     fraction = pc.multiply(pc.subsecond(ts), 1_000_000_000).cast(pa.int32())  # type: ignore
@@ -93,7 +93,7 @@ def timestamp_to_sf_struct(ts: pa.Array | pa.ChunkedArray) -> pa.Array:
         timezone = pa.array([1440] * len(ts), type=pa.int32())
 
         return pa.StructArray.from_arrays(
-            arrays=[epoch, fraction, timezone],  # type: ignore https://github.com/zen-xu/pyarrow-stubs/issues/42
+            arrays=[epoch, fraction, timezone],
             fields=[
                 pa.field("epoch", nullable=False, type=pa.int64()),
                 pa.field("fraction", nullable=False, type=pa.int32()),
@@ -102,7 +102,7 @@ def timestamp_to_sf_struct(ts: pa.Array | pa.ChunkedArray) -> pa.Array:
         )
     else:
         return pa.StructArray.from_arrays(
-            arrays=[epoch, fraction],  # type: ignore https://github.com/zen-xu/pyarrow-stubs/issues/42
+            arrays=[epoch, fraction],
             fields=[
                 pa.field("epoch", nullable=False, type=pa.int64()),
                 pa.field("fraction", nullable=False, type=pa.int32()),

fakesnow/cursor.py

@@ -14,6 +14,7 @@ import pyarrow  # needed by fetch_arrow_table()
 import snowflake.connector.converter
 import snowflake.connector.errors
 import sqlglot
+import sqlglot.errors
 from duckdb import DuckDBPyConnection
 from snowflake.connector.cursor import ResultMetadata
 from snowflake.connector.result_batch import ResultBatch
@@ -155,6 +156,11 @@ class FakeSnowflakeCursor:
         except snowflake.connector.errors.ProgrammingError as e:
             self._sqlstate = e.sqlstate
             raise e
+        except sqlglot.errors.ParseError as e:
+            self._sqlstate = "42000"
+            # strip highlight for better readability, TODO: show pointer to start of error
+            msg = str(e).replace("\x1b[4m", "").replace("\x1b[0m", "")
+            raise snowflake.connector.errors.ProgrammingError(msg=msg, errno=1003, sqlstate="42000") from None
 
     def check_db_and_schema(self, expression: exp.Expression) -> None:
         no_database, no_schema = checks.is_unqualified_table_expression(expression)

fakesnow/transforms/__init__.py

@@ -720,15 +720,21 @@ def information_schema_fs_views(expression: exp.Expression) -> exp.Expression:
     return expression
 
 
+NUMBER_38_0 = [
+    exp.DataTypeParam(this=exp.Literal(this="38", is_string=False)),
+    exp.DataTypeParam(this=exp.Literal(this="0", is_string=False)),
+]
+
+
 def integer_precision(expression: exp.Expression) -> exp.Expression:
-    """Convert integers to bigint.
+    """Convert integers and number(38,0) to bigint.
 
-    So dataframes will return them with a dtype of int64.
+    So fetch_all will return int and dataframes will return them with a dtype of int64.
     """
-
     if (
         isinstance(expression, exp.DataType)
-        and (expression.this == exp.DataType.Type.DECIMAL and not expression.expressions)
+        and expression.this == exp.DataType.Type.DECIMAL
+        and (not expression.expressions or expression.expressions == NUMBER_38_0)
     ) or expression.this in (exp.DataType.Type.INT, exp.DataType.Type.SMALLINT, exp.DataType.Type.TINYINT):
         return exp.DataType(
             this=exp.DataType.Type.BIGINT,
@@ -836,29 +842,29 @@ def object_construct(expression: exp.Expression) -> exp.Expression:
     See https://docs.snowflake.com/en/sql-reference/functions/object_construct
     """
 
-    if isinstance(expression, exp.Struct):
-        non_null_expressions = []
-        for e in expression.expressions:
-            if not (isinstance(e, exp.PropertyEQ)):
-                non_null_expressions.append(e)
-                continue
+    if not isinstance(expression, exp.Struct):
+        return expression
 
-            left = e.left
-            right = e.right
+    non_null_expressions = []
+    for e in expression.expressions:
+        if not (isinstance(e, exp.PropertyEQ)):
+            non_null_expressions.append(e)
+            continue
 
-            left_is_null = isinstance(left, exp.Null)
-            right_is_null = isinstance(right, exp.Null)
+        left = e.left
+        right = e.right
 
-            if left_is_null or right_is_null:
-                continue
+        left_is_null = isinstance(left, exp.Null)
+        right_is_null = isinstance(right, exp.Null)
 
-            non_null_expressions.append(e)
+        if left_is_null or right_is_null:
+            continue
 
-        new_struct = expression.copy()
-        new_struct.set("expressions", non_null_expressions)
-        return exp.Anonymous(this="TO_JSON", expressions=[new_struct])
+        non_null_expressions.append(e)
 
-    return expression
+    new_struct = expression.copy()
+    new_struct.set("expressions", non_null_expressions)
+    return exp.Anonymous(this="TO_JSON", expressions=[new_struct])
 
 
 def regex_replace(expression: exp.Expression) -> exp.Expression:
@@ -1023,17 +1029,10 @@ def show_objects_tables(expression: exp.Expression, current_database: str | None
         catalog = table.db or current_database
         schema = table.name
     else:
-        # all objects / tables
+        # all objects / tables - will show everything in the "account"
         catalog = None
         schema = None
 
-    tables_only = "table_type = 'BASE TABLE' and " if show == "TABLES" else ""
-    exclude_fakesnow_tables = "not (table_schema == '_fs_information_schema')"
-    # without a database will show everything in the "account"
-    table_catalog = f" and table_catalog = '{catalog}'" if catalog else ""
-    schema = f" and table_schema = '{schema}'" if schema else ""
-    limit = limit.sql() if (limit := expression.args.get("limit")) and isinstance(limit, exp.Expression) else ""
-
     columns = [
         "to_timestamp(0)::timestamptz as 'created_on'",
         "table_name as 'name'",
@@ -1041,17 +1040,29 @@ def show_objects_tables(expression: exp.Expression, current_database: str | None
         "table_catalog as 'database_name'",
         "table_schema as 'schema_name'",
     ]
-
-    terse = expression.args["terse"]
-    if not terse:
+    if not expression.args["terse"]:
         columns.append('null as "comment"')
+    columns_clause = ", ".join(columns)
+
+    where = ["not (table_schema == '_fs_information_schema')"]  # exclude fakesnow's internal schemas
+    if show == "TABLES":
+        where.append("table_type = 'BASE TABLE'")
+    if catalog:
+        where.append(f"table_catalog = '{catalog}'")
+    if schema:
+        where.append(f"table_schema = '{schema}'")
+    if (like := expression.args.get("like")) and isinstance(like, exp.Expression):
+        where.append(f"table_name ilike {like.sql()}")
+    where_clause = " AND ".join(where)
 
-    columns_str = ", ".join(columns)
+    limit = limit.sql() if (limit := expression.args.get("limit")) and isinstance(limit, exp.Expression) else ""
 
-    query = (
-        f"SELECT {columns_str} from information_schema.tables "
-        f"where {tables_only}{exclude_fakesnow_tables}{table_catalog}{schema}{limit}"
-    )
+    query = f"""
+        SELECT {columns_clause}
+        from information_schema.tables
+        where {where_clause}
+        {limit}
+    """
 
     return sqlglot.parse_one(query, read="duckdb")
 
@@ -1296,8 +1307,6 @@ def _get_to_number_args(e: exp.ToNumber) -> tuple[exp.Expression | None, exp.Exp
                 # to_number('100', 'TM9', 10, 2)
                 if arg_scale:
                     _scale = arg_scale
-            else:
-                pass
         else:
             # to_number('100', 10, ...)
             # arg_format is not a string, so it must be precision.
@@ -1307,12 +1316,10 @@ def _get_to_number_args(e: exp.ToNumber) -> tuple[exp.Expression | None, exp.Exp
             # And arg_precision must be scale
             if arg_precision:
                 _scale = arg_precision
-    else:
-        # If format is not provided, just check for precision and scale directly
-        if arg_precision:
-            _precision = arg_precision
-            if arg_scale:
-                _scale = arg_scale
+    elif arg_precision:
+        _precision = arg_precision
+        if arg_scale:
+            _scale = arg_scale
 
     return _format, _precision, _scale
 
@@ -1633,8 +1640,7 @@ def show_keys(
                   AND table_name NOT LIKE '_fs_%'
                 """
 
-        scope_kind = expression.args.get("scope_kind")
-        if scope_kind:
+        if scope_kind := expression.args.get("scope_kind"):
             table = expression.args["scope"]
 
             if scope_kind == "SCHEMA":

{fakesnow-0.9.30.dist-info → fakesnow-0.9.31.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: fakesnow
-Version: 0.9.30
+Version: 0.9.31
 Summary: Fake Snowflake Connector for Python. Run, mock and test Snowflake DB locally.
 License:                                  Apache License
                                    Version 2.0, January 2004
@@ -220,7 +220,7 @@ 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~=4.0; extra == "dev"
-Requires-Dist: pyarrow-stubs==10.0.1.9; extra == "dev"
+Requires-Dist: pyarrow-stubs==17.19; extra == "dev"
 Requires-Dist: pytest~=8.0; extra == "dev"
 Requires-Dist: pytest-asyncio; extra == "dev"
 Requires-Dist: ruff~=0.11.0; extra == "dev"

{fakesnow-0.9.30.dist-info → fakesnow-0.9.31.dist-info}/RECORD

@@ -1,10 +1,10 @@
 fakesnow/__init__.py,sha256=qUfgucQYPdELrJaxczalhJgWAWQ6cfTCUAHx6nUqRaI,3528
 fakesnow/__main__.py,sha256=GDrGyNTvBFuqn_UfDjKs7b3LPtU6gDv1KwosVDrukIM,76
-fakesnow/arrow.py,sha256=MwatkdZX5AFADzXvxhBFmcRJVxbW4D39VoqLyhpTbl0,5057
+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=HGhFKErKWvAfVEy3QSc0tfNmzGh_T7FtvRfWuDBy_CQ,5744
-fakesnow/cursor.py,sha256=KP8aDhq_m10ibpLjiL4retcwUCh_8PsY5sZfEFY_3No,20970
+fakesnow/cursor.py,sha256=tlcQK_w4r6BZHs3BTNHUFGk2hZz-lSnOdIrNLaJ-rBw,21349
 fakesnow/expr.py,sha256=CAxuYIUkwI339DQIBzvFF0F-m1tcVGKEPA5rDTzmH9A,892
 fakesnow/fakes.py,sha256=JQTiUkkwPeQrJ8FDWhPFPK6pGwd_aR2oiOrNzCWznlM,187
 fakesnow/fixtures.py,sha256=G-NkVeruSQAJ7fvSS2fR2oysUn0Yra1pohHlOvacKEk,455
@@ -16,11 +16,11 @@ fakesnow/py.typed,sha256=B-DLSjYBi7pkKjwxCSdpVj2J02wgfJr-E7B1wOUyxYU,80
 fakesnow/rowtype.py,sha256=QUp8EaXD5LT0Xv8BXk5ze4WseEn52xoJ6R05pJjs5mM,2729
 fakesnow/server.py,sha256=-jKyEVuD2TEr88jUSA1Lu86MAymel7LQAiNlytHqhTg,5934
 fakesnow/variables.py,sha256=WXyPnkeNwD08gy52yF66CVe2twiYC50tztNfgXV4q1k,3032
-fakesnow/transforms/__init__.py,sha256=ENBHnwfQHAlC9PWOq4tdz-9-YQGy2E48xJB5ce7qEA0,60345
+fakesnow/transforms/__init__.py,sha256=a-WNNqIo6IhY9p_m0zfdre7eb6KDmz-dbyrNkhpsHQk,60506
 fakesnow/transforms/merge.py,sha256=Pg7_rwbAT_vr1U4ocBofUSyqaK8_e3qdIz_2SDm2S3s,8320
-fakesnow-0.9.30.dist-info/LICENSE,sha256=kW-7NWIyaRMQiDpryfSmF2DObDZHGR1cJZ39s6B1Svg,11344
-fakesnow-0.9.30.dist-info/METADATA,sha256=ryFSnkBU9Vb4BrYI9g9q16zN4M2env6w8-syT7LbJ1k,18109
-fakesnow-0.9.30.dist-info/WHEEL,sha256=52BFRY2Up02UkjOa29eZOS2VxUrpPORXg1pkohGGUS8,91
-fakesnow-0.9.30.dist-info/entry_points.txt,sha256=2riAUgu928ZIHawtO8EsfrMEJhi-EH-z_Vq7Q44xKPM,47
-fakesnow-0.9.30.dist-info/top_level.txt,sha256=500evXI1IFX9so82cizGIEMHAb_dJNPaZvd2H9dcKTA,24
-fakesnow-0.9.30.dist-info/RECORD,,
+fakesnow-0.9.31.dist-info/LICENSE,sha256=kW-7NWIyaRMQiDpryfSmF2DObDZHGR1cJZ39s6B1Svg,11344
+fakesnow-0.9.31.dist-info/METADATA,sha256=RXWi0hLbvLkUtzDw0SHtcu-N3-Ebd1BC5K7JfoPnZ38,18106
+fakesnow-0.9.31.dist-info/WHEEL,sha256=beeZ86-EfXScwlR_HKu4SllMC9wUEj_8Z_4FJ3egI2w,91
+fakesnow-0.9.31.dist-info/entry_points.txt,sha256=2riAUgu928ZIHawtO8EsfrMEJhi-EH-z_Vq7Q44xKPM,47
+fakesnow-0.9.31.dist-info/top_level.txt,sha256=500evXI1IFX9so82cizGIEMHAb_dJNPaZvd2H9dcKTA,24
+fakesnow-0.9.31.dist-info/RECORD,,

{fakesnow-0.9.30.dist-info → fakesnow-0.9.31.dist-info}/WHEEL

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