主页 / 问题 / 1065720
Accepted
Keymaster
Asked:2020-01-04 08:21:30 +0000 UTC2020-01-04 08:21:30 +0000 UTC

目录树绘制算法

  • 772

我试图在 python 中编写 DOS 树实用程序的类似物。这是绘制目录树的那个。像这样：

在此处输入图像描述

我所能达到的就是：

在此处输入图像描述

如您所见，一些标记为红色的路径我不知道如何绘制。或者更确切地说，有一些想法，甚至有些事情已经解决了，但是剧本里有这么多垃圾，我自己几乎无法理解它是如何运作的。在我看来，我最初错误地接近了问题的解决方案。因此，我请求聪明人的帮助。我知道 os.system('tree')。但意义不在于结果，而在于过程。

import os


def tree(my_path='.'):

    print(my_path)
    for path, dirs, files in os.walk(my_path):
        deep = path.replace(my_path, '').count(os.sep)
        space = '   ' * (deep - 1)
        if deep == 1:
            print('{}|__{}'.format(space, os.path.basename(path)))
        elif deep > 1:
            print('|{}|__{}'.format(space, os.path.basename(path)))


tree()
python

1 个回答

  1. Best Answer
    from pathlib import Path

fork_string   = '├──'
corner_string = '└──'
wall_string   = '│  '
space_string  = '   '

def drawtree(path, seen, head='', tail=''):
    if path.is_dir() and path.resolve() not in seen:
        yield head + path.name
        seen.add(path.resolve()) # на случай зацикленных ссылок
        entries = sorted(filter(Path.is_dir, path.iterdir()))

        for i, entry in enumerate(entries):
            if i < len(entries) - 1:
                yield from drawtree(entry, seen, tail + fork_string, tail + wall_string)
            else:
                yield from drawtree(entry, seen, tail + corner_string, tail + space_string)

def tree(my_path='.'):
    for line in drawtree(Path(my_path), set()):
        print(line)

tree('root')

    root
├──Folder
│  ├──a
│  │  ├──aa
│  │  ├──ab
│  │  │  └──aab
│  │  └──ac
│  ├──b
│  └──c
│     └──cc
│        ├──ccc
│        └──cce
└──something

    更少的功能，没有检查，但更简洁和描述性

    from pathlib import Path

def tree(path='.', head='', tail=''):
    path = Path(path)

    if path.is_dir():
        print(head + path.name)
        entries = sorted(filter(Path.is_dir, path.iterdir()))

        for i, entry in enumerate(entries):
            if i < len(entries) - 1:
                tree(entry, tail + '├──', tail + '│  ')
            else:
                tree(entry, tail + '└──', tail + '   ')

tree('root')
    • 15

相关问题