Python实现扫描指定目录下的子目录及文件的方法
程序员文章站
2022-04-28 10:39:52
本文介绍了使用python来扫描指定目录下的文件,或者匹配指定后缀和前缀的函数。步骤如下:
如果要扫描指定目录下的文件,包括子目录,需要调用scan_files("/ex...
本文介绍了使用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
下一篇: #leetcode刷题之路42-接雨水