ChatGPT编程—实现小工具软件(文件查找和筛选)
今天借助[小蜜蜂AI]网站的ChatGPT编程实现一个功能:根据特定需求结合通配符和其他条件来进行文件查找和筛选。在这个例子中,我们将创建一个函数find_files,它接受用户输入的目录路径、文件扩展名以及其他条件作为参数,并返回符合条件的文件列表。
环境:Pycharm 2021
系统:Mac OS
向ChatGPT输入如下内容:
ChatGPT收到后,返回如下结果:
创建自定义函数进行文件查找和筛选:
import os
import fnmatch
def find_files(directory, pattern='*', keyword=None, min_size=None, max_size=None):
matched_files = []
for root, dirs, files in os.walk(directory):
for file_name in files:
if fnmatch.fnmatch(file_name, pattern):
file_path = os.path.join(root, file_name)
file_size = os.path.getsize(file_path)
# 根据关键字筛选文件
if keyword and keyword not in file_name:
continue
# 根据文件大小筛选文件
if min_size and file_size < min_size:
continue
if max_size and file_size > max_size:
continue
matched_files.append(file_path)
return matched_files
# 示例用法:查找目录下所有以 .txt 结尾并包含关键字 'example' 的文件
found_files = find_files('path/to/directory', pattern='*.txt', keyword='example')
for file_path in found_files:
print(file_path)
在上述代码中,find_files函数允许您基于目录路径、文件扩展名、关键字、最小/最大文件大小等条件来定制文件查找和筛选操作。您可以根据实际需求扩展或修改该函数,以满足您对文件查找和筛选的特定需求。
根据自己的需求,对上面的代码进行修改,如下:
import os
import fnmatch
def find_files(directory, pattern='*', keyword=None, min_size=None, max_size=None):
matched_files = []
for root, dirs, files in os.walk(directory):
for file_name in files:
if fnmatch.fnmatch(file_name, pattern):
file_path = os.path.join(root, file_name)
file_size = os.path.getsize(file_path)
# 根据关键字筛选文件
if keyword and keyword not in file_name:
continue
# 根据文件大小筛选文件
if min_size and file_size < min_size:
continue
if max_size and file_size > max_size:
continue
matched_files.append(file_path)
return matched_files
# 定义变量,输入自定义的需求
file_path = input('请输入路径:')
file_type = input('请输入文件类型:')
keyword = input('请输入包含的关键字:')
# 示例用法:查找目录下所有以 .txt 结尾并包含关键字 'example' 的文件
found_files = find_files(file_path, pattern=file_type, keyword=keyword)
for file_path in found_files:
print(file_path)
这个案例说明了,AI编程太高效了。用好AI提高效率,提升自我。