add lab4 to infi=ormatics

Author: NF-coder

informatics/sem1/lab4/converters/AST/AST.py

@@ -0,0 +1,52 @@
+from .Tree.Tree import TreeAPI
+
+class AST():
+    def __init__(self, to_json: bool = True) -> None:
+        self.to_json = to_json
+
+    def main(self, file: str) -> dict:
+        file = file.replace(">", ">\n").replace("\n\n", "\n")
+        lines = file.split("\n")
+        Tree = TreeAPI("root")
+        tag_stack = [Tree]
+
+        for line in lines:
+            TAG_FLAG = False
+            OPEN_TAG, CLOSE_TAG = False, False
+            tag_name_cache = ""
+            buffer = ""
+
+            if line[:2] == "<?": continue
+            
+            line = line.replace("\t", "") # удаляем лтшние табы
+            
+            for idx, sym in enumerate(line): 
+                if TAG_FLAG:
+                    if line[idx-1] == "<": # перепроверяем начало тэга
+                        # флаги типа тэга: открывающий, закрывающий
+                        OPEN_TAG, CLOSE_TAG = sym != "/", sym == "/"
+                        # тут же добавим символ к имени открывающего тэга т.к. его захватило
+                        if OPEN_TAG: tag_name_cache += sym
+                    elif sym == ">" or sym == " ":
+                        TAG_FLAG = False # отключаем сбор информации о тэге
+                        if OPEN_TAG:
+                            tag_stack.append(
+                                tag_stack[-1].add_child(tag_name_cache)
+                            )
+                        elif CLOSE_TAG:
+                            if buffer != "": tag_stack[-1].add_child(buffer)
+                            if buffer == "" and len(tag_stack[-1].Tree.children) == 0: tag_stack[-1].add_child({})
+                            buffer = ""
+                            tag_stack = tag_stack[:-1]
+                    else: tag_name_cache += sym
+                
+                elif sym == ">": # Inner args parser
+                    for arg in buffer.split():
+                        key, value = arg.split("=")
+                        tag_stack[-1].add_child(key).add_child(value)
+                    buffer = ""
+                elif sym == "<": TAG_FLAG = True # детектируем начало тэга
+                else: buffer += sym
+
+        if self.to_json: return Tree.read_tree()["root"]
+        else: return Tree

informatics/sem1/lab4/converters/AST/AST_Regex.py

@@ -0,0 +1,37 @@
+from .Tree.Tree import TreeAPI
+import re
+
+'''
+    УПРОЩЁННАЯ ВЕРСИЯ
+    НЕ ПОДДЕРЖИИВАЮТСЯ ДАННЫЕ ВНУТРИ ТЭГА!!!
+'''
+
+class AST():
+    def __init__(self, to_json: bool = True) -> None:
+        self.R_TYPE_1 = re.compile(
+            r"(?i)((?<=<)[A-z]+)>((.|\n)*?)(</\1>)"
+        )
+        self.Tree = TreeAPI("root")
+        self.to_json = to_json
+
+    def recursive_builder(self, data: str) -> object:
+
+        array = re.findall(self.R_TYPE_1, data)
+        for item in array:
+            TAG, DATA, _, END = map(str, item)
+            self.tag_stack.append(
+                self.tag_stack[-1].add_child(TAG)
+            )
+            self.recursive_builder(DATA)
+
+        if array == [] and data != "": self.tag_stack[-1].add_child(data)
+        elif array == [] and data == "": self.tag_stack[-1].add_child({})
+
+        self.tag_stack = self.tag_stack[:-1]
+
+    def main(self, file: str) -> dict:
+        self.tag_stack = [self.Tree]
+        self.recursive_builder(file)
+
+        if self.to_json: return self.Tree.read_tree()["root"]
+        else: return self.Tree

informatics/sem1/lab4/converters/AST/Tree/Tree.py

@@ -0,0 +1,39 @@
+class Node:
+    def __init__(self, name: str):
+        self.children = []
+        self.name = name
+
+class TreeAPI:
+    def __init__(self, base: str = "") -> None:
+        self.Tree = Node(base)
+
+    def __repr__(self):
+        return f"{self.Tree.name}: {self.Tree.children}"
+
+    def read_tree(self): # + some postprocessing
+        children = self.Tree.children
+
+        if len(children) == 0: return self.Tree.name        
+        out = []
+        for child in children: out.append(child.read_tree())
+        
+        # postprocessing block (delete unused lists)
+        if len(out) == 1: out = out[0]
+        else:
+            buffer = {}
+            for elem in out:
+                for k,v in elem.items():
+                    buffer.setdefault(k, [])
+                    buffer[k].append(v)
+            
+            for elem in buffer:
+                if len(buffer[elem]) == 1: 
+                    buffer[elem] = buffer[elem][0]
+            out = buffer
+
+        return {self.Tree.name: out}
+    
+    def add_child(self, child_name: str) -> object:
+        _ = TreeAPI(child_name)
+        self.Tree.children.append(_)
+        return _

informatics/sem1/lab4/converters/AST/Tree/__pycache__/Tree.cpython-312.pyc

@@ -0,0 +1,5 @@
+#from .task1 import Converter as task1
+from .task1 import Converter as С1
+from .task2 import Converter as С2
+from .task3 import Converter as C3
+from .protobuf_converter import Converter as C4