《春晓》
唐·孟浩然
春眠不觉晓，处处闻啼鸟。夜来风雨声，花落知多少。

上一篇：python3把日期转换为星期
下一篇：python3封装文件相关操作
封装的方法可用于文件夹创建、文件夹删除、查找文件夹下所有文件。

#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Time     : 2023/12/20 9:53
# @Author   : chb
# @File     : folder_operation
import json
import os
import shutil


class FolderOperation:
    """
    文件夹创建、删除、查找文件夹下所有文件
    code码如下:
    2000:文件夹创建成功
    2001:文件夹已存在
    2003:文件夹删除成功
    2004:文件查找完成
    2005:查找文件的目录不存在
    2006:删除失败
    """

    FOLDER_CREATED = '文件夹创建成功'
    FOLDER_EXISTS = '文件夹已存在'
    FOLDER_REMOVED = '文件夹删除成功'
    FILE_SEARCH_OK = '文件查找完成'
    FOLDER_NOT_EXISTS = '查找文件的目录不存在'

    def __init__(self, folder_path: str):
        self.folder_path = folder_path

    @staticmethod
    def __return_json(code=None, data=None, msg=None):
        return_dict = {'code': str(code), 'data': data, 'msg': str(msg)}
        return json.dumps(return_dict, ensure_ascii=False)

    def creat_folder(self):
        """
        创建文件夹，不存在创建，存在提示2001
        :return: json字符串
        """
        if not os.path.exists(self.folder_path):
            os.makedirs(self.folder_path)
            return self.__return_json('2000', self.folder_path, self.FOLDER_CREATED)
        else:
            return self.__return_json('2001', self.folder_path, self.FOLDER_EXISTS)

    def del_folder(self):
        """
        删除某一目录下的所有文件和文件夹，慎重使用
        :return:只显示删除的文件夹，不显示文件夹中的其他文件
        """

        del_list = os.listdir(self.folder_path)
        for file in del_list:
            file_path = os.path.join(self.folder_path, file)

            if os.path.isfile(file_path):
                try:
                    os.remove(file_path)
                except PermissionError as err:
                    return self.__return_json('2006', file_path, err)
            elif os.path.isdir(file_path):
                shutil.rmtree(file_path)
        return self.__return_json('2003', del_list, self.FOLDER_REMOVED)

    def get_all_files(self, filename_content: str = None):
        """
        获取某个文件夹下所有的文件，包含嵌套文件夹内的文件，按照文件名称模糊查找
        :param filename_content: 文件名称包含的内容
        :return:
        """

        file_path_list = []
        if not os.path.exists(self.folder_path):
            return self.__return_json('2005', file_path_list, self.FOLDER_NOT_EXISTS)
        for root, dirs, files in os.walk(self.folder_path):
            for file_name in files:
                if not filename_content:
                    file_path = os.path.join(root, file_name)
                    file_path_list.append(file_path)
                else:
                    if filename_content in file_name:
                        file_path = os.path.join(root, file_name)
                        file_path_list.append(file_path)
        return self.__return_json('2004', file_path_list, self.FILE_SEARCH_OK)


if __name__ == '__main__':
    file_op = FolderOperation(r'F:\py_project')
    print(file_op.get_all_files())

如果感觉本文对您有帮助可以点个赞哦

本文为学习笔记，转载请标明出处

本文仅供交流学习，请勿用于非法途径

仅是个人意见，如有想法，欢迎留言