diff options
Diffstat (limited to 'sqlglot/dataframe/sql/_typing.py')
-rw-r--r-- | sqlglot/dataframe/sql/_typing.py | 18 |
1 files changed, 18 insertions, 0 deletions
diff --git a/sqlglot/dataframe/sql/_typing.py b/sqlglot/dataframe/sql/_typing.py new file mode 100644 index 0000000..fb46026 --- /dev/null +++ b/sqlglot/dataframe/sql/_typing.py @@ -0,0 +1,18 @@ +from __future__ import annotations + +import datetime +import typing as t + +from sqlglot import expressions as exp + +if t.TYPE_CHECKING: + from sqlglot.dataframe.sql.column import Column + from sqlglot.dataframe.sql.types import StructType + +ColumnLiterals = t.Union[str, float, int, bool, t.List, t.Tuple, datetime.date, datetime.datetime] +ColumnOrName = t.Union[Column, str] +ColumnOrLiteral = t.Union[ + Column, str, float, int, bool, t.List, t.Tuple, datetime.date, datetime.datetime +] +SchemaInput = t.Union[str, t.List[str], StructType, t.Dict[str, t.Optional[str]]] +OutputExpressionContainer = t.Union[exp.Select, exp.Create, exp.Insert] |