105 lines
2.8 KiB
Python
105 lines
2.8 KiB
Python
"""
|
||
@project:userbuild
|
||
@File:treeClass.py
|
||
@IDE:PyCharm
|
||
@Author:徐彬程
|
||
@Date:2023/11/27 16:00
|
||
"""
|
||
|
||
|
||
class PythonCodeTreeNode:
|
||
def __init__(self, data, parent=None, last_node=None, indentation=0, serial_number=0, extent_data=None):
|
||
if extent_data is None:
|
||
extent_data = {}
|
||
self.data = data
|
||
self.parent = parent
|
||
self.children = []
|
||
self.indentation = indentation
|
||
self.extent_data = extent_data
|
||
self.serial_number = serial_number
|
||
self.last_node = last_node
|
||
|
||
|
||
# 树结构类（非平衡非二叉树）
|
||
class PythonCodeTreeController:
|
||
|
||
# 增
|
||
@classmethod
|
||
def add_data(cls, node):
|
||
node.parent.children.append(node)
|
||
|
||
# 删
|
||
@classmethod
|
||
def del_data(cls):
|
||
pass
|
||
|
||
# 改
|
||
@classmethod
|
||
def edit_data(cls):
|
||
pass
|
||
|
||
# 查
|
||
@classmethod
|
||
def find_data(cls, root_node: PythonCodeTreeNode, i, s):
|
||
if root_node.indentation == i and root_node.serial_number == s:
|
||
return root_node
|
||
for child in root_node.children:
|
||
result = cls.find_data(child, i, s)
|
||
if result:
|
||
return result
|
||
return None
|
||
|
||
@classmethod
|
||
def find_data_by_path(cls, root_node, path, depth=0):
|
||
if root_node is None:
|
||
return None
|
||
|
||
components = path.split('/')
|
||
current_node = root_node
|
||
for component in components:
|
||
if not component: # 跳过空的组件（例如，路径开头为 '/'）
|
||
continue
|
||
|
||
found = False
|
||
for child in current_node.children:
|
||
if child.data["name"] == component:
|
||
current_node = child
|
||
found = True
|
||
break
|
||
|
||
if not found:
|
||
return None # 如果未找到路径中的某个组件，返回None
|
||
|
||
return current_node
|
||
|
||
# 输出
|
||
@classmethod
|
||
def to_array(cls, node):
|
||
result = {}
|
||
if not isinstance(node, str):
|
||
result = node.data.copy()
|
||
result["indentation"] = node.indentation
|
||
result["serial_number"] = node.serial_number
|
||
result["children"] = [cls.to_array(child) for child in node.children]
|
||
return result
|
||
|
||
# 输出
|
||
@classmethod
|
||
def to_array2(cls, node):
|
||
result = {}
|
||
if not isinstance(node, str):
|
||
result = node.data.copy()
|
||
result["indentation"] = node.indentation
|
||
result["serial_number"] = node.serial_number
|
||
return result
|
||
|
||
@classmethod
|
||
def to_string(cls, code_json):
|
||
# 输出当前节点的值
|
||
if not code_json['syntax_statement'] == 'Root':
|
||
print(" " * (code_json["syntax_indentation"] - 4) + code_json['syntax_statement'])
|
||
|
||
# 递归遍历子节点
|
||
for child in code_json['children']:
|
||
cls.to_string(child)
|