changes

unknowncoder05 · unknowncoder05 · commit c005fa73b10d · 2021-08-28T20:27:34.000-05:00
diff --git a/compiler/python_compiler/engines/py3_8/Conditional.py b/compiler/python_compiler/engines/py3_8/Conditional.py
@@ -0,0 +1,33 @@
+from .Fragment import Fragment
+from utils.flags import *
+from utils.CustomLogging import CustomLogging
+from python_compiler.engines.utils.types import get_python_type_str, ANY
+
+def get_conditional_condition(fragment) -> str:
+    if not (condition := fragment.get(ATTRIBUTE_CONDITIONAL_CONDITION)):
+        CustomLogging.critical(f"Fragment type conditional '{ATTRIBUTE_CONDITIONAL_CONDITION}' attribute does not exist")
+    return condition
+
+def get_conditional_code(fragment) -> list:
+    if not (code := fragment.get(ATTRIBUTE_CONDITIONAL_CODE)):
+        CustomLogging.critical(f"Fragment type conditional '{ATTRIBUTE_CONDITIONAL_CODE}' attribute does not exist")
+    return code
+class Conditional(Fragment):
+    condition:str
+    code:list
+    def __init__(self, blueprint, *args, **kwargs) -> None:
+        super().__init__( blueprint, *args, **kwargs)
+        self.condition = get_conditional_condition(blueprint)
+        self.code = get_conditional_code(blueprint)
+    def code_lines_compile(self) -> list:
+        code_build_lines = []
+        for line in self.code:
+            code_build_lines.append(TAB*(self.level+1)+self.general_compile(line))
+        return code_build_lines
+    def compile(self)->str:
+        fragment_lines = []
+        fragment_lines.append(f"if {self.condition}:")
+        fragment_lines.extend(self.code_lines_compile())
+        fragment_build = "\n".join(fragment_lines)
+        
+        return fragment_build
diff --git a/compiler/python_compiler/engines/py3_8/FunctionCall.py b/compiler/python_compiler/engines/py3_8/FunctionCall.py
@@ -0,0 +1,43 @@
+from .Fragment import Fragment
+from utils.flags import *
+from utils.CustomLogging import CustomLogging
+from python_compiler.engines.utils.types import get_python_type_str, ANY
+
+def get_function_name(fragment) -> str:
+    if not (function_name := fragment.get(ATTRIBUTE_FUNCTION_CALL_NAME)):
+        CustomLogging.critical(f"Fragment type function_call '{ATTRIBUTE_FUNCTION_CALL_NAME}' attribute does not exist")
+    return function_name
+
+def get_function_args(fragment) -> dict:
+    if not (function_args := fragment.get(ATTRIBUTE_FUNCTION_ARGS)):
+        function_args = {}
+    return function_args
+
+def get_function_kwargs(fragment) -> dict:
+    if not (function_kwargs := fragment.get(ATTRIBUTE_FUNCTION_KWARGS)):
+        function_kwargs = {}
+    return function_kwargs
+
+class FunctionCall(Fragment):
+    name:str
+    args:dict
+    kwargs:dict
+    def __init__(self, blueprint, *args, **kwargs) -> None:
+        super().__init__( blueprint, *args, **kwargs)
+        self.name = get_function_name(blueprint)
+        self.args = get_function_args(blueprint)
+        self.kwargs = get_function_kwargs(blueprint)
+    def inputs_compile(self) -> str:
+        inputs = []
+        if self.args:
+            inputs = self.args
+        
+        if self.kwargs:
+            inputs.extend([ f"{x}:{self.kwargs[x]}" for x in self.kwargs])
+
+        inputs_build = ", ".join(inputs)
+        return inputs_build
+    def compile(self)->str:
+        fragment_build = f"{self.name}({self.inputs_compile()})"
+        
+        return fragment_build
