Python扫描指定目录下(包括子目录)的文件

最新推荐文章于 2024-05-10 17:18:48 发布

原创最新推荐文章于 2024-05-10 17:18:48 发布 · 6k 阅读

4 ·

CC 4.0 BY-SA版权

文章标签：

#jython #python #自动化测试

python 同时被 2 个专栏收录

53 篇文章

订阅专栏

jython

5 篇文章

订阅专栏

本文介绍了一个实用的Python函数，用于扫描指定目录及其子目录下的文件。可根据后缀名或文件名前缀筛选文件。

扫描指定目录下的文件，或者匹配指定后缀和前缀的函数。

如果要扫描指定目录下的文件，包括子目录，调用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