sqlframe 0.1.dev3__py3-none-any.whl → 1.1.0__py3-none-any.whl

sqlframe/_version.py

@@ -12,5 +12,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '0.1.dev3'
-__version_tuple__ = version_tuple = (0, 1, 'dev3')
+__version__ = version = '1.1.0'
+__version_tuple__ = version_tuple = (1, 1, 0)

sqlframe/base/catalog.py

@@ -8,7 +8,7 @@ from sqlglot import MappingSchema, exp
 
 from sqlframe.base.decorators import normalize
 from sqlframe.base.exceptions import TableSchemaError
-from sqlframe.base.util import to_schema
+from sqlframe.base.util import ensure_column_mapping, to_schema
 
 if t.TYPE_CHECKING:
     from sqlglot.schema import ColumnMapping
@@ -82,6 +82,7 @@ class _BaseCatalog(t.Generic[SESSION, DF]):
                 raise TableSchemaError(
                     "This session does not have access to a catalog that can lookup column information. See docs for explicitly defining columns or using a session that can automatically determine this."
                 )
+        column_mapping = ensure_column_mapping(column_mapping)  # type: ignore
         self._schema.add_table(table, column_mapping, dialect=self.session.input_dialect)
 
     @normalize(["dbName"])

sqlframe/base/dataframe.py

@@ -417,7 +417,7 @@ class _BaseDataFrame(t.Generic[SESSION, WRITER, NA, STAT, GROUP_DATA]):
         from sqlframe.base.session import _BaseSession
 
         value = expression.sql(dialect=_BaseSession().input_dialect).encode("utf-8")
-        hash = f"t{zlib.crc32(value)}"[:6]
+        hash = f"t{zlib.crc32(value)}"[:9]
         return self.session._normalize_string(hash)
 
     def _get_select_expressions(
@@ -606,8 +606,11 @@ class _BaseDataFrame(t.Generic[SESSION, WRITER, NA, STAT, GROUP_DATA]):
         return df._convert_leaf_to_cte(sequence_id=new_sequence_id)
 
     @operation(Operation.WHERE)
-    def where(self, column: t.Union[Column, bool], **kwargs) -> Self:
-        col = self._ensure_and_normalize_col(column)
+    def where(self, column: t.Union[Column, str, bool], **kwargs) -> Self:
+        if isinstance(column, str):
+            col = sqlglot.parse_one(column, dialect=self.session.input_dialect)
+        else:
+            col = self._ensure_and_normalize_col(column)
         return self.copy(expression=self.expression.where(col.expression))
 
     filter = where
@@ -662,7 +665,7 @@ class _BaseDataFrame(t.Generic[SESSION, WRITER, NA, STAT, GROUP_DATA]):
         | 16|  Bob|    85|
         +---+-----+------+
         """
-        return self.join.__wrapped__(self, other, how="cross")
+        return self.join.__wrapped__(self, other, how="cross")  # type: ignore
 
     @operation(Operation.FROM)
     def join(
@@ -769,7 +772,7 @@ class _BaseDataFrame(t.Generic[SESSION, WRITER, NA, STAT, GROUP_DATA]):
         new_df = self.copy(expression=join_expression)
         new_df.pending_join_hints.extend(self.pending_join_hints)
         new_df.pending_hints.extend(other_df.pending_hints)
-        new_df = new_df.select.__wrapped__(new_df, *select_column_names)
+        new_df = new_df.select.__wrapped__(new_df, *select_column_names)  # type: ignore
         return new_df
 
     @operation(Operation.ORDER_BY)
@@ -1094,7 +1097,7 @@ class _BaseDataFrame(t.Generic[SESSION, WRITER, NA, STAT, GROUP_DATA]):
         )
         if existing_col_index:
             expression = self.expression.copy()
-            expression.expressions[existing_col_index] = col.expression
+            expression.expressions[existing_col_index] = col.alias(colName).expression
             return self.copy(expression=expression)
         return self.copy().select(col.alias(colName), append=True)
 

sqlframe/base/decorators.py

@@ -11,7 +11,7 @@ if t.TYPE_CHECKING:
     from sqlframe.base.catalog import _BaseCatalog
 
 
-def normalize(normalize_kwargs: t.List[str]):
+def normalize(normalize_kwargs: t.List[str]) -> t.Callable[[t.Callable], t.Callable]:
     """
     Decorator used around DataFrame methods to indicate what type of operation is being performed from the
     ordered Operation enums. This is used to determine which operations should be performed on a CTE vs.
@@ -23,9 +23,9 @@ def normalize(normalize_kwargs: t.List[str]):
     in cases where there is overlap in names.
     """
 
-    def decorator(func: t.Callable):
+    def decorator(func: t.Callable) -> t.Callable:
         @functools.wraps(func)
-        def wrapper(self: _BaseCatalog, *args, **kwargs):
+        def wrapper(self: _BaseCatalog, *args, **kwargs) -> _BaseCatalog:
             kwargs.update(dict(zip(func.__code__.co_varnames[1:], args)))
             for kwarg in normalize_kwargs:
                 if kwarg in kwargs:
@@ -43,9 +43,9 @@ def normalize(normalize_kwargs: t.List[str]):
     return decorator
 
 
-def func_metadata(unsupported_engines: t.Optional[t.Union[str, t.List[str]]] = None):
-    def _metadata(func):
-        func.unsupported_engines = ensure_list(unsupported_engines) if unsupported_engines else []
+def func_metadata(unsupported_engines: t.Optional[t.Union[str, t.List[str]]] = None) -> t.Callable:
+    def _metadata(func: t.Callable) -> t.Callable:
+        func.unsupported_engines = ensure_list(unsupported_engines) if unsupported_engines else []  # type: ignore
         return func
 
     return _metadata

sqlframe/base/mixins/readwriter_mixins.py

@@ -108,6 +108,9 @@ class PandasWriterMixin(_BaseDataFrameWriter, t.Generic[SESSION, DF]):
                 raise NotImplementedError("Append mode is not supported for parquet.")
             pandas_df.to_parquet(path, **kwargs)
         elif format == "json":
+            # Pandas versions are inconsistent on how to handle True/False index so we just remove it
+            # since in all versions it will not result in an index column in the output.
+            del kwargs["index"]
             kwargs["mode"] = mode
             kwargs["orient"] = "records"
             pandas_df.to_json(path, lines=True, **kwargs)

sqlframe/base/operations.py

@@ -23,7 +23,7 @@ class Operation(IntEnum):
     LIMIT = 7
 
 
-def operation(op: Operation):
+def operation(op: Operation) -> t.Callable[[t.Callable], t.Callable]:
     """
     Decorator used around DataFrame methods to indicate what type of operation is being performed from the
     ordered Operation enums. This is used to determine which operations should be performed on a CTE vs.
@@ -35,9 +35,9 @@ def operation(op: Operation):
     in cases where there is overlap in names.
     """
 
-    def decorator(func: t.Callable):
+    def decorator(func: t.Callable) -> t.Callable:
         @functools.wraps(func)
-        def wrapper(self: _BaseDataFrame, *args, **kwargs):
+        def wrapper(self: _BaseDataFrame, *args, **kwargs) -> _BaseDataFrame:
             if self.last_op == Operation.INIT:
                 self = self._convert_leaf_to_cte()
                 self.last_op = Operation.NO_OP
@@ -47,7 +47,7 @@ def operation(op: Operation):
                 self = self._convert_leaf_to_cte()
             df: t.Union[_BaseDataFrame, _BaseGroupedData] = func(self, *args, **kwargs)
             df.last_op = new_op  # type: ignore
-            return df
+            return df  # type: ignore
 
         wrapper.__wrapped__ = func  # type: ignore
         return wrapper
@@ -55,7 +55,7 @@ def operation(op: Operation):
     return decorator
 
 
-def group_operation(op: Operation):
+def group_operation(op: Operation) -> t.Callable[[t.Callable], t.Callable]:
     """
     Decorator used around DataFrame methods to indicate what type of operation is being performed from the
     ordered Operation enums. This is used to determine which operations should be performed on a CTE vs.
@@ -67,9 +67,9 @@ def group_operation(op: Operation):
     in cases where there is overlap in names.
     """
 
-    def decorator(func: t.Callable):
+    def decorator(func: t.Callable) -> t.Callable:
         @functools.wraps(func)
-        def wrapper(self: _BaseGroupedData, *args, **kwargs):
+        def wrapper(self: _BaseGroupedData, *args, **kwargs) -> _BaseDataFrame:
             if self._df.last_op == Operation.INIT:
                 self._df = self._df._convert_leaf_to_cte()
                 self._df.last_op = Operation.NO_OP

sqlframe/base/session.py

@@ -11,9 +11,9 @@ from collections import defaultdict
 from functools import cached_property
 
 import sqlglot
-from more_itertools import take
 from sqlglot import Dialect, exp
 from sqlglot.expressions import parse_identifier
+from sqlglot.helper import seq_get
 from sqlglot.optimizer import optimize
 from sqlglot.optimizer.normalize_identifiers import normalize_identifiers
 from sqlglot.optimizer.qualify_columns import (
@@ -211,10 +211,10 @@ class _BaseSession(t.Generic[CATALOG, READER, WRITER, DF, CONN]):
                     row_types.append((row_name, default_type))
                 return "struct<" + ", ".join(f"{k}: {v}" for (k, v) in row_types) + ">"
             elif isinstance(value, dict):
-                sample_row = take(1, value.items())
+                sample_row = seq_get(list(value.items()), 0)
                 if not sample_row:
                     return None
-                key, value = sample_row[0]
+                key, value = sample_row
                 default_key = get_default_data_type(key)
                 default_value = get_default_data_type(value)
                 if not default_key or not default_value:
@@ -313,24 +313,16 @@ class _BaseSession(t.Generic[CATALOG, READER, WRITER, DF, CONN]):
         sel_expression = exp.Select(**select_kwargs)
         if empty_df:
             sel_expression = sel_expression.where(exp.false())
-        # if empty_df:
-        #     if not column_mapping:
-        #         # If we don't have rows or columns then we just return a null with a false expression
-        #         sel_expression = (
-        #             exp.Select().select("null").from_("VALUES (NULL)").where(exp.false())
-        #         )
-        #     else:
-        #         # Ensure no results are returned if the dataframe is expected to be empty instead of
-        #         # a row of null values
-        #         sel_expression = sel_expression.where(exp.false())
         return self._create_df(sel_expression)
 
-    def sql(self, sqlQuery: t.Union[str, exp.Expression]) -> DF:
-        expression = self._optimize(
+    def sql(self, sqlQuery: t.Union[str, exp.Expression], optimize: bool = True) -> DF:
+        expression = (
             sqlglot.parse_one(sqlQuery, read=self.input_dialect)
             if isinstance(sqlQuery, str)
             else sqlQuery
         )
+        if optimize:
+            expression = self._optimize(expression)
         if self.temp_views:
             replacement_mapping = {}
             for table in expression.find_all(exp.Table):

sqlframe/base/transforms.py

@@ -5,7 +5,9 @@ import typing as t
 from sqlglot import expressions as exp
 
 
-def replace_id_value(node, replacement_mapping: t.Dict[exp.Identifier, exp.Identifier]):
+def replace_id_value(
+    node: exp.Expression, replacement_mapping: t.Dict[exp.Identifier, exp.Identifier]
+) -> exp.Expression:
     if isinstance(node, exp.Identifier) and node in replacement_mapping:
         node = node.replace(replacement_mapping[node].copy())
     return node

sqlframe/base/util.py

@@ -9,6 +9,7 @@ from sqlglot.dialects.dialect import Dialect, DialectType
 from sqlglot.schema import ensure_column_mapping as sqlglot_ensure_column_mapping
 
 if t.TYPE_CHECKING:
+    from pandas.core.frame import DataFrame as PandasDataFrame
     from pyspark.sql.dataframe import SparkSession as PySparkSession
 
     from sqlframe.base import types
@@ -97,7 +98,7 @@ def get_tables_from_expression_with_join(expression: exp.Select) -> t.List[exp.T
     return [left_table] + other_tables
 
 
-def to_csv(options: t.Dict[str, OptionalPrimitiveType], equality_char: str = "="):
+def to_csv(options: t.Dict[str, OptionalPrimitiveType], equality_char: str = "=") -> str:
     return ", ".join(
         [f"{k}{equality_char}{v}" for k, v in (options or {}).items() if v is not None]
     )
@@ -112,11 +113,13 @@ def ensure_column_mapping(schema: t.Union[str, StructType]) -> t.Dict:
         }
     # TODO: Make a protocol with a `simpleString` attribute as what it looks for instead of the actual
     # `StructType` object.
+    elif hasattr(schema, "simpleString"):
+        return {struct_field.name: struct_field.dataType.simpleString() for struct_field in schema}
     return sqlglot_ensure_column_mapping(schema)  # type: ignore
 
 
 # SO: https://stackoverflow.com/questions/37513355/converting-pandas-dataframe-into-spark-dataframe-error
-def get_equivalent_spark_type(pandas_type):
+def get_equivalent_spark_type(pandas_type) -> types.DataType:
     """
     This method will retrieve the corresponding spark type given a pandas
     type.
@@ -139,7 +142,7 @@ def get_equivalent_spark_type(pandas_type):
     return type_map.get(str(pandas_type).lower(), types.StringType())
 
 
-def pandas_to_spark_schema(pandas_df):
+def pandas_to_spark_schema(pandas_df: PandasDataFrame) -> types.StructType:
     """
     This method will return a spark dataframe schema given a pandas dataframe.
 

sqlframe/bigquery/catalog.py

@@ -3,7 +3,6 @@ from __future__ import annotations
 import fnmatch
 import typing as t
 
-from google.cloud.bigquery import StandardSqlDataType
 from sqlglot import exp
 
 from sqlframe.base.catalog import CatalogMetadata, Column, Function
@@ -16,8 +15,10 @@ from sqlframe.base.mixins.catalog_mixins import (
 from sqlframe.base.util import schema_, to_schema
 
 if t.TYPE_CHECKING:
-    from sqlframe.bigquery.session import BigQuerySession  # noqa
+    from google.cloud.bigquery import StandardSqlDataType
+
     from sqlframe.bigquery.dataframe import BigQueryDataFrame  # noqa
+    from sqlframe.bigquery.session import BigQuerySession  # noqa
 
 
 class BigQueryCatalog(

sqlframe/bigquery/session.py

@@ -11,9 +11,10 @@ from sqlframe.bigquery.readwriter import (
 )
 
 if t.TYPE_CHECKING:
-    from google.cloud import bigquery
+    from google.cloud.bigquery.client import Client as BigQueryClient
     from google.cloud.bigquery.dbapi.connection import Connection as BigQueryConnection
 else:
+    BigQueryClient = t.Any
     BigQueryConnection = t.Any
 
 
@@ -48,7 +49,7 @@ class BigQuerySession(
             self.default_dataset = default_dataset
 
     @property
-    def _client(self) -> bigquery.client.Client:
+    def _client(self) -> BigQueryClient:
         assert self._connection
         return self._connection._client
 

sqlframe/duckdb/readwriter.py

@@ -5,6 +5,9 @@ from __future__ import annotations
 import logging
 import typing as t
 
+from sqlglot import exp
+from sqlglot.helper import ensure_list
+
 from sqlframe.base.readerwriter import _BaseDataFrameReader, _BaseDataFrameWriter
 from sqlframe.base.util import ensure_column_mapping, to_csv
 
@@ -69,13 +72,22 @@ class DuckDBDataFrameReader(_BaseDataFrameReader["DuckDBSession", "DuckDBDataFra
         |100|NULL|
         +---+----+
         """
+        if schema:
+            column_mapping = ensure_column_mapping(schema)
+            select_columns = [x.expression for x in self._to_casted_columns(column_mapping)]
+            if format == "csv":
+                duckdb_columns = ", ".join(
+                    [f"'{column}': '{dtype}'" for column, dtype in column_mapping.items()]
+                )
+                options["columns"] = "{" + duckdb_columns + "}"
+        else:
+            select_columns = [exp.Star()]
         if format:
-            sql = f"SELECT * FROM read_{format}('{path}', {to_csv(options)})"
+            paths = ",".join([f"'{path}'" for path in ensure_list(path)])
+            from_clause = f"read_{format}([{paths}], {to_csv(options)})"
         else:
-            sql = f"select * from '{path}'"
-        df = self.session.sql(sql)
-        if schema:
-            df = df.select(*self._to_casted_columns(ensure_column_mapping(schema)))
+            from_clause = f"'{path}'"
+        df = self.session.sql(exp.select(*select_columns).from_(from_clause), optimize=False)
         self.session._last_loaded_file = path  # type: ignore
         return df
 
@@ -87,7 +99,7 @@ class DuckDBDataFrameWriter(_BaseDataFrameWriter["DuckDBSession", "DuckDBDataFra
             return
         if mode == "append":
             raise NotImplementedError("Append mode not supported")
-        options = to_csv(options, equality_char=" ")
+        options = to_csv(options, equality_char=" ")  # type: ignore
         sqls = self._df.sql(pretty=False, optimize=False, as_list=True)
         for i, sql in enumerate(sqls):
             if i < len(sqls) - 1:

{sqlframe-0.1.dev3.dist-info → sqlframe-1.1.0.dist-info}/METADATA

@@ -1,11 +1,12 @@
 Metadata-Version: 2.1
 Name: sqlframe
-Version: 0.1.dev3
-Summary: PySpark Dataframe API Compatible SQL Generator
+Version: 1.1.0
+Summary: Taking the Spark out of PySpark by converting to SQL
 Home-page: https://github.com/eakmanrq/sqlframe
 Author: Ryan Eakman
 Author-email: eakmanrq@gmail.com
 License: MIT
+Platform: UNKNOWN
 Classifier: Development Status :: 5 - Production/Stable
 Classifier: Intended Audience :: Developers
 Classifier: Intended Audience :: Science/Research
@@ -16,60 +17,63 @@ Classifier: Programming Language :: Python :: 3 :: Only
 Requires-Python: >=3.8
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: prettytable
-Requires-Dist: sqlglot
+Requires-Dist: prettytable (<3.11.0)
+Requires-Dist: sqlglot (<24.1,>=24.0.0)
 Provides-Extra: bigquery
-Requires-Dist: google-cloud-bigquery[pandas] ; extra == 'bigquery'
-Requires-Dist: google-cloud-bigquery-storage ; extra == 'bigquery'
+Requires-Dist: google-cloud-bigquery-storage (<3,>=2) ; extra == 'bigquery'
+Requires-Dist: google-cloud-bigquery[pandas] (<4,>=3) ; extra == 'bigquery'
+Requires-Dist: pandas (<3,>=2) ; extra == 'bigquery'
 Provides-Extra: dev
-Requires-Dist: duckdb ; extra == 'dev'
-Requires-Dist: mkdocs ==1.4.2 ; extra == 'dev'
-Requires-Dist: mkdocs-include-markdown-plugin ==4.0.3 ; extra == 'dev'
-Requires-Dist: mkdocs-material ==9.0.5 ; extra == 'dev'
-Requires-Dist: mkdocs-material-extensions ==1.1.1 ; extra == 'dev'
-Requires-Dist: mypy ; extra == 'dev'
-Requires-Dist: pandas ; extra == 'dev'
-Requires-Dist: pymdown-extensions ; extra == 'dev'
-Requires-Dist: psycopg ; extra == 'dev'
-Requires-Dist: pyarrow ; extra == 'dev'
-Requires-Dist: pyspark ; extra == 'dev'
-Requires-Dist: pytest ; extra == 'dev'
-Requires-Dist: pytest-postgresql ; extra == 'dev'
-Requires-Dist: pytest-xdist ; extra == 'dev'
-Requires-Dist: pre-commit ; extra == 'dev'
-Requires-Dist: ruff ; extra == 'dev'
-Requires-Dist: typing-extensions ; extra == 'dev'
-Requires-Dist: types-psycopg2 ; extra == 'dev'
+Requires-Dist: duckdb (<0.11,>=0.9) ; extra == 'dev'
+Requires-Dist: mypy (<1.11,>=1.10.0) ; extra == 'dev'
+Requires-Dist: pandas-stubs (<3,>=2) ; extra == 'dev'
+Requires-Dist: pandas (<3,>=2) ; extra == 'dev'
+Requires-Dist: psycopg (<4,>=3.1) ; extra == 'dev'
+Requires-Dist: pyarrow (<17,>=10) ; extra == 'dev'
+Requires-Dist: pyspark (<3.6,>=2) ; extra == 'dev'
+Requires-Dist: pytest-postgresql (<7,>=6) ; extra == 'dev'
+Requires-Dist: pytest-xdist (<3.7,>=3.6) ; extra == 'dev'
+Requires-Dist: pytest (<8.3,>=8.2.0) ; extra == 'dev'
+Requires-Dist: ruff (<0.5,>=0.4.4) ; extra == 'dev'
+Requires-Dist: types-psycopg2 (<3,>=2.9) ; extra == 'dev'
+Requires-Dist: typing-extensions (<5,>=4.11) ; extra == 'dev'
+Requires-Dist: pre-commit (>=3.5) ; (python_version == "3.8") and extra == 'dev'
+Requires-Dist: pre-commit (<3.8,>=3.7) ; (python_version >= "3.9") and extra == 'dev'
+Provides-Extra: docs
+Requires-Dist: mkdocs-include-markdown-plugin (==4.0.3) ; extra == 'docs'
+Requires-Dist: mkdocs-material-extensions (==1.1.1) ; extra == 'docs'
+Requires-Dist: mkdocs-material (==9.0.5) ; extra == 'docs'
+Requires-Dist: mkdocs (==1.4.2) ; extra == 'docs'
+Requires-Dist: pymdown-extensions ; extra == 'docs'
 Provides-Extra: duckdb
-Requires-Dist: duckdb ; extra == 'duckdb'
-Requires-Dist: pandas ; extra == 'duckdb'
+Requires-Dist: duckdb (<0.11,>=0.9) ; extra == 'duckdb'
+Requires-Dist: pandas (<3,>=2) ; extra == 'duckdb'
 Provides-Extra: postgres
-Requires-Dist: psycopg2 ; extra == 'postgres'
+Requires-Dist: pandas (<3,>=2) ; extra == 'postgres'
+Requires-Dist: psycopg2 (<3,>=2.8) ; extra == 'postgres'
 Provides-Extra: redshift
-Requires-Dist: redshift-connector ; extra == 'redshift'
+Requires-Dist: pandas (<3,>=2) ; extra == 'redshift'
+Requires-Dist: redshift-connector (<2.2.0,>=2.1.1) ; extra == 'redshift'
 Provides-Extra: snowflake
-Requires-Dist: snowflake-connector-python[pandas,secure-local-storage] ; extra == 'snowflake'
+Requires-Dist: pandas (<3,>=2) ; extra == 'snowflake'
+Requires-Dist: snowflake-connector-python[pandas,secure-local-storage] (<3.11,>=3.10.0) ; extra == 'snowflake'
 Provides-Extra: spark
-Requires-Dist: pyspark ; extra == 'spark'
+Requires-Dist: pyspark (<3.6,>=2) ; extra == 'spark'
 
 <div align="center">
-  <img src="docs/images/sqlframe_logo.png" alt="SQLFrame Logo" width="400"/>
+  <img src="https://sqlframe.readthedocs.io/en/latest/docs/images/sqlframe_logo.png" alt="SQLFrame Logo" width="400"/>
 </div>
 
-![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)
-
 SQLFrame implements the PySpark DataFrame API in order to enable running transformation pipelines directly on database engines - no Spark clusters or dependencies required.
 
-![SQLFrame Diagram](docs/images/sqlframe_diagram.png)
-
-SQLFrame currently supports the following engines:
+SQLFrame currently supports the following engines (many more in development):
 
-* [BigQuery](docs/bigquery.md)
-* [DuckDB](docs/duckdb.md)
-* [Postgres](docs/postgres.md)
+* [BigQuery](https://sqlframe.readthedocs.io/en/latest/bigquery/)
+* [DuckDB](https://sqlframe.readthedocs.io/en/latest/duckdb)
+* [Postgres](https://sqlframe.readthedocs.io/en/latest/postgres)
 
 SQLFrame also has a "Standalone" session that be used to generate SQL without any connection to a database engine.
-* [Standalone](docs/standalone.md)
+* [Standalone](https://sqlframe.readthedocs.io/en/latest/standalone)
 
 SQLFrame is great for:
 
@@ -101,7 +105,7 @@ from sqlframe.bigquery import Window
 
 session = BigQuerySession()
 table_path = "bigquery-public-data.samples.natality"
-# Get the top 5 years with the greatest year-over-year % change in new families with a single child
+# Top 5 years with the greatest year-over-year % change in new families with single child
 df = (
     session.table(table_path)
     .where(F.col("ever_born") == 1)
@@ -118,17 +122,15 @@ df = (
     )
     .orderBy(F.abs(F.col("percent_change")).desc())
     .select(
-        F.col("year").alias("Year"),
-        F.format_number("num_single_child_families", 0).alias("number of new families single child"),
+        F.col("year").alias("year"),
+        F.format_number("num_single_child_families", 0).alias("new families single child"),
         F.format_number(F.col("percent_change") * 100, 2).alias("percent change"),
     )
     .limit(5)
 )
 ```
 ```python
-df.sql()
-```
-```sql
+>>> df.sql()
 WITH `t94228` AS (
   SELECT
     `natality`.`year` AS `year`,
@@ -147,7 +149,7 @@ WITH `t94228` AS (
 )
 SELECT
   `t39093`.`year` AS `year`,
-  FORMAT('%\'.0f', ROUND(CAST(`t39093`.`num_single_child_families` AS FLOAT64), 0)) AS `number of new families single child`,
+  FORMAT('%\'.0f', ROUND(CAST(`t39093`.`num_single_child_families` AS FLOAT64), 0)) AS `new families single child`,
   FORMAT('%\'.2f', ROUND(CAST((((`t39093`.`num_single_child_families` - `t39093`.`last_year_num_single_child_families`) / `t39093`.`last_year_num_single_child_families`) * 100) AS FLOAT64), 2)) AS `percent change`
 FROM `t39093` AS `t39093`
 ORDER BY
@@ -155,16 +157,16 @@ ORDER BY
 LIMIT 5
 ```
 ```python
-df.show()
-```
-```
-+------+-------------------------------------+----------------+
-| year | number of new families single child | percent change |
-+------+-------------------------------------+----------------+
-| 1989 |              1,650,246              |     25.02      |
-| 1974 |               783,448               |     14.49      |
-| 1977 |              1,057,379              |     11.38      |
-| 1985 |              1,308,476              |     11.15      |
-| 1975 |               868,985               |     10.92      |
-+------+-------------------------------------+----------------+
+>>> df.show()
++------+---------------------------+----------------+
+| year | new families single child | percent change |
++------+---------------------------+----------------+
+| 1989 |         1,650,246         |     25.02      |
+| 1974 |          783,448          |     14.49      |
+| 1977 |         1,057,379         |     11.38      |
+| 1985 |         1,308,476         |     11.15      |
+| 1975 |          868,985          |     10.92      |
++------+---------------------------+----------------+
 ```
+
+

{sqlframe-0.1.dev3.dist-info → sqlframe-1.1.0.dist-info}/RECORD

@@ -1,35 +1,34 @@
 sqlframe/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-sqlframe/_version.py,sha256=rzktxOC5O6_wrybdrzjz0JCVUczin8eenqU8ABIxLo8,419
+sqlframe/_version.py,sha256=CqDGE4B1ZqZ-56mxeOFcXRTmlxrdOh4ayrjbcPjziE4,411
 sqlframe/base/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 sqlframe/base/_typing.py,sha256=DuTay8-o9W-pw3RPZCgLunKNJLS9PkaV11G_pxXp9NY,1256
-sqlframe/base/catalog.py,sha256=Y9v7ZdpUVeFjjqcKyrRTBKK1H-IXM5SuIwVm5blSXK4,36984
+sqlframe/base/catalog.py,sha256=P55_mLrk9KUC2LRYyLgSzVAan1Lx6EBNfdDjMEtc9DA,37086
 sqlframe/base/column.py,sha256=K9TtpBjVsFK9NtEX9ZQscU6qZIKiVVh1zj3jG9HifyA,15110
-sqlframe/base/dataframe.py,sha256=OVtDEZWAwMxlavIVhH_A3rZs8R2Dm4rorJTYhWVeDX0,58785
-sqlframe/base/decorators.py,sha256=pqpscaRHQdi0BmsDWxr_Ji-CVNXW1LIwF02PLuS9Wdo,1971
+sqlframe/base/dataframe.py,sha256=lNBMm79rX1DAt5vj4qKuxDwyhJdnhDROcOPcqVDxNHE,58971
+sqlframe/base/decorators.py,sha256=fnqT1Hqa0J_gUurDcVY1Dcscj6SXFxFJ5PKAw-xe5sU,2097
 sqlframe/base/exceptions.py,sha256=pCB9hXX4jxZWzNg3JN1i38cv3BmpUlee5NoLYx3YXIQ,208
 sqlframe/base/function_alternatives.py,sha256=to0kv3MTJmQFeVTMcitz0AxBIoUJC3cu5LkEY5aJpoo,31318
 sqlframe/base/functions.py,sha256=rm-8Mmz_156G3Hdwfdf-HF4HDeVkv0pBcDCJskoRINQ,53541
 sqlframe/base/group.py,sha256=sKoaI2aLMih9nJTQfqzfJ00NbBcGQtArWXYHT40motQ,4060
 sqlframe/base/normalize.py,sha256=Ie6IcrD9dL-xBUKgDoh_c_gfLw68tBK5AmiprCA8MXE,3633
-sqlframe/base/operations.py,sha256=eU67Zs_Y6WuT48PgLHURl-L8kDlVWPszngCSJnBM8cs,3297
+sqlframe/base/operations.py,sha256=fVlAse6-WdQnEaHghRZVHXOesQ3OnKQwBnVYv5nVRiI,3457
 sqlframe/base/readerwriter.py,sha256=cgg7KuO7Eu8fScKOg1KyNFAcgnsjpU6yusPVs0o52a4,25213
-sqlframe/base/session.py,sha256=XK38XBaeykUsIllGf85STAMFDWQGnQu_idVT_hGU1GM,22381
-sqlframe/base/transforms.py,sha256=xqGx2Rgjf7SXzZdMaexxuhOsIJwxhPsCJZxBpgCF4dU,428
+sqlframe/base/session.py,sha256=evVdd-FGKkp-Wg80UG5289iRtBihLFfkqrcXTH64_R8,21926
+sqlframe/base/transforms.py,sha256=EKwUpfp83bncEs_MNmI2OO7gV6vA_Rr89ZWmE4eETSw,468
 sqlframe/base/types.py,sha256=1CwMW9Q1inYzQcPTyjv1QANtVSHha8ZmBigmopQET98,11925
-sqlframe/base/util.py,sha256=6MYy2IZfsPXokugLcX_MWMuBb5RIxRjACJh4ak9nveE,7330
+sqlframe/base/util.py,sha256=SeUC2pcSBGnsS1W5PL1p-IGC6bJG8_2a7En2hxSTmpA,7597
 sqlframe/base/window.py,sha256=yyKvoNi41vL2t7XK2Ysjp8Q2FNIu3BYv-9EPtp5og6k,4944
 sqlframe/base/mixins/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 sqlframe/base/mixins/catalog_mixins.py,sha256=ZNzNn-cWB0RwT7L1KZCWYke2JlP-cZze0MDNOzSfHew,14093
-sqlframe/base/mixins/readwriter_mixins.py,sha256=yMtKoZ9MCgGI5D-j1g2YOW6LEAcBt7_uipLs6AQh6o4,4433
+sqlframe/base/mixins/readwriter_mixins.py,sha256=N2nsXOG3A2j6O3N195U-_fYOZMkqfifGcfduxODUcxs,4656
 sqlframe/bigquery/__init__.py,sha256=i2NsMbiXOj2xphCtPuNk6cVw4iYeq5_B1I9dVI9aGAk,712
-sqlframe/bigquery/catalog.py,sha256=BvTRfTeiIGzkQsRzzuKlmkTLaX5OwLNyu320Ygo_xM8,9272
+sqlframe/bigquery/catalog.py,sha256=HdRXZfZczoyLHEQ0y30nfCFKBvTTOJ1s6t0mafN_bGk,9277
 sqlframe/bigquery/column.py,sha256=E1tUa62Y5HajkhgFuebU9zohrGyieudcHzTT8gfalio,40
 sqlframe/bigquery/dataframe.py,sha256=fPQ6043aSS_ds30WsvrYOgNZJPH0jq7BeNHGLQ2MEW4,1372
 sqlframe/bigquery/functions.py,sha256=RF8yG_4MS3at_60V0NNTE5ADERJZa7kZGYFWI4ST3jM,11149
-sqlframe/bigquery/functions.pyi,sha256=doa_vkSY6OOj5_-DsYdOFJJWh1w9TIJQlAyGRkOUPA4,11929
 sqlframe/bigquery/group.py,sha256=C3wLlyDYgNqDszTNbojoXdZQuqBYw3FRtOqpHhfaf0E,392
 sqlframe/bigquery/readwriter.py,sha256=-r8nGOACWJ51XACONppgKsoMQMwcCgTJ0fp_mXW4JOs,871
-sqlframe/bigquery/session.py,sha256=1wALkRiIplVHcbiED3ukqZ6yEeStp7At97VuPKkY9CE,2658
+sqlframe/bigquery/session.py,sha256=1-hE1Wr2b6SqfD4M_-OGMqjaSbhD6wSQd74v71xHZv8,2709
 sqlframe/bigquery/types.py,sha256=KwNyuXIo-2xVVd4bZED3YrQOobKCtemlxGrJL7DrTC8,34
 sqlframe/bigquery/window.py,sha256=6GKPzuxeSapJakBaKBeT9VpED1ACdjggDv9JRILDyV0,35
 sqlframe/duckdb/__init__.py,sha256=t85TA3ufZtL1weQNFmEs8itCSwbJFtw03-p0GT4XGf8,669
@@ -37,9 +36,8 @@ sqlframe/duckdb/catalog.py,sha256=XDFf-si0_myFBOM8rwlg2G80FdgBXBe7pkzk8xxYcuE,38
 sqlframe/duckdb/column.py,sha256=wkEPcp3xVsH5nC3kpacXqNkRv9htPtBgt-0uFRxIRNs,56
 sqlframe/duckdb/dataframe.py,sha256=9T6GV4JScaApFSA4T7fixot78HMUgkjGxU7TgjolOOM,1410
 sqlframe/duckdb/functions.py,sha256=srvzbk_Wg-wQPFGYp624dRDyYJghi47M8E-Tu7pBdY0,1507
-sqlframe/duckdb/functions.pyi,sha256=JXcPclyAzRbP9MK3kkRPHRIOwd0Cy1KbAv485dfxA5o,5033
 sqlframe/duckdb/group.py,sha256=sYTExtNprfbW74LWc_Lyjc1G6K1FogQsdILU2599Bq8,384
-sqlframe/duckdb/readwriter.py,sha256=PhuI342ak7alffj-W2bpotCq2QcXz6QjG_lm3nurWG8,3801
+sqlframe/duckdb/readwriter.py,sha256=TC0LigUmCRpcdx4B8Mb5ap5ifFBrjmbqXmhUB5rG87U,4376
 sqlframe/duckdb/session.py,sha256=TCAVsSqBGGj1Otb2iIkSkWqjbzzg1MeDAafGN928-O8,1893
 sqlframe/duckdb/types.py,sha256=KwNyuXIo-2xVVd4bZED3YrQOobKCtemlxGrJL7DrTC8,34
 sqlframe/duckdb/window.py,sha256=6GKPzuxeSapJakBaKBeT9VpED1ACdjggDv9JRILDyV0,35
@@ -48,7 +46,6 @@ sqlframe/postgres/catalog.py,sha256=EjRbgHKSZKYFdKYkicIIpXiezvA0vOhAQxNyspKNxXk,
 sqlframe/postgres/column.py,sha256=E1tUa62Y5HajkhgFuebU9zohrGyieudcHzTT8gfalio,40
 sqlframe/postgres/dataframe.py,sha256=bv_y9D9w03x-sfLdippb8n4goFQGazg1j0gZEPHe98k,1372
 sqlframe/postgres/functions.py,sha256=UNL7dE6LmzekvolwqWB-aFt8ITamxeSfuG50_NP_G8c,2133
-sqlframe/postgres/functions.pyi,sha256=xAtohxyW6NQSucLvv1IO-dmdS9BSOdku3p93FoITyPM,4767
 sqlframe/postgres/group.py,sha256=XmUndNdoYgy6-84xMMVCdCI06j7Na5QrSFhhqvUKcGw,392
 sqlframe/postgres/readwriter.py,sha256=P3gXZkKKc9R92slKph5K0FPJPjsdgMYDRX-8LPsOI5s,871
 sqlframe/postgres/session.py,sha256=oKh8-j9MN6msVheQNCYoGmej9ktFLTTHmlMP58uZ3nw,1936
@@ -94,8 +91,8 @@ sqlframe/standalone/readwriter.py,sha256=n2uoebNdL_t6_eaXNkpu7Zv2UmZ9I3rASuo01gG
 sqlframe/standalone/session.py,sha256=xWxBh-OtH--LmWtpDboOBpwKLcaBK5JV-IF2gCra5k0,1192
 sqlframe/standalone/types.py,sha256=KwNyuXIo-2xVVd4bZED3YrQOobKCtemlxGrJL7DrTC8,34
 sqlframe/standalone/window.py,sha256=6GKPzuxeSapJakBaKBeT9VpED1ACdjggDv9JRILDyV0,35
-sqlframe-0.1.dev3.dist-info/LICENSE,sha256=VZu79YgW780qxaFJMr0t5ZgbOYEh04xWoxaWOaqIGWk,1068
-sqlframe-0.1.dev3.dist-info/METADATA,sha256=QN2odqXNrZhGYEUqAHDCl5H-r3v_pQL2UQsOIXWOqys,6203
-sqlframe-0.1.dev3.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-sqlframe-0.1.dev3.dist-info/top_level.txt,sha256=T0_RpoygaZSF6heeWwIDQgaP0varUdSK1pzjeJZRjM8,9
-sqlframe-0.1.dev3.dist-info/RECORD,,
+sqlframe-1.1.0.dist-info/LICENSE,sha256=VZu79YgW780qxaFJMr0t5ZgbOYEh04xWoxaWOaqIGWk,1068
+sqlframe-1.1.0.dist-info/METADATA,sha256=RBSfrpj8FYCqz79aL88JvpDRIkKZvbASKiwT0YqwXm0,6873
+sqlframe-1.1.0.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
+sqlframe-1.1.0.dist-info/top_level.txt,sha256=T0_RpoygaZSF6heeWwIDQgaP0varUdSK1pzjeJZRjM8,9
+sqlframe-1.1.0.dist-info/RECORD,,

{sqlframe-0.1.dev3.dist-info → sqlframe-1.1.0.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: bdist_wheel (0.43.0)
+Generator: bdist_wheel (0.37.1)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

sqlframe/bigquery/functions.pyi

@@ -1,269 +0,0 @@
-import typing as t
-
-from sqlframe.base.column import Column as Column
-from sqlframe.base.function_alternatives import (
-    array_union_using_array_concat as array_union,
-)
-from sqlframe.base.function_alternatives import (
-    base64_from_blob as base64,
-)
-from sqlframe.base.function_alternatives import (
-    bit_length_from_length as bit_length,
-)
-from sqlframe.base.function_alternatives import (
-    collect_set_from_list_distinct as collect_set,
-)
-from sqlframe.base.function_alternatives import (
-    concat_ws_from_array_to_string as concat_ws,
-)
-from sqlframe.base.function_alternatives import (
-    dayofmonth_from_extract_with_day as dayofmonth,
-)
-from sqlframe.base.function_alternatives import (
-    dayofweek_from_extract as dayofweek,
-)
-from sqlframe.base.function_alternatives import (
-    dayofyear_from_extract as dayofyear,
-)
-from sqlframe.base.function_alternatives import (  # noqa
-    e_literal as e,
-)
-from sqlframe.base.function_alternatives import (
-    element_at_using_brackets as element_at,
-)
-from sqlframe.base.function_alternatives import (
-    expm1_from_exp as expm1,
-)
-from sqlframe.base.function_alternatives import (
-    factorial_from_case_statement as factorial,
-)
-from sqlframe.base.function_alternatives import (
-    format_string_with_format as format_string,
-)
-from sqlframe.base.function_alternatives import (
-    hash_from_farm_fingerprint as hash,
-)
-from sqlframe.base.function_alternatives import (
-    hex_casted_as_bytes as hex,
-)
-from sqlframe.base.function_alternatives import (
-    hour_from_extract as hour,
-)
-from sqlframe.base.function_alternatives import (
-    instr_using_strpos as instr,
-)
-from sqlframe.base.function_alternatives import (
-    isnull_using_equal as isnull,
-)
-from sqlframe.base.function_alternatives import (
-    last_day_with_cast as last_day,
-)
-from sqlframe.base.function_alternatives import (
-    log1p_from_log as log1p,
-)
-from sqlframe.base.function_alternatives import (
-    make_date_from_date_func as make_date,
-)
-from sqlframe.base.function_alternatives import (
-    minute_from_extract as minute,
-)
-from sqlframe.base.function_alternatives import (
-    month_from_extract as month,
-)
-from sqlframe.base.function_alternatives import (
-    nanvl_as_case as nanvl,
-)
-from sqlframe.base.function_alternatives import (
-    overlay_from_substr as overlay,
-)
-from sqlframe.base.function_alternatives import (
-    percentile_approx_without_accuracy_and_plural as percentile_approx,
-)
-from sqlframe.base.function_alternatives import (
-    quarter_from_extract as quarter,
-)
-from sqlframe.base.function_alternatives import (
-    rand_no_seed as rand,
-)
-from sqlframe.base.function_alternatives import (
-    regexp_extract_only_one_group as regexp_extract,
-)
-from sqlframe.base.function_alternatives import (
-    rint_from_round as rint,
-)
-from sqlframe.base.function_alternatives import (
-    second_from_extract as second,
-)
-from sqlframe.base.function_alternatives import (
-    sequence_from_generate_array as sequence,
-)
-from sqlframe.base.function_alternatives import (
-    sha1_force_sha1_and_to_hex as sha1,
-)
-from sqlframe.base.function_alternatives import (
-    split_with_split as split,
-)
-from sqlframe.base.function_alternatives import (
-    to_date_from_timestamp as to_date,
-)
-from sqlframe.base.function_alternatives import (
-    weekofyear_from_extract_as_isoweek as weekofyear,
-)
-from sqlframe.base.function_alternatives import (
-    year_from_extract as year,
-)
-from sqlframe.base.functions import abs as abs
-from sqlframe.base.functions import acos as acos
-from sqlframe.base.functions import acosh as acosh
-from sqlframe.base.functions import add_months as add_months
-from sqlframe.base.functions import approx_count_distinct as approx_count_distinct
-from sqlframe.base.functions import approxCountDistinct as approxCountDistinct
-from sqlframe.base.functions import array as array
-from sqlframe.base.functions import array_contains as array_contains
-from sqlframe.base.functions import array_join as array_join
-from sqlframe.base.functions import asc as asc
-from sqlframe.base.functions import asc_nulls_first as asc_nulls_first
-from sqlframe.base.functions import asc_nulls_last as asc_nulls_last
-from sqlframe.base.functions import ascii as ascii
-from sqlframe.base.functions import asin as asin
-from sqlframe.base.functions import asinh as asinh
-from sqlframe.base.functions import atan as atan
-from sqlframe.base.functions import atan2 as atan2
-from sqlframe.base.functions import atanh as atanh
-from sqlframe.base.functions import avg as avg
-from sqlframe.base.functions import bitwise_not as bitwise_not
-from sqlframe.base.functions import bitwiseNOT as bitwiseNOT
-from sqlframe.base.functions import cbrt as cbrt
-from sqlframe.base.functions import ceil as ceil
-from sqlframe.base.functions import ceiling as ceiling
-from sqlframe.base.functions import coalesce as coalesce
-from sqlframe.base.functions import col as col
-from sqlframe.base.functions import collect_list as collect_list
-from sqlframe.base.functions import concat as concat
-from sqlframe.base.functions import corr as corr
-from sqlframe.base.functions import cos as cos
-from sqlframe.base.functions import cosh as cosh
-from sqlframe.base.functions import cot as cot
-from sqlframe.base.functions import count as count
-from sqlframe.base.functions import covar_pop as covar_pop
-from sqlframe.base.functions import covar_samp as covar_samp
-from sqlframe.base.functions import csc as csc
-from sqlframe.base.functions import cume_dist as cume_dist
-from sqlframe.base.functions import current_date as current_date
-from sqlframe.base.functions import current_timestamp as current_timestamp
-from sqlframe.base.functions import date_add as date_add
-from sqlframe.base.functions import date_diff as date_diff
-from sqlframe.base.functions import date_format as date_format
-from sqlframe.base.functions import date_sub as date_sub
-from sqlframe.base.functions import date_trunc as date_trunc
-from sqlframe.base.functions import dense_rank as dense_rank
-from sqlframe.base.functions import desc as desc
-from sqlframe.base.functions import desc_nulls_first as desc_nulls_first
-from sqlframe.base.functions import desc_nulls_last as desc_nulls_last
-from sqlframe.base.functions import exp as exp
-from sqlframe.base.functions import explode as explode
-from sqlframe.base.functions import explode_outer as explode_outer
-from sqlframe.base.functions import expr as expr
-from sqlframe.base.functions import floor as floor
-from sqlframe.base.functions import get_json_object as get_json_object
-from sqlframe.base.functions import greatest as greatest
-from sqlframe.base.functions import initcap as initcap
-from sqlframe.base.functions import input_file_name as input_file_name
-from sqlframe.base.functions import isnan as isnan
-from sqlframe.base.functions import lag as lag
-from sqlframe.base.functions import lead as lead
-from sqlframe.base.functions import least as least
-from sqlframe.base.functions import length as length
-from sqlframe.base.functions import lit as lit
-from sqlframe.base.functions import log as log
-from sqlframe.base.functions import log2 as log2
-from sqlframe.base.functions import log10 as log10
-from sqlframe.base.functions import lower as lower
-from sqlframe.base.functions import lpad as lpad
-from sqlframe.base.functions import ltrim as ltrim
-from sqlframe.base.functions import max as max
-from sqlframe.base.functions import max_by as max_by
-from sqlframe.base.functions import md5 as md5
-from sqlframe.base.functions import mean as mean
-from sqlframe.base.functions import min as min
-from sqlframe.base.functions import min_by as min_by
-from sqlframe.base.functions import nth_value as nth_value
-from sqlframe.base.functions import ntile as ntile
-from sqlframe.base.functions import octet_length as octet_length
-from sqlframe.base.functions import percent_rank as percent_rank
-from sqlframe.base.functions import posexplode as posexplode
-from sqlframe.base.functions import posexplode_outer as posexplode_outer
-from sqlframe.base.functions import pow as pow
-from sqlframe.base.functions import rank as rank
-from sqlframe.base.functions import regexp_replace as regexp_replace
-from sqlframe.base.functions import repeat as repeat
-from sqlframe.base.functions import reverse as reverse
-from sqlframe.base.functions import round as round
-from sqlframe.base.functions import row_number as row_number
-from sqlframe.base.functions import rpad as rpad
-from sqlframe.base.functions import rtrim as rtrim
-from sqlframe.base.functions import sec as sec
-from sqlframe.base.functions import shiftLeft as shiftLeft
-from sqlframe.base.functions import shiftleft as shiftleft
-from sqlframe.base.functions import shiftRight as shiftRight
-from sqlframe.base.functions import shiftright as shiftright
-from sqlframe.base.functions import signum as signum
-from sqlframe.base.functions import sin as sin
-from sqlframe.base.functions import sinh as sinh
-from sqlframe.base.functions import size as size
-from sqlframe.base.functions import soundex as soundex
-from sqlframe.base.functions import sqrt as sqrt
-from sqlframe.base.functions import stddev as stddev
-from sqlframe.base.functions import stddev_pop as stddev_pop
-from sqlframe.base.functions import stddev_samp as stddev_samp
-from sqlframe.base.functions import struct as struct
-from sqlframe.base.functions import substring as substring
-from sqlframe.base.functions import sum as sum
-from sqlframe.base.functions import sum_distinct as sum_distinct
-from sqlframe.base.functions import sumDistinct as sumDistinct
-from sqlframe.base.functions import tan as tan
-from sqlframe.base.functions import tanh as tanh
-from sqlframe.base.functions import timestamp_seconds as timestamp_seconds
-from sqlframe.base.functions import to_timestamp as to_timestamp
-from sqlframe.base.functions import toDegrees as toDegrees
-from sqlframe.base.functions import toRadians as toRadians
-from sqlframe.base.functions import translate as translate
-from sqlframe.base.functions import trim as trim
-from sqlframe.base.functions import trunc as trunc
-from sqlframe.base.functions import unbase64 as unbase64
-from sqlframe.base.functions import unhex as unhex
-from sqlframe.base.functions import upper as upper
-from sqlframe.base.functions import var_pop as var_pop
-from sqlframe.base.functions import var_samp as var_samp
-from sqlframe.base.functions import variance as variance
-from sqlframe.base.functions import when as when
-from sqlframe.base.util import get_func_from_session as get_func_from_session
-
-if t.TYPE_CHECKING:
-    from sqlframe.base._typing import ColumnOrLiteral, ColumnOrName
-
-def array_distinct(col: ColumnOrName) -> Column: ...
-def array_max(col: ColumnOrName) -> Column: ...
-def array_min(col: ColumnOrName) -> Column: ...
-def array_position(col: ColumnOrName, value: ColumnOrLiteral) -> Column: ...
-def array_remove(col: ColumnOrName, value: ColumnOrLiteral) -> Column: ...
-def array_sort(col: ColumnOrName, asc: t.Optional[bool] = ...) -> Column: ...
-def bin(col: ColumnOrName) -> Column: ...
-def bround(col: ColumnOrName, scale: t.Optional[int] = ...) -> Column: ...
-def degrees(col: ColumnOrName) -> Column: ...
-def format_number(col: ColumnOrName, d: int) -> Column: ...
-def from_unixtime(col: ColumnOrName, format: t.Optional[str] = ...) -> Column: ...
-def months_between(
-    date1: ColumnOrName, date2: ColumnOrName, roundOff: t.Optional[bool] = ...
-) -> Column: ...
-def next_day(col: ColumnOrName, dayOfWeek: str) -> Column: ...
-def radians(col: ColumnOrName) -> Column: ...
-def slice(
-    x: ColumnOrName, start: t.Union[ColumnOrName, int], length: t.Union[ColumnOrName, int]
-) -> Column: ...
-def sort_array(col: ColumnOrName, asc: t.Optional[bool] = ...) -> Column: ...
-def substring_index(str: ColumnOrName, delim: str, count: int) -> Column: ...
-def typeof(col: ColumnOrName) -> Column: ...
-def unix_timestamp(
-    timestamp: t.Optional[ColumnOrName] = ..., format: t.Optional[str] = ...
-) -> Column: ...

sqlframe/duckdb/functions.pyi

@@ -1,183 +0,0 @@
-from sqlframe.base.function_alternatives import (  # noqa
-    e_literal as e,
-    expm1_from_exp as expm1,
-    log1p_from_log as log1p,
-    rint_from_round as rint,
-    kurtosis_from_kurtosis_pop as kurtosis,
-    collect_set_from_list_distinct as collect_set,
-    first_always_ignore_nulls as first,
-    factorial_ensure_int as factorial,
-    isnull_using_equal as isnull,
-    nanvl_as_case as nanvl,
-    percentile_approx_without_accuracy as percentile_approx,
-    rand_no_seed as rand,
-    base64_from_blob as base64,
-    decode_from_blob as decode,
-    format_string_with_pipes as format_string,
-    overlay_from_substr as overlay,
-    split_no_limit as split,
-    arrays_overlap_using_intersect as arrays_overlap,
-    slice_as_list_slice as slice,
-    array_join_null_replacement_with_transform as array_join,
-    element_at_using_brackets as element_at,
-    array_remove_using_filter as array_remove,
-    array_union_using_list_concat as array_union,
-    array_min_from_sort as array_min,
-    array_max_from_sort as array_max,
-    sequence_from_generate_series as sequence,
-)
-from sqlframe.base.functions import (
-    abs as abs,
-    acos as acos,
-    add_months as add_months,
-    approxCountDistinct as approxCountDistinct,
-    approx_count_distinct as approx_count_distinct,
-    array as array,
-    array_contains as array_contains,
-    array_distinct as array_distinct,
-    array_intersect as array_intersect,
-    array_position as array_position,
-    array_sort as array_sort,
-    asc as asc,
-    asc_nulls_first as asc_nulls_first,
-    asc_nulls_last as asc_nulls_last,
-    ascii as ascii,
-    asin as asin,
-    atan as atan,
-    atan2 as atan2,
-    avg as avg,
-    bin as bin,
-    bit_length as bit_length,
-    bitwiseNOT as bitwiseNOT,
-    bitwise_not as bitwise_not,
-    cbrt as cbrt,
-    ceil as ceil,
-    ceiling as ceiling,
-    coalesce as coalesce,
-    col as col,
-    collect_list as collect_list,
-    concat as concat,
-    concat_ws as concat_ws,
-    corr as corr,
-    cos as cos,
-    cot as cot,
-    count as count,
-    countDistinct as countDistinct,
-    count_distinct as count_distinct,
-    covar_pop as covar_pop,
-    covar_samp as covar_samp,
-    create_map as create_map,
-    cume_dist as cume_dist,
-    current_date as current_date,
-    current_timestamp as current_timestamp,
-    date_add as date_add,
-    date_diff as date_diff,
-    date_format as date_format,
-    date_sub as date_sub,
-    date_trunc as date_trunc,
-    dayofmonth as dayofmonth,
-    dayofweek as dayofweek,
-    dayofyear as dayofyear,
-    degrees as degrees,
-    dense_rank as dense_rank,
-    desc as desc,
-    desc_nulls_first as desc_nulls_first,
-    desc_nulls_last as desc_nulls_last,
-    encode as encode,
-    exp as exp,
-    explode as explode,
-    expr as expr,
-    flatten as flatten,
-    floor as floor,
-    from_unixtime as from_unixtime,
-    get_json_object as get_json_object,
-    greatest as greatest,
-    grouping_id as grouping_id,
-    hash as hash,
-    hex as hex,
-    hour as hour,
-    input_file_name as input_file_name,
-    instr as instr,
-    isnan as isnan,
-    lag as lag,
-    last as last,
-    lead as lead,
-    least as least,
-    length as length,
-    levenshtein as levenshtein,
-    lit as lit,
-    locate as locate,
-    log as log,
-    log10 as log10,
-    log2 as log2,
-    lower as lower,
-    lpad as lpad,
-    ltrim as ltrim,
-    make_date as make_date,
-    map_from_arrays as map_from_arrays,
-    max as max,
-    max_by as max_by,
-    md5 as md5,
-    mean as mean,
-    min as min,
-    min_by as min_by,
-    minute as minute,
-    month as month,
-    months_between as months_between,
-    nth_value as nth_value,
-    ntile as ntile,
-    percent_rank as percent_rank,
-    percentile as percentile,
-    pow as pow,
-    quarter as quarter,
-    radians as radians,
-    rank as rank,
-    regexp_extract as regexp_extract,
-    regexp_replace as regexp_replace,
-    repeat as repeat,
-    reverse as reverse,
-    round as round,
-    row_number as row_number,
-    rpad as rpad,
-    rtrim as rtrim,
-    second as second,
-    shiftLeft as shiftLeft,
-    shiftRight as shiftRight,
-    shiftleft as shiftleft,
-    shiftright as shiftright,
-    signum as signum,
-    sin as sin,
-    size as size,
-    skewness as skewness,
-    sort_array as sort_array,
-    soundex as soundex,
-    sqrt as sqrt,
-    stddev as stddev,
-    stddev_pop as stddev_pop,
-    stddev_samp as stddev_samp,
-    struct as struct,
-    substring as substring,
-    sum as sum,
-    sumDistinct as sumDistinct,
-    sum_distinct as sum_distinct,
-    tan as tan,
-    timestamp_seconds as timestamp_seconds,
-    toDegrees as toDegrees,
-    toRadians as toRadians,
-    to_date as to_date,
-    to_timestamp as to_timestamp,
-    translate as translate,
-    trim as trim,
-    trunc as trunc,
-    typeof as typeof,
-    unbase64 as unbase64,
-    unhex as unhex,
-    unix_timestamp as unix_timestamp,
-    upper as upper,
-    var_pop as var_pop,
-    var_samp as var_samp,
-    variance as variance,
-    weekofyear as weekofyear,
-    when as when,
-    year as year,
-)

sqlframe/postgres/functions.pyi

@@ -1,167 +0,0 @@
-from sqlframe.base.function_alternatives import (  # noqa
-    e_literal as e,
-    expm1_from_exp as expm1,
-    log1p_from_log as log1p,
-    rint_from_round as rint,
-    collect_set_from_list_distinct as collect_set,
-    isnan_using_equal as isnan,
-    isnull_using_equal as isnull,
-    nanvl_as_case as nanvl,
-    rand_no_seed as rand,
-    round_cast_as_numeric as round,
-    year_from_extract as year,
-    quarter_from_extract as quarter,
-    month_from_extract as month,
-    dayofweek_from_extract_with_isodow as dayofweek,
-    dayofmonth_from_extract_with_day as dayofmonth,
-    dayofyear_from_extract_doy as dayofyear,
-    hour_from_extract as hour,
-    minute_from_extract as minute,
-    second_from_extract as second,
-    weekofyear_from_extract_as_week as weekofyear,
-    make_date_casted_as_integer as make_date,
-    date_add_by_multiplication as date_add,
-    date_sub_by_multiplication as date_sub,
-    date_diff_with_subtraction as date_diff,
-    add_months_by_multiplication as add_months,
-    months_between_from_age_and_extract as months_between,
-    from_unixtime_from_timestamp as from_unixtime,
-    unix_timestamp_from_extract as unix_timestamp,
-    base64_from_blob as base64,
-    bas64_from_encode as base64,
-    unbase64_from_decode as unbase64,
-    decode_from_convert_from as decode,
-    encode_from_convert_to as encode,
-    format_number_from_to_char as format_number,
-    format_string_with_format as format_string,
-    split_from_regex_split_to_array as split,
-    array_contains_any as array_contains,
-    slice_with_brackets as slice,
-    element_at_using_brackets as element_at,
-    get_json_object_using_arrow_op as get_json_object,
-    array_min_from_subquery as array_min,
-    array_max_from_subquery as array_max,
-)
-from sqlframe.base.functions import (
-    abs as abs,
-    acos as acos,
-    acosh as acosh,
-    array as array,
-    array_join as array_join,
-    array_position as array_position,
-    array_remove as array_remove,
-    arrays_overlap as arrays_overlap,
-    asc as asc,
-    asc_nulls_first as asc_nulls_first,
-    asc_nulls_last as asc_nulls_last,
-    ascii as ascii,
-    asin as asin,
-    asinh as asinh,
-    atan as atan,
-    atan2 as atan2,
-    atanh as atanh,
-    avg as avg,
-    bit_length as bit_length,
-    bitwiseNOT as bitwiseNOT,
-    bitwise_not as bitwise_not,
-    cbrt as cbrt,
-    ceil as ceil,
-    ceiling as ceiling,
-    coalesce as coalesce,
-    col as col,
-    collect_list as collect_list,
-    concat as concat,
-    concat_ws as concat_ws,
-    corr as corr,
-    cos as cos,
-    cosh as cosh,
-    cot as cot,
-    count as count,
-    countDistinct as countDistinct,
-    count_distinct as count_distinct,
-    covar_pop as covar_pop,
-    covar_samp as covar_samp,
-    cume_dist as cume_dist,
-    current_date as current_date,
-    current_timestamp as current_timestamp,
-    date_format as date_format,
-    date_trunc as date_trunc,
-    degrees as degrees,
-    dense_rank as dense_rank,
-    desc as desc,
-    desc_nulls_first as desc_nulls_first,
-    desc_nulls_last as desc_nulls_last,
-    exp as exp,
-    explode as explode,
-    expr as expr,
-    factorial as factorial,
-    floor as floor,
-    greatest as greatest,
-    initcap as initcap,
-    input_file_name as input_file_name,
-    instr as instr,
-    lag as lag,
-    lead as lead,
-    least as least,
-    length as length,
-    levenshtein as levenshtein,
-    lit as lit,
-    locate as locate,
-    log as log,
-    log10 as log10,
-    log2 as log2,
-    lower as lower,
-    lpad as lpad,
-    ltrim as ltrim,
-    max as max,
-    md5 as md5,
-    mean as mean,
-    min as min,
-    nth_value as nth_value,
-    ntile as ntile,
-    octet_length as octet_length,
-    overlay as overlay,
-    percent_rank as percent_rank,
-    percentile as percentile,
-    pow as pow,
-    radians as radians,
-    rank as rank,
-    regexp_replace as regexp_replace,
-    repeat as repeat,
-    reverse as reverse,
-    row_number as row_number,
-    rpad as rpad,
-    rtrim as rtrim,
-    shiftLeft as shiftLeft,
-    shiftRight as shiftRight,
-    shiftleft as shiftleft,
-    shiftright as shiftright,
-    signum as signum,
-    sin as sin,
-    sinh as sinh,
-    size as size,
-    soundex as soundex,
-    sqrt as sqrt,
-    stddev as stddev,
-    stddev_pop as stddev_pop,
-    stddev_samp as stddev_samp,
-    substring as substring,
-    sum as sum,
-    sumDistinct as sumDistinct,
-    sum_distinct as sum_distinct,
-    tan as tan,
-    tanh as tanh,
-    timestamp_seconds as timestamp_seconds,
-    toDegrees as toDegrees,
-    toRadians as toRadians,
-    to_date as to_date,
-    to_timestamp as to_timestamp,
-    translate as translate,
-    trim as trim,
-    trunc as trunc,
-    upper as upper,
-    var_pop as var_pop,
-    var_samp as var_samp,
-    variance as variance,
-    when as when,
-)