如果特定文件不存在，则删除文件夹

Question

用于清理我的电影数据。我想删除没有扩展名为“.nfo”的文件的文件夹。

我的代码：

movies2 = (r'C:\Users\usph\AppData\Roaming\Sublime Text 3\Packages\User\Movies2')

extension = '.nfo'

for dirpath, dirs, files in os.walk(movies2):
    for i in files:
        if extension not in i:
            shutil.rmtree(dirpath)

结果并不如我所料。所有文件夹都被删除了！

请帮助更正我的代码！

Answer 1

import os, shutil

movies2 = (r'C:\Users\usph\AppData\Roaming\Sublime Text 3\Packages\User\Movies2')

for dirpath, dirs, files in os.walk(movies2):
    if [f for f in os.listdir(dirpath) if f.endswith('.nfo')] == []:
        try:
            print dirpath
            shutil.rmtree(dirpath)
        except: pass