summaryrefslogtreecommitdiffstats
path: root/sqlglot/dataframe/sql/_typing.py
diff options
context:
space:
mode:
Diffstat (limited to 'sqlglot/dataframe/sql/_typing.py')
-rw-r--r--sqlglot/dataframe/sql/_typing.py18
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]