【Python】掃描指定文件夾下特定后綴的文件

原文地址：https://www.cnblogs.com/cynchanpin/p/6781020.html

 

掃描指定文件夾下的文件。或者匹配指定后綴和前綴的函數。

 

假設要掃描指定文件夾下的文件，包含子文件夾，調用scan_files("/export/home/test/")

假設要掃描指定文件夾下的特定后綴的文件（比方jar包），包含子文件夾，調用scan_files("/export/home/test/", postfix=".jar")

假設要掃描指定文件夾下的特定前綴的文件（比方test_xxx.py）。包含子文件夾，調用scan_files("/export/home/test/", prefix="test_")

 

#!/usr/bin/env python
#coding=utf-8

import os

def scan_files(directory,prefix=None,postfix=None):
    files_list=[]
    
    for root, sub_dirs, files in os.walk(directory):
        for special_file in files:
            if postfix:
                if special_file.endswith(postfix):
                    files_list.append(os.path.join(root,special_file))
            elif prefix:
                if special_file.startswith(prefix):
                    files_list.append(os.path.join(root,special_file))
            else:
                files_list.append(os.path.join(root,special_file))
                          
    return files_list