代码实现文件夹自动生成目录
时间: 2023-08-31 21:05:04 浏览: 39
可以使用Python中的`os`和`re`模块来实现文件夹自动生成目录。具体步骤如下:
1. 遍历文件夹中的所有文件,并将它们的路径保存到一个列表中。
2. 对这个列表进行排序,以保证生成的目录按照文件名的字母顺序排列。
3. 对每个文件名进行正则表达式匹配,提取出它们的目录结构。
4. 根据目录结构生成目录字符串。
5. 将目录字符串写入一个新的文件中。
下面是一份示例代码:
```python
import os
import re
def generate_toc(root_dir):
# 遍历所有文件
file_list = []
for root, dirs, files in os.walk(root_dir):
for file in files:
file_list.append(os.path.join(root, file))
# 对文件列表进行排序
file_list.sort()
# 生成目录字符串
toc = ""
for file in file_list:
# 提取目录结构
pattern = re.compile(root_dir + r"/(.*)")
match = pattern.match(file)
if match:
file_path = match.group(1)
file_path_list = file_path.split("/")
file_path_list = [x for x in file_path_list if x != ""]
depth = len(file_path_list)
toc += " " * depth
toc += "- " + file_path_list[-1] + "\n"
# 将目录字符串写入新文件
with open("toc.txt", "w") as f:
f.write(toc)
if __name__ == "__main__":
generate_toc("my_folder")
```
注意,这份代码假设所有文件都在一个名为`my_folder`的文件夹中。如果你的情况不同,需要相应地修改代码。