python - 从预期的文本字符串构建树

Question

我有一个代表 django 中简单树的节点的模型:

class Node(models.Model):
    name = models.CharField('Name')
    parent = models.ForeignKey('self', blank=True, null=True, related_name='children')

我有一个文档，由树结构组成，带有选项卡，如下所示:

Level1.1
    Level2.1
    Level2.2
        Level3
Level1.2

当我解析此文档时，我有以下字符串:

'Level1.1\n\tLevel2.1\n\tLevel2.2\n\t\tLevel3\nLevel1.2'

如何有效地解析该字符串并创建节点？结果应该是:

Node(name='Level1.1'), Node(name='Level2.1', parent=Node(previous node)) etc.

提前致谢!

Answer 1

使用堆栈来跟踪父节点。

document = 'Level1.1\n\tLevel2.1\n\tLevel2.2\n\t\tLevel3\nLevel1.2'

stack = []  # keep track of ancestors ([0] - root, ..., [-1] - parent)
for line in x.splitlines():
    name = line.strip()
    level = line.count('\t')
    while len(stack) > level:
        stack.pop()
    node = Node(name=name, parent=stack[-1] if stack else None)
    node.save()
    stack.append(node)