sqlframe 2.0.0__tar.gz → 2.1.0__tar.gz

{sqlframe-2.0.0 → sqlframe-2.1.0}/PKG-INFO

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

{sqlframe-2.0.0 → sqlframe-2.1.0}/docs/duckdb.md

@@ -258,6 +258,7 @@ See something that you would like to see supported? [Open an issue](https://gith
 * [concat](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.concat.html)                               
     * Only works on strings (does not work on arrays)
 * [concat_ws](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.concat_ws.html)
+* [convert_timezone](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.convert_timezone.html)
 * [corr](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.corr.html)
 * [cos](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.cos.html)
 * [cot](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.cot.html)

{sqlframe-2.0.0 → sqlframe-2.1.0}/docs/snowflake.md

@@ -286,6 +286,7 @@ See something that you would like to see supported? [Open an issue](https://gith
 * [concat](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.concat.html)
   * Can only concat strings not arrays
 * [concat_ws](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.concat_ws.html)
+* [convert_timezone](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.convert_timezone.html)
 * [corr](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.corr.html)
 * [cos](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.cos.html)
 * [cosh](https://spark.apache.org/docs/latest/api/python/reference/pyspark.sql/api/pyspark.sql.functions.cosh.html)

{sqlframe-2.0.0 → sqlframe-2.1.0}/setup.py

@@ -20,7 +20,7 @@ setup(
     python_requires=">=3.8",
     install_requires=[
         "prettytable<3.11.0",
-        "sqlglot>=24.0.0,<25.9",
+        "sqlglot>=24.0.0,<25.11",
         "typing_extensions>=4.8,<5",
     ],
     extras_require={
@@ -31,7 +31,7 @@ setup(
         "dev": [
             "duckdb>=0.9,<1.1",
             "mypy>=1.10.0,<1.12",
-            "openai>=1.30,<1.38",
+            "openai>=1.30,<1.41",
             "pandas>=2,<3",
             "pandas-stubs>=2,<3",
             "psycopg>=3.1,<4",
@@ -57,7 +57,7 @@ setup(
             "pandas>=2,<3",
         ],
         "openai": [
-            "openai>=1.30,<1.38",
+            "openai>=1.30,<1.41",
         ],
         "pandas": [
             "pandas>=2,<3",

{sqlframe-2.0.0 → sqlframe-2.1.0}/sqlframe/_version.py

@@ -12,5 +12,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '2.0.0'
-__version_tuple__ = version_tuple = (2, 0, 0)
+__version__ = version = '2.1.0'
+__version_tuple__ = version_tuple = (2, 1, 0)

{sqlframe-2.0.0 → sqlframe-2.1.0}/sqlframe/base/_typing.py

@@ -24,6 +24,7 @@ OutputExpressionContainer = t.Union[exp.Select, exp.Create, exp.Insert]
 StorageLevel = str
 PathOrPaths = t.Union[str, t.List[str]]
 OptionalPrimitiveType = t.Optional[PrimitiveType]
+DataTypeOrString = t.Union[DataType, str]
 
 
 class UserDefinedFunctionLike(t.Protocol):

{sqlframe-2.0.0 → sqlframe-2.1.0}/sqlframe/base/functions.py

@@ -2070,14 +2070,19 @@ def contains(left: ColumnOrName, right: ColumnOrName) -> Column:
     return Column.invoke_anonymous_function(left, "contains", right)
 
 
-@meta(unsupported_engines="*")
+@meta(unsupported_engines=["bigquery", "postgres"])
 def convert_timezone(
     sourceTz: t.Optional[Column], targetTz: Column, sourceTs: ColumnOrName
 ) -> Column:
-    if sourceTz is None:
-        return Column.invoke_anonymous_function(targetTz, "convert_timezone", sourceTs)
-    else:
-        return Column.invoke_anonymous_function(sourceTz, "convert_timezone", targetTz, sourceTs)
+    to_timestamp = get_func_from_session("to_timestamp")
+
+    return Column(
+        expression.ConvertTimezone(
+            timestamp=to_timestamp(Column.ensure_col(sourceTs)).expression,
+            source_tz=sourceTz.expression if sourceTz else None,
+            target_tz=Column.ensure_col(targetTz).expression,
+        )
+    )
 
 
 @meta(unsupported_engines="postgres")

{sqlframe-2.0.0 → sqlframe-2.1.0}/sqlframe/base/session.py

@@ -27,6 +27,7 @@ from sqlframe.base.catalog import _BaseCatalog
 from sqlframe.base.dataframe import _BaseDataFrame
 from sqlframe.base.normalize import normalize_dict
 from sqlframe.base.readerwriter import _BaseDataFrameReader, _BaseDataFrameWriter
+from sqlframe.base.udf import _BaseUDFRegistration
 from sqlframe.base.util import (
     get_column_mapping_from_schema_input,
     normalize_string,
@@ -64,16 +65,18 @@ CATALOG = t.TypeVar("CATALOG", bound=_BaseCatalog)
 READER = t.TypeVar("READER", bound=_BaseDataFrameReader)
 WRITER = t.TypeVar("WRITER", bound=_BaseDataFrameWriter)
 DF = t.TypeVar("DF", bound=_BaseDataFrame)
+UDF_REGISTRATION = t.TypeVar("UDF_REGISTRATION", bound=_BaseUDFRegistration)
 
 _MISSING = "MISSING"
 
 
-class _BaseSession(t.Generic[CATALOG, READER, WRITER, DF, CONN]):
+class _BaseSession(t.Generic[CATALOG, READER, WRITER, DF, CONN, UDF_REGISTRATION]):
     _instance = None
     _reader: t.Type[READER]
     _writer: t.Type[WRITER]
     _catalog: t.Type[CATALOG]
     _df: t.Type[DF]
+    _udf_registration: t.Type[UDF_REGISTRATION]
 
     SANITIZE_COLUMN_NAMES = False
 
@@ -81,7 +84,6 @@ class _BaseSession(t.Generic[CATALOG, READER, WRITER, DF, CONN]):
         self,
         conn: t.Optional[CONN] = None,
         schema: t.Optional[MappingSchema] = None,
-        case_sensitive: bool = False,
         *args,
         **kwargs,
     ):
@@ -91,11 +93,6 @@ class _BaseSession(t.Generic[CATALOG, READER, WRITER, DF, CONN]):
             self.execution_dialect: Dialect = Dialect.get_or_raise(
                 self.builder.DEFAULT_EXECUTION_DIALECT
             )
-            self.case_sensitive: bool = case_sensitive
-            if self.case_sensitive:
-                self.input_dialect.NORMALIZATION_STRATEGY = NormalizationStrategy.CASE_SENSITIVE
-                self.output_dialect.NORMALIZATION_STRATEGY = NormalizationStrategy.CASE_SENSITIVE
-                self.execution_dialect.NORMALIZATION_STRATEGY = NormalizationStrategy.CASE_SENSITIVE
             self.known_ids: t.Set[str] = set()
             self.known_branch_ids: t.Set[str] = set()
             self.known_sequence_ids: t.Set[str] = set()
@@ -176,6 +173,13 @@ class _BaseSession(t.Generic[CATALOG, READER, WRITER, DF, CONN]):
     def _has_connection(self) -> bool:
         return hasattr(self, "_connection") and bool(self._connection)
 
+    @property
+    def udf(self) -> UDF_REGISTRATION:
+        return self._udf_registration(self)
+
+    def getActiveSession(self) -> Self:
+        return self
+
     def range(self, *args):
         start = 0
         step = 1
@@ -573,53 +577,10 @@ class _BaseSession(t.Generic[CATALOG, READER, WRITER, DF, CONN]):
             converted_values.append(cls._to_value(value))
         return _create_row(columns, converted_values)
 
-    @property
-    def _is_standalone(self) -> bool:
-        from sqlframe.standalone.session import StandaloneSession
-
-        return isinstance(self, StandaloneSession)
-
-    @property
-    def _is_duckdb(self) -> bool:
-        from sqlframe.duckdb.session import DuckDBSession
-
-        return isinstance(self, DuckDBSession)
-
-    @property
-    def _is_postgres(self) -> bool:
-        from sqlframe.postgres.session import PostgresSession
-
-        return isinstance(self, PostgresSession)
-
-    @property
-    def _is_spark(self) -> bool:
-        from sqlframe.spark.session import SparkSession
-
-        return isinstance(self, SparkSession)
-
-    @property
-    def _is_bigquery(self) -> bool:
-        from sqlframe.bigquery.session import BigQuerySession
-
-        return isinstance(self, BigQuerySession)
-
-    @property
-    def _is_redshift(self) -> bool:
-        from sqlframe.redshift.session import RedshiftSession
-
-        return isinstance(self, RedshiftSession)
-
-    @property
-    def _is_snowflake(self) -> bool:
-        from sqlframe.snowflake.session import SnowflakeSession
-
-        return isinstance(self, SnowflakeSession)
-
     class Builder:
         SQLFRAME_INPUT_DIALECT_KEY = "sqlframe.input.dialect"
         SQLFRAME_OUTPUT_DIALECT_KEY = "sqlframe.output.dialect"
         SQLFRAME_EXECUTION_DIALECT_KEY = "sqlframe.execution.dialect"
-        SQLFRAME_CASE_SENSITIVE_KEY = "spark.sql.caseSensitive"
         SQLFRAME_CONN_KEY = "sqlframe.conn"
         SQLFRAME_SCHEMA_KEY = "sqlframe.schema"
         DEFAULT_INPUT_DIALECT = "spark"
@@ -665,8 +626,6 @@ class _BaseSession(t.Generic[CATALOG, READER, WRITER, DF, CONN]):
                     self._session_kwargs["conn"] = value
                 elif key == self.SQLFRAME_SCHEMA_KEY:
                     self._session_kwargs["schema"] = value
-                elif key == self.SQLFRAME_CASE_SENSITIVE_KEY:
-                    self._session_kwargs["case_sensitive"] = value
                 else:
                     self._session_kwargs[key] = value
             if map:
@@ -676,8 +635,6 @@ class _BaseSession(t.Generic[CATALOG, READER, WRITER, DF, CONN]):
                     self.output_dialect = map[self.SQLFRAME_OUTPUT_DIALECT_KEY]
                 if self.SQLFRAME_EXECUTION_DIALECT_KEY in map:
                     self.execution_dialect = map[self.SQLFRAME_EXECUTION_DIALECT_KEY]
-                if self.SQLFRAME_CASE_SENSITIVE_KEY in map:
-                    self._session_kwargs["case_sensitive"] = map[self.SQLFRAME_CASE_SENSITIVE_KEY]
                 if self.SQLFRAME_CONN_KEY in map:
                     self._session_kwargs["conn"] = map[self.SQLFRAME_CONN_KEY]
                 if self.SQLFRAME_SCHEMA_KEY in map:
@@ -700,15 +657,5 @@ class _BaseSession(t.Generic[CATALOG, READER, WRITER, DF, CONN]):
             self.session.execution_dialect = Dialect.get_or_raise(self.execution_dialect)
             if hasattr(self.session, "_connection") and not self.session._connection:
                 self.session._connection = self._conn
-            if self.session.case_sensitive:
-                self.session.input_dialect.NORMALIZATION_STRATEGY = (
-                    NormalizationStrategy.CASE_SENSITIVE
-                )
-                self.session.output_dialect.NORMALIZATION_STRATEGY = (
-                    NormalizationStrategy.CASE_SENSITIVE
-                )
-                self.session.execution_dialect.NORMALIZATION_STRATEGY = (
-                    NormalizationStrategy.CASE_SENSITIVE
-                )
 
     builder = Builder()

sqlframe-2.1.0/sqlframe/base/udf.py

@@ -0,0 +1,36 @@
+# This code is based on code from Apache Spark under the license found in the LICENSE file located in the 'sqlframe' folder.
+from __future__ import annotations
+
+import typing as t
+
+if t.TYPE_CHECKING:
+    from sqlframe.base._typing import DataTypeOrString, UserDefinedFunctionLike
+    from sqlframe.base.session import _BaseSession
+
+    SESSION = t.TypeVar("SESSION", bound=_BaseSession)
+else:
+    SESSION = t.TypeVar("SESSION")
+
+
+class _BaseUDFRegistration(t.Generic[SESSION]):
+    def __init__(self, sparkSession: SESSION):
+        self.sparkSession = sparkSession
+
+    def register(
+        self,
+        name: str,
+        f: t.Union[t.Callable[..., t.Any], UserDefinedFunctionLike],
+        returnType: t.Optional[DataTypeOrString] = None,
+    ) -> UserDefinedFunctionLike:
+        raise NotImplementedError
+
+    def registerJavaFunction(
+        self,
+        name: str,
+        javaClassName: str,
+        returnType: t.Optional[DataTypeOrString] = None,
+    ) -> None:
+        raise NotImplementedError
+
+    def registerJavaUDAF(self, name: str, javaClassName: str) -> None:
+        raise NotImplementedError

{sqlframe-2.0.0 → sqlframe-2.1.0}/sqlframe/bigquery/session.py

@@ -9,6 +9,7 @@ from sqlframe.bigquery.readwriter import (
     BigQueryDataFrameReader,
     BigQueryDataFrameWriter,
 )
+from sqlframe.bigquery.udf import BigQueryUDFRegistration
 
 if t.TYPE_CHECKING:
     from google.cloud.bigquery.client import Client as BigQueryClient
@@ -25,12 +26,14 @@ class BigQuerySession(
         BigQueryDataFrameWriter,
         BigQueryDataFrame,
         BigQueryConnection,
+        BigQueryUDFRegistration,
     ],
 ):
     _catalog = BigQueryCatalog
     _reader = BigQueryDataFrameReader
     _writer = BigQueryDataFrameWriter
     _df = BigQueryDataFrame
+    _udf_registration = BigQueryUDFRegistration
 
     QUALIFY_INFO_SCHEMA_WITH_DATABASE = True
     SANITIZE_COLUMN_NAMES = True

sqlframe-2.1.0/sqlframe/bigquery/udf.py

@@ -0,0 +1,11 @@
+from __future__ import annotations
+
+import typing as t
+
+from sqlframe.base.udf import _BaseUDFRegistration
+
+if t.TYPE_CHECKING:
+    from sqlframe.bigquery.session import BigQuerySession
+
+
+class BigQueryUDFRegistration(_BaseUDFRegistration["BigQuerySession"]): ...

{sqlframe-2.0.0 → sqlframe-2.1.0}/sqlframe/duckdb/session.py

@@ -11,6 +11,7 @@ from sqlframe.duckdb.readwriter import (
     DuckDBDataFrameReader,
     DuckDBDataFrameWriter,
 )
+from sqlframe.duckdb.udf import DuckDBUDFRegistration
 
 if t.TYPE_CHECKING:
     from duckdb import DuckDBPyConnection
@@ -26,12 +27,14 @@ class DuckDBSession(
         DuckDBDataFrameWriter,
         DuckDBDataFrame,
         DuckDBPyConnection,
+        DuckDBUDFRegistration,
     ]
 ):
     _catalog = DuckDBCatalog
     _reader = DuckDBDataFrameReader
     _writer = DuckDBDataFrameWriter
     _df = DuckDBDataFrame
+    _udf_registration = DuckDBUDFRegistration
 
     def __init__(self, conn: t.Optional[DuckDBPyConnection] = None, *args, **kwargs):
         import duckdb

sqlframe-2.1.0/sqlframe/duckdb/udf.py

@@ -0,0 +1,19 @@
+from __future__ import annotations
+
+import typing as t
+
+from sqlframe.base.udf import _BaseUDFRegistration
+
+if t.TYPE_CHECKING:
+    from sqlframe.base._typing import DataTypeOrString, UserDefinedFunctionLike
+    from sqlframe.duckdb.session import DuckDBSession
+
+
+class DuckDBUDFRegistration(_BaseUDFRegistration["DuckDBSession"]):
+    def register(  # type: ignore
+        self,
+        name: str,
+        f: t.Union[t.Callable[..., t.Any], UserDefinedFunctionLike],
+        returnType: t.Optional[DataTypeOrString] = None,
+    ) -> UserDefinedFunctionLike:
+        self.sparkSession._conn.create_function(name, f, return_type=returnType)  # type: ignore

{sqlframe-2.0.0 → sqlframe-2.1.0}/sqlframe/postgres/session.py

@@ -11,6 +11,7 @@ from sqlframe.postgres.readwriter import (
     PostgresDataFrameReader,
     PostgresDataFrameWriter,
 )
+from sqlframe.postgres.udf import PostgresUDFRegistration
 
 if t.TYPE_CHECKING:
     from psycopg2.extensions import connection as psycopg2_connection
@@ -27,12 +28,14 @@ class PostgresSession(
         PostgresDataFrameWriter,
         PostgresDataFrame,
         psycopg2_connection,
+        PostgresUDFRegistration,
     ],
 ):
     _catalog = PostgresCatalog
     _reader = PostgresDataFrameReader
     _writer = PostgresDataFrameWriter
     _df = PostgresDataFrame
+    _udf_registration = PostgresUDFRegistration
 
     def __init__(self, conn: t.Optional[psycopg2_connection] = None):
         if not hasattr(self, "_conn"):

sqlframe-2.1.0/sqlframe/postgres/udf.py

@@ -0,0 +1,11 @@
+from __future__ import annotations
+
+import typing as t
+
+from sqlframe.base.udf import _BaseUDFRegistration
+
+if t.TYPE_CHECKING:
+    from sqlframe.postgres.session import PostgresSession
+
+
+class PostgresUDFRegistration(_BaseUDFRegistration["PostgresSession"]): ...

{sqlframe-2.0.0 → sqlframe-2.1.0}/sqlframe/redshift/session.py

@@ -10,6 +10,7 @@ from sqlframe.redshift.readwriter import (
     RedshiftDataFrameReader,
     RedshiftDataFrameWriter,
 )
+from sqlframe.redshift.udf import RedshiftUDFRegistration
 
 if t.TYPE_CHECKING:
     from redshift_connector.core import Connection as RedshiftConnection
@@ -24,12 +25,14 @@ class RedshiftSession(
         RedshiftDataFrameWriter,
         RedshiftDataFrame,
         RedshiftConnection,
+        RedshiftUDFRegistration,
     ],
 ):
     _catalog = RedshiftCatalog
     _reader = RedshiftDataFrameReader
     _writer = RedshiftDataFrameWriter
     _df = RedshiftDataFrame
+    _udf_registration = RedshiftUDFRegistration
 
     def __init__(self, conn: t.Optional[RedshiftConnection] = None):
         warnings.warn(

sqlframe-2.1.0/sqlframe/redshift/udf.py

@@ -0,0 +1,11 @@
+from __future__ import annotations
+
+import typing as t
+
+from sqlframe.base.udf import _BaseUDFRegistration
+
+if t.TYPE_CHECKING:
+    from sqlframe.redshift.session import RedshiftSession
+
+
+class RedshiftUDFRegistration(_BaseUDFRegistration["RedshiftSession"]): ...

{sqlframe-2.0.0 → sqlframe-2.1.0}/sqlframe/snowflake/session.py

@@ -4,6 +4,8 @@ import json
 import typing as t
 import warnings
 
+from sqlframe.snowflake.udf import SnowflakeUDFRegistration
+
 try:
     from snowflake.connector.converter import SnowflakeConverter
 except ImportError:
@@ -50,12 +52,14 @@ class SnowflakeSession(
         SnowflakeDataFrameWriter,
         SnowflakeDataFrame,
         SnowflakeConnection,
+        SnowflakeUDFRegistration,
     ],
 ):
     _catalog = SnowflakeCatalog
     _reader = SnowflakeDataFrameReader
     _writer = SnowflakeDataFrameWriter
     _df = SnowflakeDataFrame
+    _udf_registration = SnowflakeUDFRegistration
 
     def __init__(self, conn: t.Optional[SnowflakeConnection] = None):
         import snowflake

sqlframe-2.1.0/sqlframe/snowflake/udf.py

@@ -0,0 +1,11 @@
+from __future__ import annotations
+
+import typing as t
+
+from sqlframe.base.udf import _BaseUDFRegistration
+
+if t.TYPE_CHECKING:
+    from sqlframe.snowflake.session import SnowflakeSession
+
+
+class SnowflakeUDFRegistration(_BaseUDFRegistration["SnowflakeSession"]): ...

{sqlframe-2.0.0 → sqlframe-2.1.0}/sqlframe/spark/session.py

@@ -14,6 +14,7 @@ from sqlframe.spark.readwriter import (
     SparkDataFrameWriter,
 )
 from sqlframe.spark.types import Row
+from sqlframe.spark.udf import SparkUDFRegistration
 
 if t.TYPE_CHECKING:
     import pandas as pd
@@ -32,12 +33,14 @@ class SparkSession(
         SparkDataFrameWriter,
         SparkDataFrame,
         PySparkSession,
+        SparkUDFRegistration,
     ],
 ):
     _catalog = SparkCatalog
     _reader = SparkDataFrameReader
     _writer = SparkDataFrameWriter
     _df = SparkDataFrame
+    _udf_registration = SparkUDFRegistration
 
     def __init__(self, conn: t.Optional[PySparkSession] = None, *args, **kwargs):
         from pyspark.sql.session import DataFrame, SparkSession

sqlframe-2.1.0/sqlframe/spark/udf.py

@@ -0,0 +1,34 @@
+from __future__ import annotations
+
+import typing as t
+
+from sqlframe.base.udf import _BaseUDFRegistration
+
+if t.TYPE_CHECKING:
+    from sqlframe.base._typing import DataTypeOrString, UserDefinedFunctionLike
+    from sqlframe.spark.session import SparkSession  # type: ignore
+
+
+class SparkUDFRegistration(_BaseUDFRegistration["SparkSession"]):
+    def register(
+        self,
+        name: str,
+        f: t.Union[t.Callable[..., t.Any], UserDefinedFunctionLike],
+        returnType: t.Optional[DataTypeOrString] = None,
+    ) -> UserDefinedFunctionLike:
+        return self.sparkSession.spark_session.udf.register(name, f, returnType=returnType)  # type: ignore
+
+    def registerJavaFunction(
+        self,
+        name: str,
+        javaClassName: str,
+        returnType: t.Optional[DataTypeOrString] = None,
+    ) -> None:
+        self.sparkSession.spark_session.udf.registerJavaFunction(
+            name,
+            javaClassName,
+            returnType=returnType,  # type: ignore
+        )
+
+    def registerJavaUDAF(self, name: str, javaClassName: str) -> None:
+        self.sparkSession.spark_session.udf.registerJavaUDAF(name, javaClassName)

{sqlframe-2.0.0 → sqlframe-2.1.0}/sqlframe/standalone/session.py

@@ -9,6 +9,7 @@ from sqlframe.standalone.readwriter import (
     StandaloneDataFrameReader,
     StandaloneDataFrameWriter,
 )
+from sqlframe.standalone.udf import StandaloneUDFRegistration
 
 
 class StandaloneSession(
@@ -18,12 +19,14 @@ class StandaloneSession(
         StandaloneDataFrameWriter,
         StandaloneDataFrame,
         object,
+        StandaloneUDFRegistration,
     ]
 ):  # type: ignore
     _catalog = StandaloneCatalog
     _reader = StandaloneDataFrameReader
     _writer = StandaloneDataFrameWriter
     _df = StandaloneDataFrame
+    _udf_registration = StandaloneUDFRegistration
 
     class Builder(_BaseSession.Builder):
         DEFAULT_INPUT_DIALECT = "spark"

sqlframe-2.1.0/sqlframe/standalone/udf.py

@@ -0,0 +1,11 @@
+from __future__ import annotations
+
+import typing as t
+
+from sqlframe.base.udf import _BaseUDFRegistration
+
+if t.TYPE_CHECKING:
+    from sqlframe.standalone.session import StandaloneSession
+
+
+class StandaloneUDFRegistration(_BaseUDFRegistration["StandaloneSession"]): ...

{sqlframe-2.0.0 → sqlframe-2.1.0}/sqlframe.egg-info/PKG-INFO

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

{sqlframe-2.0.0 → sqlframe-2.1.0}/sqlframe.egg-info/SOURCES.txt

@@ -72,6 +72,7 @@ sqlframe/base/readerwriter.py
 sqlframe/base/session.py
 sqlframe/base/transforms.py
 sqlframe/base/types.py
+sqlframe/base/udf.py
 sqlframe/base/util.py
 sqlframe/base/window.py
 sqlframe/base/mixins/__init__.py
@@ -88,6 +89,7 @@ sqlframe/bigquery/group.py
 sqlframe/bigquery/readwriter.py
 sqlframe/bigquery/session.py
 sqlframe/bigquery/types.py
+sqlframe/bigquery/udf.py
 sqlframe/bigquery/window.py
 sqlframe/duckdb/__init__.py
 sqlframe/duckdb/catalog.py
@@ -99,6 +101,7 @@ sqlframe/duckdb/group.py
 sqlframe/duckdb/readwriter.py
 sqlframe/duckdb/session.py
 sqlframe/duckdb/types.py
+sqlframe/duckdb/udf.py
 sqlframe/duckdb/window.py
 sqlframe/postgres/__init__.py
 sqlframe/postgres/catalog.py
@@ -110,6 +113,7 @@ sqlframe/postgres/group.py
 sqlframe/postgres/readwriter.py
 sqlframe/postgres/session.py
 sqlframe/postgres/types.py
+sqlframe/postgres/udf.py
 sqlframe/postgres/window.py
 sqlframe/redshift/__init__.py
 sqlframe/redshift/catalog.py
@@ -120,6 +124,7 @@ sqlframe/redshift/group.py
 sqlframe/redshift/readwriter.py
 sqlframe/redshift/session.py
 sqlframe/redshift/types.py
+sqlframe/redshift/udf.py
 sqlframe/redshift/window.py
 sqlframe/snowflake/__init__.py
 sqlframe/snowflake/catalog.py
@@ -131,6 +136,7 @@ sqlframe/snowflake/group.py
 sqlframe/snowflake/readwriter.py
 sqlframe/snowflake/session.py
 sqlframe/snowflake/types.py
+sqlframe/snowflake/udf.py
 sqlframe/snowflake/window.py
 sqlframe/spark/__init__.py
 sqlframe/spark/catalog.py
@@ -142,6 +148,7 @@ sqlframe/spark/group.py
 sqlframe/spark/readwriter.py
 sqlframe/spark/session.py
 sqlframe/spark/types.py
+sqlframe/spark/udf.py
 sqlframe/spark/window.py
 sqlframe/standalone/__init__.py
 sqlframe/standalone/catalog.py
@@ -152,6 +159,7 @@ sqlframe/standalone/group.py
 sqlframe/standalone/readwriter.py
 sqlframe/standalone/session.py
 sqlframe/standalone/types.py
+sqlframe/standalone/udf.py
 sqlframe/standalone/window.py
 sqlframe/testing/__init__.py
 sqlframe/testing/utils.py
@@ -186,6 +194,7 @@ tests/integration/engines/duck/test_duckdb_catalog.py
 tests/integration/engines/duck/test_duckdb_dataframe.py
 tests/integration/engines/duck/test_duckdb_reader.py
 tests/integration/engines/duck/test_duckdb_session.py
+tests/integration/engines/duck/test_duckdb_udf.py
 tests/integration/engines/postgres/__init__.py
 tests/integration/engines/postgres/test_postgres_catalog.py
 tests/integration/engines/postgres/test_postgres_dataframe.py

{sqlframe-2.0.0 → sqlframe-2.1.0}/sqlframe.egg-info/requires.txt

@@ -1,5 +1,5 @@
 prettytable<3.11.0
-sqlglot<25.9,>=24.0.0
+sqlglot<25.11,>=24.0.0
 typing_extensions<5,>=4.8
 
 [bigquery]
@@ -9,7 +9,7 @@ google-cloud-bigquery[pandas]<4,>=3
 [dev]
 duckdb<1.1,>=0.9
 mypy<1.12,>=1.10.0
-openai<1.38,>=1.30
+openai<1.41,>=1.30
 pandas-stubs<3,>=2
 pandas<3,>=2
 psycopg<4,>=3.1
@@ -39,7 +39,7 @@ duckdb<1.1,>=0.9
 pandas<3,>=2
 
 [openai]
-openai<1.38,>=1.30
+openai<1.41,>=1.30
 
 [pandas]
 pandas<3,>=2