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

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
sqlframe/__init__.py CHANGED
@@ -6,6 +6,10 @@ import typing as t
6
6
  from contextlib import contextmanager
7
7
  from unittest.mock import MagicMock
8
8
 
9
+ from sqlglot.dialects import spark
10
+
11
+ spark.Spark.Generator.readparquet_sql = lambda self, x: x.sql(dialect="duckdb") # type: ignore
12
+
9
13
  if t.TYPE_CHECKING:
10
14
  from sqlframe.base.session import CONN
11
15
 
sqlframe/_version.py CHANGED
@@ -28,7 +28,7 @@ version_tuple: VERSION_TUPLE
28
28
  commit_id: COMMIT_ID
29
29
  __commit_id__: COMMIT_ID
30
30
 
31
- __version__ = version = '3.43.6'
32
- __version_tuple__ = version_tuple = (3, 43, 6)
31
+ __version__ = version = '3.43.7'
32
+ __version_tuple__ = version_tuple = (3, 43, 7)
33
33
 
34
- __commit_id__ = commit_id = 'g7c6cd19ef'
34
+ __commit_id__ = commit_id = 'g745c96966'
@@ -414,7 +414,7 @@ def radians(col: ColumnOrName) -> Column:
414
414
  if session._is_bigquery:
415
415
  return radians_bgutil(col)
416
416
 
417
- return Column.invoke_anonymous_function(col, "RADIANS")
417
+ return Column.invoke_expression_over_column(col, expression.Radians)
418
418
 
419
419
 
420
420
  toRadians = radians
@@ -1078,7 +1078,7 @@ def hour(col: ColumnOrName) -> Column:
1078
1078
  if session._is_bigquery or session._is_postgres:
1079
1079
  return hour_from_extract(col)
1080
1080
 
1081
- return Column.invoke_anonymous_function(col, "HOUR")
1081
+ return Column.invoke_expression_over_column(col, expression.Hour)
1082
1082
 
1083
1083
 
1084
1084
  @meta()
@@ -1090,7 +1090,7 @@ def minute(col: ColumnOrName) -> Column:
1090
1090
  if session._is_bigquery or session._is_postgres:
1091
1091
  return minute_from_extract(col)
1092
1092
 
1093
- return Column.invoke_anonymous_function(col, "MINUTE")
1093
+ return Column.invoke_expression_over_column(col, expression.Minute)
1094
1094
 
1095
1095
 
1096
1096
  @meta()
@@ -1102,7 +1102,7 @@ def second(col: ColumnOrName) -> Column:
1102
1102
  if session._is_bigquery or session._is_postgres:
1103
1103
  return second_from_extract(col)
1104
1104
 
1105
- return Column.invoke_anonymous_function(col, "SECOND")
1105
+ return Column.invoke_expression_over_column(col, expression.Second)
1106
1106
 
1107
1107
 
1108
1108
  @meta()
@@ -1398,7 +1398,7 @@ def next_day(col: ColumnOrName, dayOfWeek: str) -> Column:
1398
1398
  if session._is_bigquery:
1399
1399
  return next_day_bgutil(col, dayOfWeek)
1400
1400
 
1401
- return Column.invoke_anonymous_function(col, "NEXT_DAY", lit(dayOfWeek))
1401
+ return Column.invoke_expression_over_column(col, expression.NextDay, expression=lit(dayOfWeek))
1402
1402
 
1403
1403
 
1404
1404
  @meta()
@@ -2160,7 +2160,7 @@ def bit_count(col: ColumnOrName) -> Column:
2160
2160
  if session._is_duckdb:
2161
2161
  return Column.invoke_anonymous_function(col, "BIT_COUNT")
2162
2162
 
2163
- return Column.invoke_expression_over_column(col, expression.BitwiseCountAgg)
2163
+ return Column.invoke_expression_over_column(col, expression.BitwiseCount)
2164
2164
 
2165
2165
 
2166
2166
  @meta(unsupported_engines="*")
@@ -4108,7 +4108,7 @@ def left(str: ColumnOrName, len: ColumnOrName) -> Column:
4108
4108
  return Column.invoke_expression_over_column(str, expression.Left, expression=len)
4109
4109
 
4110
4110
 
4111
- @meta(unsupported_engines="*")
4111
+ @meta(unsupported_engines="bigquery")
4112
4112
  def like(
4113
4113
  str: ColumnOrName, pattern: ColumnOrName, escapeChar: t.Optional["Column"] = None
4114
4114
  ) -> Column:
@@ -6805,7 +6805,7 @@ def weekday(col: ColumnOrName) -> Column:
6805
6805
  return Column.invoke_anonymous_function(col, "weekday")
6806
6806
 
6807
6807
 
6808
- @meta(unsupported_engines="*")
6808
+ @meta(unsupported_engines=["bigquery", "duckdb", "postgres"])
6809
6809
  def width_bucket(
6810
6810
  v: ColumnOrName,
6811
6811
  min: ColumnOrName,
@@ -6854,7 +6854,9 @@ def width_bucket(
6854
6854
  +----------------------------+
6855
6855
  """
6856
6856
  numBucket = lit(numBucket) if isinstance(numBucket, int) else numBucket
6857
- return Column.invoke_anonymous_function(v, "width_bucket", min, max, numBucket)
6857
+ return Column.invoke_expression_over_column(
6858
+ v, expression.WidthBucket, min_value=min, max_value=max, num_buckets=numBucket
6859
+ )
6858
6860
 
6859
6861
 
6860
6862
  @meta(unsupported_engines=["*", "spark"])
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: sqlframe
3
- Version: 3.43.6
3
+ Version: 3.43.7
4
4
  Summary: Turning PySpark Into a Universal DataFrame API
5
5
  Home-page: https://github.com/eakmanrq/sqlframe
6
6
  Author: Ryan Eakman
@@ -18,7 +18,7 @@ Description-Content-Type: text/markdown
18
18
  License-File: LICENSE
19
19
  Requires-Dist: more-itertools
20
20
  Requires-Dist: prettytable <4
21
- Requires-Dist: sqlglot <27.27,>=24.0.0
21
+ Requires-Dist: sqlglot <27.29,>=24.0.0
22
22
  Requires-Dist: typing-extensions
23
23
  Provides-Extra: bigquery
24
24
  Requires-Dist: google-cloud-bigquery-storage <3,>=2 ; extra == 'bigquery'
@@ -1,5 +1,5 @@
1
- sqlframe/__init__.py,sha256=SB80yLTITBXHI2GCDS6n6bN5ObHqgPjfpRPAUwxaots,3403
2
- sqlframe/_version.py,sha256=y6ZWAFQfKzruNZ3B9AngA6WOri7aL6KlfRfNwyPr8T8,714
1
+ sqlframe/__init__.py,sha256=Et338oqN6tgrUNzSHpaIyjTiXcXS9lze7qeLFYdRNVc,3536
2
+ sqlframe/_version.py,sha256=rxxVqq9tM144VQj_9aHUEqtWXVIJXvf2oSYA3-oytsk,714
3
3
  sqlframe/py.typed,sha256=Nqnn8clbgv-5l0PgxcTOldg8mkMKrFn4TvPL-rYUUGg,1
4
4
  sqlframe/base/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
5
5
  sqlframe/base/_typing.py,sha256=b2clI5HI1zEZKB_3Msx3FeAJQyft44ubUifJwQRVXyQ,1298
@@ -9,7 +9,7 @@ sqlframe/base/dataframe.py,sha256=iKdiJ9OnMEbL0GAwydP9hQz0XsTKs1mpKi_6ajrCZ9I,90
9
9
  sqlframe/base/decorators.py,sha256=IhE5xNQDkwJHacCvulq5WpUKyKmXm7dL2A3o5WuKGP4,2131
10
10
  sqlframe/base/exceptions.py,sha256=9Uwvqn2eAkDpqm4BrRgbL61qM-GMCbJEMAW8otxO46s,370
11
11
  sqlframe/base/function_alternatives.py,sha256=aTu3nQhIAkZoxrI1IpjpaHEAMxBNms0AnhS0EMR-TwY,51727
12
- sqlframe/base/functions.py,sha256=F725utyWrBreiSvo-a-bM-497JfXq84ywPgNNjjqMAE,229640
12
+ sqlframe/base/functions.py,sha256=FtX9TAoST0Z_OrIlS9bOJ3V--lGmDE8o8rAEiNIL-5Q,229808
13
13
  sqlframe/base/group.py,sha256=fBm8EUve7W7xz11nybTXr09ih-yZxL_vvEiZVE1eb_0,12025
14
14
  sqlframe/base/normalize.py,sha256=YPeopWr8ZRjevArYfrM-DZBkQp4t4UfAEwynoj4VvcU,11773
15
15
  sqlframe/base/operations.py,sha256=v8BA80eDKBOOpfHB_Zxsdi2RrDkSl1-_gtA-GuwO8qY,4438
@@ -130,8 +130,8 @@ sqlframe/standalone/udf.py,sha256=azmgtUjHNIPs0WMVNId05SHwiYn41MKVBhKXsQJ5dmY,27
130
130
  sqlframe/standalone/window.py,sha256=6GKPzuxeSapJakBaKBeT9VpED1ACdjggDv9JRILDyV0,35
131
131
  sqlframe/testing/__init__.py,sha256=VVCosQhitU74A3NnE52O4mNtGZONapuEXcc20QmSlnQ,132
132
132
  sqlframe/testing/utils.py,sha256=PFsGZpwNUE_4-g_f43_vstTqsK0AQ2lBneb5Eb6NkFo,13008
133
- sqlframe-3.43.6.dist-info/LICENSE,sha256=VZu79YgW780qxaFJMr0t5ZgbOYEh04xWoxaWOaqIGWk,1068
134
- sqlframe-3.43.6.dist-info/METADATA,sha256=UjA8mFuqq0yTQVw_TAU3t7XEbzDoA9dn_nhrfibhoS0,9070
135
- sqlframe-3.43.6.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
136
- sqlframe-3.43.6.dist-info/top_level.txt,sha256=T0_RpoygaZSF6heeWwIDQgaP0varUdSK1pzjeJZRjM8,9
137
- sqlframe-3.43.6.dist-info/RECORD,,
133
+ sqlframe-3.43.7.dist-info/LICENSE,sha256=VZu79YgW780qxaFJMr0t5ZgbOYEh04xWoxaWOaqIGWk,1068
134
+ sqlframe-3.43.7.dist-info/METADATA,sha256=pOTnxd_hgmoNGSbRzFF-4fWMETe2sq8VKRL3_HhMg0o,9070
135
+ sqlframe-3.43.7.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
136
+ sqlframe-3.43.7.dist-info/top_level.txt,sha256=T0_RpoygaZSF6heeWwIDQgaP0varUdSK1pzjeJZRjM8,9
137
+ sqlframe-3.43.7.dist-info/RECORD,,