网络编程
位置:首页>> 网络编程>> Python编程>> Python实现扫描指定目录下的子目录及文件的方法

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

作者:shichen2014  发布时间:2022-09-04 23:44:40 

标签:Python,扫描,方法

本文介绍了使用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
0
投稿

猜你喜欢

手机版 网络编程 asp之家 www.aspxhome.com