add models

Author: oleksandr.volha

uncoder-core/app/translator/core/models/functions/join.py

@@ -0,0 +1,26 @@
+from dataclasses import dataclass, field
+from typing import Union
+
+from app.translator.core.custom_types.functions import FunctionType
+from app.translator.core.models.field import Alias, Field
+from app.translator.core.models.functions.base import Function
+from app.translator.core.models.identifier import Identifier
+from app.translator.core.models.query_container import TokenizedQueryContainer
+from app.translator.tools.custom_enum import CustomEnum
+
+
+class JoinType(CustomEnum):
+    inner = "inner"
+    left = "left"
+    right = "right"
+    cross = "cross"
+
+
+@dataclass
+class JoinFunction(Function):
+    name: str = FunctionType.join
+    alias: Alias = None
+    type_: str = JoinType.inner
+    tokenized_query_container: TokenizedQueryContainer = None
+    condition: list[Union[Alias, Field, Identifier]] = field(default_factory=list)
+    preset_log_source_str: str = None

uncoder-core/app/translator/core/models/functions/union.py

@@ -0,0 +1,12 @@
+from dataclasses import dataclass
+
+from app.translator.core.custom_types.functions import FunctionType
+from app.translator.core.models.functions.base import Function
+from app.translator.core.models.query_container import TokenizedQueryContainer
+
+
+@dataclass
+class UnionFunction(Function):
+    name: str = FunctionType.union
+    tokenized_query_container: TokenizedQueryContainer = None
+    preset_log_source_str: str = None