python - os walk 改变了目录结构

Question

我目前想改进我曾经写过的一个 python 脚本。

其中的结构大致是这样的：

def x(args):
    for root, dirs, files in os.walk(args):
        do_something_that_changes_the_directory_structure()

    for root, dirs, files in os.walk(args):
        do_someting_with_that_changed_directory()

问题是我更改了目录结构并且必须重新读取该结构才能对其进行处理。是否可以只用一个 for 循环来做到这一点？我在考虑while循环，但在考虑之后我认为这行不通！

Answer 1

您应该更新dirs并files根据您对文件夹所做的更改。例如：

for root, dirs, files in os.walk(args):
    with file('{}/test.file'.format(root)) as f:
        f.write('test')

    files.append('test')