ChatGPT编程—实现小工具软件(文件查找和筛选)

今天借助[小蜜蜂AI][https://zglg.work]网站的ChatGPT编程实现一个功能：根据特定需求结合通配符和其他条件来进行文件查找和筛选。在这个例子中，我们将创建一个函数find_files，它接受用户输入的目录路径、文件扩展名以及其他条件作为参数，并返回符合条件的文件列表。

环境：Pycharm 2021

系统：Mac OS

向ChatGPT输入如下内容：

Screenshot 2024-03-16 at 11.33.51

ChatGPT收到后，返回如下结果：

创建自定义函数进行文件查找和筛选：

import os
import fnmatchdef 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:continueif max_size and file_size > max_size:continuematched_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 fnmatchdef 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:continueif max_size and file_size > max_size:continuematched_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)