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:查找文件成功
    """

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

    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):
                os.remove(file_path)
            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):
        """
        获取某个文件夹下所有的文件，包含嵌套文件夹内的文件
        :return:
        """
        file_path_list = []
        for root, dirs, files in os.walk(self.folder_path):
            for filename in files:
                file_path = os.path.join(root, filename)
                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())

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

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

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

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