sqlframe 3.43.6__py3-none-any.whl → 3.43.8__py3-none-any.whl

sqlframe/__init__.py

@@ -6,6 +6,10 @@ import typing as t
 from contextlib import contextmanager
 from unittest.mock import MagicMock
 
+from sqlglot.dialects import spark
+
+spark.Spark.Generator.readparquet_sql = lambda self, x: x.sql(dialect="duckdb")  # type: ignore
+
 if t.TYPE_CHECKING:
     from sqlframe.base.session import CONN
 

sqlframe/_version.py

@@ -28,7 +28,7 @@ version_tuple: VERSION_TUPLE
 commit_id: COMMIT_ID
 __commit_id__: COMMIT_ID
 
-__version__ = version = '3.43.6'
-__version_tuple__ = version_tuple = (3, 43, 6)
+__version__ = version = '3.43.8'
+__version_tuple__ = version_tuple = (3, 43, 8)
 
-__commit_id__ = commit_id = 'g7c6cd19ef'
+__commit_id__ = commit_id = 'g371f414f3'

sqlframe/base/functions.py

@@ -60,7 +60,11 @@ def lit(value: t.Optional[t.Any] = None) -> Column:
     if isinstance(value, str):
         return Column(expression.Literal.string(value))
     if isinstance(value, float) and value in {float("inf"), float("-inf")}:
-        return Column(expression.Literal.string(str(value)))
+        return Column(
+            expression.cast(
+                expression.Literal.string(str(value)), to=expression.DataType.Type.FLOAT
+            )
+        )
     return Column(value)
 
 
@@ -414,7 +418,7 @@ def radians(col: ColumnOrName) -> Column:
     if session._is_bigquery:
         return radians_bgutil(col)
 
-    return Column.invoke_anonymous_function(col, "RADIANS")
+    return Column.invoke_expression_over_column(col, expression.Radians)
 
 
 toRadians = radians
@@ -1078,7 +1082,7 @@ def hour(col: ColumnOrName) -> Column:
     if session._is_bigquery or session._is_postgres:
         return hour_from_extract(col)
 
-    return Column.invoke_anonymous_function(col, "HOUR")
+    return Column.invoke_expression_over_column(col, expression.Hour)
 
 
 @meta()
@@ -1090,7 +1094,7 @@ def minute(col: ColumnOrName) -> Column:
     if session._is_bigquery or session._is_postgres:
         return minute_from_extract(col)
 
-    return Column.invoke_anonymous_function(col, "MINUTE")
+    return Column.invoke_expression_over_column(col, expression.Minute)
 
 
 @meta()
@@ -1102,7 +1106,7 @@ def second(col: ColumnOrName) -> Column:
     if session._is_bigquery or session._is_postgres:
         return second_from_extract(col)
 
-    return Column.invoke_anonymous_function(col, "SECOND")
+    return Column.invoke_expression_over_column(col, expression.Second)
 
 
 @meta()
@@ -1398,7 +1402,7 @@ def next_day(col: ColumnOrName, dayOfWeek: str) -> Column:
     if session._is_bigquery:
         return next_day_bgutil(col, dayOfWeek)
 
-    return Column.invoke_anonymous_function(col, "NEXT_DAY", lit(dayOfWeek))
+    return Column.invoke_expression_over_column(col, expression.NextDay, expression=lit(dayOfWeek))
 
 
 @meta()
@@ -2160,7 +2164,7 @@ def bit_count(col: ColumnOrName) -> Column:
     if session._is_duckdb:
         return Column.invoke_anonymous_function(col, "BIT_COUNT")
 
-    return Column.invoke_expression_over_column(col, expression.BitwiseCountAgg)
+    return Column.invoke_expression_over_column(col, expression.BitwiseCount)
 
 
 @meta(unsupported_engines="*")
@@ -3308,9 +3312,9 @@ def endswith(str: ColumnOrName, suffix: ColumnOrName) -> Column:
     )
 
 
-@meta(unsupported_engines="*")
+@meta(unsupported_engines=["bigquery", "duckdb", "postgres"])
 def equal_null(col1: ColumnOrName, col2: ColumnOrName) -> Column:
-    return Column.invoke_anonymous_function(col1, "equal_null", col2)
+    return Column.invoke_expression_over_column(col1, expression.EqualNull, expression=col2)
 
 
 @meta(unsupported_engines="*")
@@ -4108,7 +4112,7 @@ def left(str: ColumnOrName, len: ColumnOrName) -> Column:
     return Column.invoke_expression_over_column(str, expression.Left, expression=len)
 
 
-@meta(unsupported_engines="*")
+@meta(unsupported_engines="bigquery")
 def like(
     str: ColumnOrName, pattern: ColumnOrName, escapeChar: t.Optional["Column"] = None
 ) -> Column:
@@ -6805,7 +6809,7 @@ def weekday(col: ColumnOrName) -> Column:
     return Column.invoke_anonymous_function(col, "weekday")
 
 
-@meta(unsupported_engines="*")
+@meta(unsupported_engines=["bigquery", "duckdb", "postgres"])
 def width_bucket(
     v: ColumnOrName,
     min: ColumnOrName,
@@ -6854,7 +6858,9 @@ def width_bucket(
     +----------------------------+
     """
     numBucket = lit(numBucket) if isinstance(numBucket, int) else numBucket
-    return Column.invoke_anonymous_function(v, "width_bucket", min, max, numBucket)
+    return Column.invoke_expression_over_column(
+        v, expression.WidthBucket, min_value=min, max_value=max, num_buckets=numBucket
+    )
 
 
 @meta(unsupported_engines=["*", "spark"])

{sqlframe-3.43.6.dist-info → sqlframe-3.43.8.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: sqlframe
-Version: 3.43.6
+Version: 3.43.8
 Summary: Turning PySpark Into a Universal DataFrame API
 Home-page: https://github.com/eakmanrq/sqlframe
 Author: Ryan Eakman
@@ -18,7 +18,7 @@ Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: more-itertools
 Requires-Dist: prettytable <4
-Requires-Dist: sqlglot <27.27,>=24.0.0
+Requires-Dist: sqlglot <27.30,>=24.0.0
 Requires-Dist: typing-extensions
 Provides-Extra: bigquery
 Requires-Dist: google-cloud-bigquery-storage <3,>=2 ; extra == 'bigquery'
@@ -28,13 +28,14 @@ Requires-Dist: databricks-sql-connector[pyarrow] <5,>=3.6 ; extra == 'databricks
 Provides-Extra: dev
 Requires-Dist: duckdb <1.5,>=1.2 ; extra == 'dev'
 Requires-Dist: findspark <3,>=2 ; extra == 'dev'
+Requires-Dist: mirakuru <3.0.0 ; extra == 'dev'
 Requires-Dist: mypy <1.19,>=1.10.0 ; extra == 'dev'
 Requires-Dist: openai <3,>=1.30 ; extra == 'dev'
 Requires-Dist: pandas-stubs <3,>=2 ; extra == 'dev'
 Requires-Dist: pandas <3,>=2 ; extra == 'dev'
 Requires-Dist: pre-commit <5,>=3.7 ; extra == 'dev'
 Requires-Dist: psycopg <4,>=3.1 ; extra == 'dev'
-Requires-Dist: pyarrow <22,>=10 ; extra == 'dev'
+Requires-Dist: pyarrow <23,>=10 ; extra == 'dev'
 Requires-Dist: pyspark <3.6,>=2 ; extra == 'dev'
 Requires-Dist: pytest-forked ; extra == 'dev'
 Requires-Dist: pytest-postgresql <8,>=6 ; extra == 'dev'

{sqlframe-3.43.6.dist-info → sqlframe-3.43.8.dist-info}/RECORD

@@ -1,5 +1,5 @@
-sqlframe/__init__.py,sha256=SB80yLTITBXHI2GCDS6n6bN5ObHqgPjfpRPAUwxaots,3403
-sqlframe/_version.py,sha256=y6ZWAFQfKzruNZ3B9AngA6WOri7aL6KlfRfNwyPr8T8,714
+sqlframe/__init__.py,sha256=Et338oqN6tgrUNzSHpaIyjTiXcXS9lze7qeLFYdRNVc,3536
+sqlframe/_version.py,sha256=h6y12CzI4Xz_j0jGD8riwX7mlvg7zIGwFg0P-8kKcC4,714
 sqlframe/py.typed,sha256=Nqnn8clbgv-5l0PgxcTOldg8mkMKrFn4TvPL-rYUUGg,1
 sqlframe/base/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 sqlframe/base/_typing.py,sha256=b2clI5HI1zEZKB_3Msx3FeAJQyft44ubUifJwQRVXyQ,1298
@@ -9,7 +9,7 @@ sqlframe/base/dataframe.py,sha256=iKdiJ9OnMEbL0GAwydP9hQz0XsTKs1mpKi_6ajrCZ9I,90
 sqlframe/base/decorators.py,sha256=IhE5xNQDkwJHacCvulq5WpUKyKmXm7dL2A3o5WuKGP4,2131
 sqlframe/base/exceptions.py,sha256=9Uwvqn2eAkDpqm4BrRgbL61qM-GMCbJEMAW8otxO46s,370
 sqlframe/base/function_alternatives.py,sha256=aTu3nQhIAkZoxrI1IpjpaHEAMxBNms0AnhS0EMR-TwY,51727
-sqlframe/base/functions.py,sha256=F725utyWrBreiSvo-a-bM-497JfXq84ywPgNNjjqMAE,229640
+sqlframe/base/functions.py,sha256=bp2Mt1ajgv0JOEYfy9AVNcGO9NVLrVnrRDTjYWTgJrM,229966
 sqlframe/base/group.py,sha256=fBm8EUve7W7xz11nybTXr09ih-yZxL_vvEiZVE1eb_0,12025
 sqlframe/base/normalize.py,sha256=YPeopWr8ZRjevArYfrM-DZBkQp4t4UfAEwynoj4VvcU,11773
 sqlframe/base/operations.py,sha256=v8BA80eDKBOOpfHB_Zxsdi2RrDkSl1-_gtA-GuwO8qY,4438
@@ -130,8 +130,8 @@ sqlframe/standalone/udf.py,sha256=azmgtUjHNIPs0WMVNId05SHwiYn41MKVBhKXsQJ5dmY,27
 sqlframe/standalone/window.py,sha256=6GKPzuxeSapJakBaKBeT9VpED1ACdjggDv9JRILDyV0,35
 sqlframe/testing/__init__.py,sha256=VVCosQhitU74A3NnE52O4mNtGZONapuEXcc20QmSlnQ,132
 sqlframe/testing/utils.py,sha256=PFsGZpwNUE_4-g_f43_vstTqsK0AQ2lBneb5Eb6NkFo,13008
-sqlframe-3.43.6.dist-info/LICENSE,sha256=VZu79YgW780qxaFJMr0t5ZgbOYEh04xWoxaWOaqIGWk,1068
-sqlframe-3.43.6.dist-info/METADATA,sha256=UjA8mFuqq0yTQVw_TAU3t7XEbzDoA9dn_nhrfibhoS0,9070
-sqlframe-3.43.6.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
-sqlframe-3.43.6.dist-info/top_level.txt,sha256=T0_RpoygaZSF6heeWwIDQgaP0varUdSK1pzjeJZRjM8,9
-sqlframe-3.43.6.dist-info/RECORD,,
+sqlframe-3.43.8.dist-info/LICENSE,sha256=VZu79YgW780qxaFJMr0t5ZgbOYEh04xWoxaWOaqIGWk,1068
+sqlframe-3.43.8.dist-info/METADATA,sha256=WsHDXv-x51IagBt9ALqEfYrTpg5Q47znIc3lM0DIUTk,9118
+sqlframe-3.43.8.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
+sqlframe-3.43.8.dist-info/top_level.txt,sha256=T0_RpoygaZSF6heeWwIDQgaP0varUdSK1pzjeJZRjM8,9
+sqlframe-3.43.8.dist-info/RECORD,,