74-递归函数2：tree功能显示

Posted 2020-08-17 白杨

使用递归函数实现tree功能显示目录结构：

import os
import sys

def list_files(path):
    if os.path.isdir(path):
        print(path + ‘:‘)
        content = os.listdir(path)
        print(content)
        for fname in content:
            fname = os.path.join(path, fname)
            list_files(fname)

if __name__ == ‘__main__‘:
    list_files(sys.argv[1])   # python3 dir.py /data/weblog

结果输出：

[email protected]:~$ python dir.py /data/weblog
/data/weblog:
[‘nginx‘]
/data/weblog/nginx:
[‘my-sweetheart.cn.error.log‘, ‘www.test.com.error.log‘, ‘www.test.com.access.log‘, ‘my-sweetheart.cn.access.log‘]
[email protected]-0-2-ubuntu:~$ tree /data/weblog/
/data/weblog/
└── nginx
    ├── my-sweetheart.cn.access.log
    ├── my-sweetheart.cn.error.log
    ├── www.test.com.access.log
    └── www.test.com.error.log

1 directory, 4 files
[email protected]-0-2-ubuntu:~$