python生成樹目錄

【1】代碼

import sys
from pathlib import Path


class DirectionTree(object):
    """生成目錄樹
    @ pathname: 目標目錄
    @ filename: 要保存成文件的名稱
    """

    def __init__(self, pathname='.', filename='tree.txt'):
        super(DirectionTree, self).__init__()
        self.pathname = Path(pathname)
        self.filename = filename
        self.tree = ''

    def set_path(self, pathname):
        self.pathname = Path(pathname)

    def set_filename(self, filename):
        self.filename = filename

    def generate_tree(self, n=0):
        if self.pathname.is_file():
            self.tree += '    |' * n + '-' * 2 + self.pathname.name + '\n'
        elif self.pathname.is_dir():
            self.tree += '    |' * n + '-' * 2 + \
                str(self.pathname.relative_to(self.pathname.parent)) + '\\' + '\n'

            for cp in self.pathname.iterdir():
                self.pathname = Path(cp)
                self.generate_tree(n + 1)

    def save_file(self):
        with open(self.filename, 'w', encoding='utf-8') as f:
            f.write(self.tree)


if __name__ == '__main__':
    dirtree = DirectionTree()
    # 命令參數個數為1，生成當前目錄的目錄樹
    if len(sys.argv) == 1:
        dirtree.set_path(Path.cwd())
        dirtree.generate_tree()
        print(dirtree.tree)
    # 命令參數個數為2並且目錄存在存在
    elif len(sys.argv) == 2 and Path(sys.argv[1]).exists():
        dirtree.set_path(sys.argv[1])
        dirtree.generate_tree()
        print(dirtree.tree)
    # 命令參數個數為3並且目錄存在存在
    elif len(sys.argv) == 3 and Path(sys.argv[1]).exists():
        dirtree.set_path(sys.argv[1])
        dirtree.generate_tree()
        dirtree.set_filename(sys.argv[2])
        dirtree.save_file()
    else:  # 參數個數太多，無法解析
        print('命令行參數太多，請檢查！')

 

【2】測試命令

同樣可以使用以下三條命令進行測試：

• python dirtree.py ：打印當前目錄的目錄樹；

• python dirtree.py E:\Programming\Python\applications ：打印指定目錄的目錄樹；

• python dirtree.py E:\Programming\Python\applications dirtree.txt ：打印指定目錄的目錄樹並保存成文件

--選課系統\
    |--bin\
    |    |--start.py
    |--conf\
    |    |--__init__.py
    |--core\
    |    |--__init__.py
    |--data\
    |    |--__init__.py
    |--doc\
    |    |--__init__.py
    |    |--思路整理
    |    |--面向對象作業題目
    |--log\
    |    |--__init__.py
    |--modules\
    |    |--__init__.py
    |--test.py

 

【參考文檔】

參考轉自：https://blog.csdn.net/sinat_38682860/article/details/80255083