PyPI - sqlframe - Versions diffs - 3.19.0__tar.gz → 3.20.0__tar.gz - Mend

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: sqlframe
-Version: 3.19.0
+Version: 3.20.0
 Summary: Turning PySpark Into a Universal DataFrame API
 Home-page: https://github.com/eakmanrq/sqlframe
 Author: Ryan Eakman

@@ -266,7 +266,7 @@ See something that you would like to see supported? [Open an issue](https://gith
 * [acos](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.acos.html)
 * [add_months](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.add_months.html)
 * [any_value](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.any_value.html)
-  * Always ignores nulls
+    * Always ignores nulls
 * [approxCountDistinct](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.approxCountDistinct.html)
 * [approx_count_distinct](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.approx_count_distinct.html)
 * [array](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.array.html)

@@ -12,5 +12,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
-__version__ = version = '3.19.0'
-__version_tuple__ = version_tuple = (3, 19, 0)
+__version__ = version = '3.20.0'
+__version_tuple__ = version_tuple = (3, 20, 0)

@@ -3133,20 +3133,14 @@ def datepart(field: ColumnOrName, source: ColumnOrName) -> Column:
 @meta(unsupported_engines=["bigquery", "postgres", "snowflake"])
 def day(col: ColumnOrName) -> Column:
-    from sqlframe.base.function_alternatives import day_with_try_to_timestamp
     session = _get_session()
     if session._is_duckdb:
         try_to_timestamp = get_func_from_session("try_to_timestamp")
         to_date = get_func_from_session("to_date")
-        when = get_func_from_session("when")
         _is_string = get_func_from_session("_is_string")
         coalesce = get_func_from_session("coalesce")
-        col = when(
-            _is_string(col),
-            coalesce(try_to_timestamp(col), to_date(col)),
-        ).otherwise(col)
+        col = coalesce(try_to_timestamp(Column.ensure_col(col).cast("VARCHAR")), to_date(col))
     return Column.invoke_expression_over_column(col, expression.Day)

@@ -316,6 +316,7 @@ def sqlglot_to_spark(sqlglot_dtype: exp.DataType) -> types.DataType:
         exp.DataType.Type.INT: types.IntegerType,
         exp.DataType.Type.BIGINT: types.LongType,
         exp.DataType.Type.SMALLINT: types.ShortType,
+        exp.DataType.Type.TINYINT: types.ByteType,
         exp.DataType.Type.FLOAT: types.FloatType,
         exp.DataType.Type.DOUBLE: types.DoubleType,
         exp.DataType.Type.DECIMAL: types.DecimalType,

@@ -3640,6 +3640,10 @@ def test_day(get_session_and_func, get_func):
     session, day = get_session_and_func("day")
     df = session.createDataFrame([("2015-04-08",)], ["dt"])
     assert df.select(day("dt").alias("day")).first()[0] == 8
+    df = session.createDataFrame([(datetime.datetime(2025, 4, 8),)], ["dt"])
+    assert df.select(day("dt").alias("day")).first()[0] == 8
+    df = session.createDataFrame([("2015-04-08 01:00:00",)], ["dt"])
+    assert df.select(day("dt").alias("day")).first()[0] == 8
 def test_elt(get_session_and_func, get_func):

@@ -39,6 +39,7 @@ def test_quote_preserving_alias_or_name(expression: t.Union[exp.Column, exp.Alia
         ("INTEGER", types.IntegerType()),
         ("BIGINT", types.LongType()),
         ("SMALLINT", types.ShortType()),
+        ("TINYINT", types.ByteType()),
         ("FLOAT", types.FloatType()),
         ("DOUBLE", types.DoubleType()),
         ("BOOLEAN", types.BooleanType()),

sqlframe 3.19.0__tar.gz → 3.20.0__tar.gz