Python实现读取当前文件夹目录,子目录,子文件
作者：林九生
这篇文章主要为大家详细介绍了如何使用Python的os模块遍历目录结构,将文件和子目录以树形结构写入到txt文件中,帮助程序员理解并整理复杂的文件系统,感兴趣的小伙伴可以了解下
需求
整理文件是程序员的一个好习惯，在整理文件的时候，目录和文件太多，但想知道目录结构是怎么样的，再来构思怎么整理文件。
环境
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:
到此这篇关于Python实现读取当前文件夹目录,子目录,子文件的文章就介绍到这了,更多相关Python读取文件目录内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！