首页 > 编程 > Python > 正文

Python实现扫描指定目录下的子目录及文件的方法

2019-11-25 18:19:23
字体:
来源:转载
供稿:网友

本文介绍了使用Python来扫描指定目录下的文件,或者匹配指定后缀和前缀的函数。步骤如下:

如果要扫描指定目录下的文件,包括子目录,需要调用scan_files("/export/home/test/")

如果要扫描指定目录下的特定后缀的文件(比如jar包),包括子目录,调用scan_files("/export/home/test/", postfix=".jar")

如果要扫描指定目录下的特定前缀的文件(比如test_xxx.py),包括子目录,调用scan_files("/export/home/test/", postfix="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
发表评论 共有条评论
用户名: 密码:
验证码: 匿名发表