sqlframe 3.14.2__tar.gz → 3.15.1__tar.gz

{sqlframe-3.14.2 → sqlframe-3.15.1}/PKG-INFO

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

{sqlframe-3.14.2 → sqlframe-3.15.1}/docs/bigquery.md

@@ -339,6 +339,8 @@ See something that you would like to see supported? [Open an issue](https://gith
 * [ceil](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.ceil.html)
 * [ceiling](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.ceiling.html)
 * [char](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.char.html)
+* [char_length](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.char_length.html)
+* [character_length](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.character_length.html)
 * [coalesce](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.coalesce.html)
 * [col](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.col.html)
 * [collect_list](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.collect_list.html)
@@ -449,11 +451,13 @@ See something that you would like to see supported? [Open an issue](https://gith
 * [radians](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.radians.html)
 * [rand](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.rand.html)
 * [rank](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.rank.html)
+* [regexp](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.regexp.html)
 * [regexp_extract](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.regexp_extract.html)
     * Single capture group is supported
 * [regexp_like](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.regexp_like.html)
 * [regexp_replace](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.regexp_replace.html)
 * [repeat](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.repeat.html)
+* [replace](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.replace.html)
 * [reverse](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.reverse.html)
     * Only works on strings (does not work on arrays)
 * [right](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.right.html)
@@ -509,6 +513,9 @@ See something that you would like to see supported? [Open an issue](https://gith
 * [ucase](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.ucase.html)
 * [unbase64](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.unbase64.html)
 * [unhex](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.unhex.html)
+* [unix_micros](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.unix_micros.html)
+* [unix_millis](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.unix_millis.html)
+* [unix_seconds](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.unix_seconds.html)
 * [unix_timestamp](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.unix_timestamp.html)
 * [upper](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.upper.html)
 * [var_pop](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.var_pop.html)

{sqlframe-3.14.2 → sqlframe-3.15.1}/docs/duckdb.md

@@ -301,6 +301,8 @@ See something that you would like to see supported? [Open an issue](https://gith
 * [ceil](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.ceil.html)
 * [ceiling](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.ceiling.html)
 * [char](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.char.html)
+* [char_length](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.char_length.html)
+* [character_length](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.character_length.html)
 * [coalesce](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.coalesce.html)
 * [col](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.col.html)
 * [collect_list](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.collect_list.html)
@@ -417,10 +419,12 @@ See something that you would like to see supported? [Open an issue](https://gith
 * [radians](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.radians.html)
 * [rand](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.rand.html)
 * [rank](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.rank.html)
+* [regexp](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.regexp.html)
 * [regexp_extract](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.regexp_extract.html)
 * [regexp_like](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.regexp_like.html)
 * [regexp_replace](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.regexp_replace.html)
 * [repeat](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.repeat.html)
+* [replace](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.replace.html)
 * [reverse](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.reverse.html)
     * Only works on strings (does not work on arrays)
 * [right](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.right.html)
@@ -473,6 +477,10 @@ See something that you would like to see supported? [Open an issue](https://gith
 * [ucase](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.ucase.html)
 * [unbase64](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.unbase64.html)
 * [unhex](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.unhex.html)
+* [unix_micros](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.unix_micros.html)
+* [unix_millis](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.unix_millis.html)
+* [unix_seconds](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.unix_seconds.html)
+* [unix_seconds](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.unix_seconds.html)
 * [unix_timestamp](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.unix_timestamp.html)
 * [upper](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.upper.html)
 * [var_pop](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.var_pop.html)

{sqlframe-3.14.2 → sqlframe-3.15.1}/docs/postgres.md

@@ -311,6 +311,8 @@ See something that you would like to see supported? [Open an issue](https://gith
 * [ceil](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.ceil.html)
 * [ceiling](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.ceiling.html)
 * [char](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.char.html)
+* [char_length](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.char_length.html)
+* [character_length](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.character_length.html)
 * [coalesce](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.coalesce.html)
 * [col](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.col.html)
 * [collect_list](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.collect_list.html)
@@ -412,9 +414,11 @@ See something that you would like to see supported? [Open an issue](https://gith
 * [radians](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.radians.html)
 * [rand](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.rand.html)
 * [rank](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.rank.html)
+* [regexp](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.regexp.html)
 * [regexp_like](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.regexp_like.html)
 * [regexp_replace](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.regexp_replace.html)
 * [repeat](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.repeat.html)
+* [replace](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.replace.html)
 * [reverse](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.reverse.html)
     * Only works on strings (does not work on arrays)
 * [right](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.right.html)
@@ -462,6 +466,9 @@ See something that you would like to see supported? [Open an issue](https://gith
 * [typeof](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.typeof.html)
 * [ucase](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.ucase.html)
 * [unbase64](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.unbase64.html)
+* [unix_micros](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.unix_micros.html)
+* [unix_millis](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.unix_millis.html)
+* [unix_seconds](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.unix_seconds.html)
 * [unix_timestamp](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.unix_timestamp.html)
 * [upper](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.upper.html)
 * [var_pop](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.var_pop.html)

{sqlframe-3.14.2 → sqlframe-3.15.1}/docs/snowflake.md

@@ -337,6 +337,8 @@ See something that you would like to see supported? [Open an issue](https://gith
 * [ceil](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.ceil.html)
 * [ceiling](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.ceiling.html)
 * [char](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.char.html)
+* [char_length](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.char_length.html)
+* [character_length](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.character_length.html)
 * [coalesce](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.coalesce.html)
 * [col](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.col.html)
 * [collect_list](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.collect_list.html)
@@ -460,6 +462,7 @@ See something that you would like to see supported? [Open an issue](https://gith
 * [regexp_extract](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.regexp_extract.html)
 * [regexp_replace](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.regexp_replace.html)
 * [repeat](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.repeat.html)
+* [replace](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.replace.html)
 * [right](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.right.html)
 * [rint](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.rint.html)
 * [round](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.round.html)
@@ -513,6 +516,8 @@ See something that you would like to see supported? [Open an issue](https://gith
 * [ucase](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.ucase.html)
 * [unbase64](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.unbase64.html)
 * [unhex](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.unhex.html)
+* [unix_micros](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.unix_micros.html)
+* [unix_millis](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.unix_millis.html)
 * [unix_seconds](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.unix_seconds.html)
 * [unix_timestamp](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.unix_timestamp.html)
 * [upper](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.upper.html)

{sqlframe-3.14.2 → sqlframe-3.15.1}/sqlframe/_version.py

@@ -12,5 +12,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '3.14.2'
-__version_tuple__ = version_tuple = (3, 14, 2)
+__version__ = version = '3.15.1'
+__version_tuple__ = version_tuple = (3, 15, 1)

{sqlframe-3.14.2 → sqlframe-3.15.1}/sqlframe/base/function_alternatives.py

@@ -6,9 +6,7 @@ import re
 import typing as t
 
 from sqlglot import exp as expression
-from sqlglot.dialects.dialect import build_formatted_time
 from sqlglot.helper import ensure_list
-from sqlglot.helper import flatten as _flatten
 
 from sqlframe.base.column import Column
 from sqlframe.base.util import (
@@ -1410,6 +1408,14 @@ def regexp_replace_global_option(
     )
 
 
+def regexp_with_matches(str: ColumnOrName, regexp: ColumnOrName) -> Column:
+    return Column.invoke_anonymous_function(str, "REGEXP_MATCHES", regexp)
+
+
+def regexp_with_contains(str: ColumnOrName, regexp: ColumnOrName) -> Column:
+    return Column.invoke_anonymous_function(str, "REGEXP_CONTAINS", regexp)
+
+
 def degrees_bgutil(col: ColumnOrName) -> Column:
     return Column(
         expression.Anonymous(
@@ -1519,6 +1525,43 @@ def unix_timestamp_bgutil(
     )
 
 
+def unix_seconds_extract_epoch(col: ColumnOrName) -> Column:
+    return Column(
+        expression.Extract(
+            this=expression.Var(this="EPOCH"),
+            expression=Column.ensure_col(col).column_expression,
+        )
+    )
+
+
+def unix_millis_multiply_epoch(col: ColumnOrName) -> Column:
+    unix_seconds = get_func_from_session("unix_seconds")
+
+    return Column(
+        expression.Cast(
+            this=expression.Mul(
+                this=unix_seconds(col).column_expression,
+                expression=expression.Literal.number(1000),
+            ),
+            to=expression.DataType.build("bigint"),
+        )
+    )
+
+
+def unix_micros_multiply_epoch(col: ColumnOrName) -> Column:
+    unix_seconds = get_func_from_session("unix_seconds")
+
+    return Column(
+        expression.Cast(
+            this=expression.Mul(
+                this=unix_seconds(col).column_expression,
+                expression=expression.Literal.number(1000000),
+            ),
+            to=expression.DataType.build("bigint"),
+        )
+    )
+
+
 def format_number_bgutil(col: ColumnOrName, d: int) -> Column:
     round = get_func_from_session("round")
     lit = get_func_from_session("lit")

{sqlframe-3.14.2 → sqlframe-3.15.1}/sqlframe/base/functions.py

@@ -51,6 +51,8 @@ def col(column_name: t.Union[ColumnOrName, t.Any]) -> Column:
 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(value)
 
 
@@ -961,12 +963,15 @@ def dayofweek(col: ColumnOrName) -> Column:
         return dayofweek_from_extract(col)
 
     if session._is_postgres:
-        return dayofweek_from_extract_with_isodow(col)
+        return dayofweek_from_extract_with_isodow(col) + 1
 
-    return Column.invoke_expression_over_column(
+    result = Column.invoke_expression_over_column(
         Column(expression.TsOrDsToDate(this=Column.ensure_col(col).column_expression)),
         expression.DayOfWeek,
     )
+    if session._is_duckdb or session._is_snowflake:
+        return result + 1
+    return result
 
 
 @meta()
@@ -2962,14 +2967,14 @@ def char(col: ColumnOrName) -> Column:
     return Column(expression.Chr(expressions=Column.ensure_col(col).column_expression))
 
 
-@meta(unsupported_engines="*")
+@meta()
 def char_length(str: ColumnOrName) -> Column:
-    return Column.invoke_anonymous_function(str, "char_length")
+    return Column.invoke_expression_over_column(str, expression.Length)
 
 
-@meta(unsupported_engines="*")
+@meta()
 def character_length(str: ColumnOrName) -> Column:
-    return Column.invoke_anonymous_function(str, "character_length")
+    return Column.invoke_expression_over_column(str, expression.Length)
 
 
 @meta(unsupported_engines=["bigquery", "postgres"])
@@ -4946,7 +4951,7 @@ def reflect(*cols: ColumnOrName) -> Column:
     return Column.invoke_anonymous_function(cols[0], "reflect")
 
 
-@meta(unsupported_engines="*")
+@meta(unsupported_engines="snowflake")
 def regexp(str: ColumnOrName, regexp: ColumnOrName) -> Column:
     r"""Returns true if `str` matches the Java regex `regexp`, or false otherwise.
 
@@ -4996,12 +5001,21 @@ def regexp(str: ColumnOrName, regexp: ColumnOrName) -> Column:
     |               true|
     +-------------------+
     """
-    from sqlframe.base.function_alternatives import regexp_extract_only_one_group
+    from sqlframe.base.function_alternatives import (
+        regexp_with_contains,
+        regexp_with_matches,
+    )
 
     session = _get_session()
 
+    if session._is_duckdb:
+        return regexp_with_matches(str, regexp)
+
+    if session._is_postgres:
+        return Column.invoke_expression_over_column(str, expression.RegexpILike, expression=regexp)
+
     if session._is_bigquery:
-        return regexp_extract_only_one_group(str, regexp)  # type: ignore
+        return regexp_with_contains(str, regexp)
 
     return Column.invoke_anonymous_function(str, "regexp", regexp)
 
@@ -5485,7 +5499,7 @@ def regr_syy(y: ColumnOrName, x: ColumnOrName) -> Column:
     return Column.invoke_anonymous_function(y, "regr_syy", x)
 
 
-@meta(unsupported_engines="*")
+@meta()
 def replace(
     src: ColumnOrName, search: ColumnOrName, replace: t.Optional[ColumnOrName] = None
 ) -> Column:
@@ -5513,6 +5527,11 @@ def replace(
     >>> df.select(replace(df.a, df.b).alias('r')).collect()
     [Row(r='ABC')]
     """
+    if replace is None and (
+        _get_session()._is_duckdb or _get_session()._is_postgres or _get_session()._is_bigquery
+    ):
+        replace = expression.Literal.string("")  # type: ignore
+
     if replace is not None:
         return Column.invoke_anonymous_function(src, "replace", search, replace)
     else:
@@ -6397,7 +6416,7 @@ def unix_date(col: ColumnOrName) -> Column:
     return Column.invoke_expression_over_column(col, expression.UnixDate)
 
 
-@meta(unsupported_engines="*")
+@meta()
 def unix_micros(col: ColumnOrName) -> Column:
     """Returns the number of microseconds since 1970-01-01 00:00:00 UTC.
 
@@ -6411,10 +6430,20 @@ def unix_micros(col: ColumnOrName) -> Column:
     [Row(n=1437584400000000)]
     >>> spark.conf.unset("spark.sql.session.timeZone")
     """
+    from sqlframe.base.function_alternatives import unix_micros_multiply_epoch
+
+    if (
+        _get_session()._is_bigquery
+        or _get_session()._is_duckdb
+        or _get_session()._is_postgres
+        or _get_session()._is_snowflake
+    ):
+        return unix_micros_multiply_epoch(col)
+
     return Column.invoke_anonymous_function(col, "unix_micros")
 
 
-@meta(unsupported_engines="*")
+@meta()
 def unix_millis(col: ColumnOrName) -> Column:
     """Returns the number of milliseconds since 1970-01-01 00:00:00 UTC.
     Truncates higher levels of precision.
@@ -6429,10 +6458,20 @@ def unix_millis(col: ColumnOrName) -> Column:
     [Row(n=1437584400000)]
     >>> spark.conf.unset("spark.sql.session.timeZone")
     """
+    from sqlframe.base.function_alternatives import unix_millis_multiply_epoch
+
+    if (
+        _get_session()._is_bigquery
+        or _get_session()._is_duckdb
+        or _get_session()._is_postgres
+        or _get_session()._is_snowflake
+    ):
+        return unix_millis_multiply_epoch(col)
+
     return Column.invoke_anonymous_function(col, "unix_millis")
 
 
-@meta(unsupported_engines=["bigquery", "duckdb", "postgres"])
+@meta()
 def unix_seconds(col: ColumnOrName) -> Column:
     """Returns the number of seconds since 1970-01-01 00:00:00 UTC.
     Truncates higher levels of precision.
@@ -6447,6 +6486,27 @@ def unix_seconds(col: ColumnOrName) -> Column:
     [Row(n=1437584400)]
     >>> spark.conf.unset("spark.sql.session.timeZone")
     """
+    from sqlframe.base.function_alternatives import unix_seconds_extract_epoch
+
+    if _get_session()._is_postgres:
+        return unix_seconds_extract_epoch(col)
+
+    if _get_session()._is_bigquery:
+        return Column(
+            expression.Anonymous(
+                this="UNIX_SECONDS",
+                expressions=[
+                    expression.Anonymous(
+                        this="TIMESTAMP",
+                        expressions=[
+                            Column.ensure_col(col).column_expression,
+                            expression.Literal.string("UTC"),
+                        ],
+                    )
+                ],
+            )
+        )
+
     return Column.invoke_expression_over_column(col, expression.UnixSeconds)
 
 

{sqlframe-3.14.2 → sqlframe-3.15.1}/sqlframe/base/session.py

@@ -108,6 +108,11 @@ class _BaseSession(t.Generic[CATALOG, READER, WRITER, DF, TABLE, CONN, UDF_REGIS
         if not getattr(self, "schema", None) or schema:
             self._schema = schema
 
+    # https://github.com/eakmanrq/sqlframe/issues/262
+    @property
+    def execution_dialect_name(self) -> str:
+        return self.execution_dialect.__class__.__name__.lower()
+
     @property
     def read(self) -> READER:
         return self._reader(self)

{sqlframe-3.14.2 → sqlframe-3.15.1}/sqlframe.egg-info/PKG-INFO

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

{sqlframe-3.14.2 → sqlframe-3.15.1}/tests/integration/engines/bigquery/test_bigquery_session.py

@@ -18,3 +18,4 @@ def test_session_from_config():
     session = BigQuerySession.builder.config("default_dataset", "sqlframe.db1").getOrCreate()
     columns = session.catalog.get_columns("db1.test_table")
     assert columns == {"cola": exp.DataType.build("BIGINT"), "colb": exp.DataType.build("TEXT")}
+    assert session.execution_dialect_name == "bigquery"

{sqlframe-3.14.2 → sqlframe-3.15.1}/tests/integration/engines/databricks/test_databricks_session.py

@@ -45,3 +45,4 @@ def test_session_from_config(cleanup_connector: DatabricksConnection):
         "cola": exp.DataType.build("INT", dialect=session.output_dialect),
         "colb": exp.DataType.build("STRING", dialect=session.output_dialect),
     }
+    assert session.execution_dialect_name == "databricks"

{sqlframe-3.14.2 → sqlframe-3.15.1}/tests/integration/engines/duck/test_duckdb_session.py

@@ -11,3 +11,4 @@ def test_session_from_config():
     session = DuckDBSession.builder.config("sqlframe.conn", conn).getOrCreate()
     columns = session.catalog.get_columns("test_table")
     assert columns == {"cola": exp.DataType.build("INT"), "colb": exp.DataType.build("TEXT")}
+    assert session.execution_dialect_name == "duckdb"

{sqlframe-3.14.2 → sqlframe-3.15.1}/tests/integration/engines/postgres/test_postgres_session.py

@@ -17,3 +17,4 @@ def test_session_from_config(function_scoped_postgres):
         "cola": exp.DataType.build("INT", dialect=session.output_dialect),
         "colb": exp.DataType.build("STRING", dialect=session.output_dialect),
     }
+    assert session.execution_dialect_name == "postgres"

{sqlframe-3.14.2 → sqlframe-3.15.1}/tests/integration/engines/redshift/test_redshift_session.py

@@ -45,3 +45,4 @@ def test_session_from_config(cleanup_connector: RedshiftConnection):
         "cola": exp.DataType.build("INT", dialect=session.output_dialect),
         "colb": exp.DataType.build("STRING", dialect=session.output_dialect),
     }
+    assert session.execution_dialect_name == "redshift"

{sqlframe-3.14.2 → sqlframe-3.15.1}/tests/integration/engines/snowflake/test_snowflake_session.py

@@ -45,3 +45,4 @@ def test_session_from_config(cleanup_connector: SnowflakeConnection):
         "cola": exp.DataType.build("DECIMAL(38, 0)", dialect=session.output_dialect),
         "colb": exp.DataType.build("TEXT", dialect=session.output_dialect),
     }
+    assert session.execution_dialect_name == "snowflake"

{sqlframe-3.14.2 → sqlframe-3.15.1}/tests/integration/engines/test_int_functions.py

@@ -1279,7 +1279,7 @@ def test_month(get_session_and_func):
 def test_dayofweek(get_session_and_func):
     session, dayofweek = get_session_and_func("dayofweek")
     df = session.createDataFrame([("2015-04-08",)], ["dt"])
-    assert df.select(dayofweek("dt").alias("day")).first()[0] in (3, 4)
+    assert df.select(dayofweek("dt").alias("day")).first()[0] == 4
 
 
 def test_dayofmonth(get_session_and_func):
@@ -5114,3 +5114,15 @@ def test_is_array(get_session_and_func, get_func):
         .collect()
     )
     assert result == [Row(v1=True, v2=True, v3=False)]
+
+
+# https://github.com/eakmanrq/sqlframe/issues/265
+def test_infinite(get_session_and_func):
+    session, lit = get_session_and_func("lit")
+    df = session.createDataFrame(
+        [
+            {"a": float("inf")},
+            {"a": float("-inf")},
+        ]
+    )
+    assert df.collect() == [Row(a=float("inf")), Row(a=float("-inf"))]

{sqlframe-3.14.2 → sqlframe-3.15.1}/tests/unit/standalone/test_functions.py

@@ -42,6 +42,8 @@ def test_invoke_anonymous(name, func):
         ),
         (SF.lit({"cola": 1, "colb": "test"}), "MAP('cola', 1, 'colb', 'test')"),
         (SF.lit(Row(cola=1, colb="test")), "STRUCT(1 AS cola, 'test' AS colb)"),
+        (SF.lit(float("inf")), "'inf'"),
+        (SF.lit(float("-inf")), "'-inf'"),
     ],
 )
 def test_lit(expression, expected):
@@ -3285,8 +3287,8 @@ def test_char(expression, expected):
 @pytest.mark.parametrize(
     "expression, expected",
     [
-        (SF.char_length("cola"), "CHAR_LENGTH(cola)"),
-        (SF.char_length(SF.col("cola")), "CHAR_LENGTH(cola)"),
+        (SF.char_length("cola"), "LENGTH(cola)"),
+        (SF.char_length(SF.col("cola")), "LENGTH(cola)"),
     ],
 )
 def test_char_length(expression, expected):
@@ -3296,8 +3298,8 @@ def test_char_length(expression, expected):
 @pytest.mark.parametrize(
     "expression, expected",
     [
-        (SF.character_length("cola"), "CHARACTER_LENGTH(cola)"),
-        (SF.character_length(SF.col("cola")), "CHARACTER_LENGTH(cola)"),
+        (SF.character_length("cola"), "LENGTH(cola)"),
+        (SF.character_length(SF.col("cola")), "LENGTH(cola)"),
     ],
 )
 def test_character_length(expression, expected):