import os
from treelib import Node, Tree

def add_to_tree(tree, node_path, parent=None):
    if parent is None:
        tree.create_node(os.path.basename(node_path), node_path)
    else:
        tree.create_node(os.path.basename(node_path), node_path, parent=parent)

    if os.path.isdir(node_path):
        for filename in os.listdir(node_path):
            add_to_tree(tree, os.path.join(node_path, filename), node_path)

def print_dir_structure(startpath):
    tree = Tree()
    add_to_tree(tree, startpath)
    tree.save2file('wocao.txt')

# 使用示例，将路径替换为你要查看的目录路径，例如使用当前工作目录
print_dir_structure('.')