在目录和子目录中查找具有扩展名的文件?

时间:2012-04-22 23:40:24

标签: python

我想创建一个Python脚本,以便根据扩展名将我桌面上的文件快速整理到文件夹中。基本上,我如何使用循环来获取文件,对其执行某些操作,继续下一个文件,等等?

2 个答案:

答案 0 :(得分:4)

您需要的所有内容可能都包含在os库中,更具体地说,位于os.path位和shutil库中。

要探索您可以使用os.walk的目录树并移动文件,可以使用shutil.move


编辑:我一起入侵的小脚本让你走了:

import os
import shutil as sh
from collections import defaultdict

DESKTOP = '/home/mac/Desktop'

#This dictionary will contain: <extension>: <list_of_files> mappings
register = defaultdict(list)

#Populate the register
for dir_, dirs, fnames in os.walk('/home/mac/Desktop'):
    for fname in fnames:
        register[fname.rsplit('.', 1)[1]].append(os.path.join(dir_, fname))

#Iterate over the register, creating the directory and moving the files 
#with that extension in it.
for dirname, files in register.iteritems():
    dirname = os.path.join(DESKTOP, dirname)
    if not os.path.exists(dirname):
        os.makedirs(dirname)
    for file_ in files:
        sh.move(file_, dirname)

答案 1 :(得分:0)

我建议使用文件名oslist模块中的os.walk