python扫描指定目录中子目录与文件的例子

发布时间:2020-09-11编辑:脚本学堂
本文介绍了Python扫描指定目录下的子目录及文件的方法,python目录与文件扫描实例。

使用python来扫描指定目录下的文件,或匹配指定后缀和前缀的函数。

具体步骤:
1,扫描指定目录下的文件,包括子目录,需要调用scan_files("/export/home/test/")
2,扫描指定目录下的特定后缀的文件(比如jar包),包括子目录,调用scan_files("/export/home/test/", postfix=".jar")
3,扫描指定目录下的特定前缀的文件(比如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