【Python】读取当前文件夹目录、子目录、子文件

需求

整理文件是程序员的一个好习惯，在整理文件的时候，目录和文件太多，但想知道目录结构是怎么样的，再来构思怎么整理文件。

环境

• Python：3.6

代码

import os

# 目录遍历方法
def checkDir(path, folder, count):
    # 检查是否是第一级目录,不是的话在目录命名前加斜杠\
    if folder != "":
        folder = folder + '\\'
    # 遍历path目录
    for i, j, k in os.walk(path):
        # 如果该目录下存在非目录文件，则先写入
        if k != []:
            for kk in k:
                print("├{} {}{}".format("─" * count, folder, kk))
                w_file.write("├{} {}{}\n".format("─" * count, folder, kk))
                w_file.flush()
        # 如果存在子目录，则进入下一个目录遍历方法
        if j != []:
            for jj in j:
                print("├{} {}{}".format("─" * count, folder, jj))
                w_file.write("├{} {}{}\n".format("─" * count, folder, jj))
                w_file.flush()
                checkDir("{}\\{}".format(i, jj), jj, count + 1)
        break


if __name__ == '__main__':
    # 新建txt，以上一级文件夹名命名
    w_file = open('{}.txt'.format(os.getcwd().split('\\')[-1]), 'w', encoding='utf-8')
    # 获取当前路径
    filePath = os.getcwd()
    # 写入当前文件路径
    w_file.write("{}".format(filePath))
    print("{}".format(filePath))
    # ”─“计数器
    count = 0
    # 调用目录遍历方法
    checkDir(filePath, "", 0)
    # 写入文件结束标志符
    w_file.write("└ #")
    print("└ #")
    # 关闭文件
    w_file.close()

示例目录：

运行结果

• 生成的txt: