python 用于搜索特定文件类型的目录的脚本

Question

下面脚本中的大家好，我首先尝试列出主目录中的每个文件和目录，然后检查是否有任何具有特定扩展名的文件，例如（.py、.mkv）。

它适用于位于主目录中的文件，但是当我想让它检查其他目录以查看那里是否有任何文件时，它无法工作。

这是我的代码：

import os

class Sorter(object):
    path = os.environ['HOME']
    all_dirs = list()
    all_items = list()
    address = None
    movies = list()


    def __init__(self):
        pass

    def list_directories(self):
        dirs = os.listdir(self.path)
        for d in dirs:
            if os.path.isdir(os.path.join(self.path,d)):
                self.all_dirs.append(d)

            elif os.path.isfile(os.path.join(self.path,d)):
                self.all_items.append(d)

    def find_movies(self):

        for item in self.all_items:
            if os.path.splitext(os.path.join(self.path,item))[1] in ['.mp3','.mkv']:
                self.movies.append(item)
        for directory in self.all_dirs:
            try:
                os.chdir(os.path.join(self.path,directory))
                for i in directory:
                    if os.path.splitext(os.path.join(self.path,item))[1] in ['.mp3','.mkv']:
                        self.movies.append(item)
                os.chdir(self.path)
            except:
                pass

Answer 1

这似乎过于复杂了。请参阅使用 os.walk 和列表理解来过滤文件的方法：

import os

创建文件：

dirs = [r"./subdir",r"./subdir/tata",r"subdir/tarumpa",r"./dir2b"]
files = ["k.mp4","some.txt","cool.mp3"]

def touch(p,fn):
    with open(os.path.join(p,fn),"w") as f:
        f.write(" ")

for d in dirs:
    os.mkdir(d)
    for f in files:
        touch(d,f)

查找文件：

movie = []        
music = []        

# os.walk recurses into subdirectories, it returns a generator for each directory
# including the directory its in as root, all subdirs in dirs and all files in 
# files: then it steps into each of the dirs and does the same ...
for root,dirs,files in os.walk("./"):
    # root is the dir we are currently in, f the filename that ends on ...
    movie.extend( (os.path.join(root,f) for f in files if f.endswith(".mp4")) )
    music.extend( (os.path.join(root,f) for f in files if f.endswith(".mp3")) )

print(movie)
print(music)

输出：

# movies
['./subdir/k.mp4', './subdir/tarumpa/k.mp4', './subdir/tata/k.mp4', './dir2b/k.mp4']

# music
['./subdir/cool.mp3', './subdir/tarumpa/cool.mp3', 
 './subdir/tata/cool.mp3', './dir2b/cool.mp3']

Answer 2

您可以使用标准库 pathlib 模块和 glob 按文件扩展名搜索文件。

glob 方言不如 bash 强大，但您可以使用 ** 进行递归子目录匹配。您不能使用 bash 样式的大括号扩展 *.{mp3,mkv}。相反，您可以链接多个 glob 搜索的结果。

from pathlib import Path

def find_files(root, extensions):
    for ext in extensions:
        yield from Path(root).glob(f'**/*.{ext}')

for movie in find_files(Path.home() / 'Videos', ['mp4', 'mkv', 'avi']):
    print(movie)

请注意，Path 对象重载了 / 运算符，因此 Path.home() / 'Videos' 将生成表示 /home/username/Videos/

的路径对象

python 用于搜索特定文件类型的目录的脚本

python script to search directories for specific file type

python

python-3.x

os.path

创建文件：

查找文件：

输出：