add models

oleksandr.volha · oleksandr.volha · commit 1719f09f1bd4 · 2024-06-24T13:16:37.000+03:00
diff --git a/uncoder-core/app/translator/core/models/functions/join.py b/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
diff --git a/uncoder-core/app/translator/core/models/functions/union.py b/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
