《赠汪伦》
唐·李白
李白乘舟将欲行，忽闻岸上踏歌声。桃花潭水深千尺，不及汪伦送我情。

上一篇：python3封装文件夹相关操作
下一篇：python3封装图片和base64相互转换
封装的文件操作方法可用于文件创建、文件删除、获取文件信息。

#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Time     : 2024/7/12 10:41
# @Author   : chb
# @File     : file_operation
import json
import os
from datetime import datetime


class FileOperation:
    """
    文件创建、删除
    code码如下:
    2000:文件创建成功
    2001:文件已存在
    2002:文件不存在
    2003:文件删除成功
    2004:文件删除失败
    2005:文件信息获取成功
    2006:目录不存在
    2007:路径参数缺失
    """

    FILE_CREATED = '文件创建成功'
    FILE_EXISTS = '文件已存在'
    FILE_NOT_EXISTS = '文件不存在'
    FILE_REMOVED = '文件删除成功'
    FILE_INFO = '文件信息获取成功'
    DIRECTORY_NOT_EXISTS = '目录不存在'
    PATH_ERROR = '路径参数缺失'

    def __init__(self, directory=None):
        if directory:
            if not os.path.exists(directory):
                raise NotADirectoryError(self.__return_json('2006', directory, self.DIRECTORY_NOT_EXISTS))
            else:
                self.directory = directory
        self.directory = directory

    @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 create_file(self, file_name=None, full_path=None):
        """
        创建空文件，不存在创建，存在提示2001
        :param full_path: 文件绝对路径
        :param file_name: 文件名
        """
        if self.directory and file_name:
            file_path = os.path.join(self.directory, file_name)
            if not os.path.exists(file_path):
                open(file_path, 'w').close()
                return self.__return_json('2000', file_path, self.FILE_CREATED)
            else:
                return self.__return_json('2001', file_path, self.FILE_EXISTS)
        elif full_path:
            if not os.path.exists(full_path):
                open(full_path, 'w').close()
                return self.__return_json('2000', full_path, self.FILE_CREATED)
            else:
                return self.__return_json('2001', full_path, self.FILE_EXISTS)
        else:
            data = {
                '目录': self.directory, '文件': file_name, '全路径': full_path
            }
            return self.__return_json('2007', data, self.PATH_ERROR)

    def del_file(self, file_name=None, full_path=None):
        """
        删除文件，不存在提示2002。
        :param full_path: 文件绝对路径
        :param file_name: 文件名
        """
        if self.directory and file_name:
            file_path = os.path.join(self.directory, file_name)
            if not os.path.isfile(file_path):
                return self.__return_json('2002', file_path, self.FILE_NOT_EXISTS)
            else:
                try:
                    os.remove(file_path)
                except PermissionError as err:
                    return self.__return_json('2004', file_path, err)
                else:
                    self.__return_json('2003', file_path, self.FILE_REMOVED)
        elif full_path:
            if not os.path.isfile(full_path):
                return self.__return_json('2002', full_path, self.FILE_NOT_EXISTS)
            else:
                try:
                    os.remove(full_path)
                except PermissionError as err:
                    return self.__return_json('2004', full_path, err)
                else:
                    self.__return_json('2003', full_path, self.FILE_REMOVED)
        else:
            data = {
                '目录': self.directory, '文件': file_name, '全路径': full_path
            }
            return self.__return_json('2007', data, self.PATH_ERROR)

    def get_file_info(self, file_name=None, full_path=None):
        """
        获取文件信息，文件大小、文件修改时间等
        :param file_name: 文件名
        :param full_path: 文件绝对路径
        :return:
        """
        if self.directory and file_name:
            file_path = os.path.join(self.directory, file_name)
            if not os.path.isfile(file_path):
                return self.__return_json('2002', file_path, self.FILE_NOT_EXISTS)
            else:
                filename, suffix = os.path.splitext(file_name)
                file_stat = os.stat(file_path)
                file_info = {
                    '文件名': filename,
                    '文件扩展名': suffix,
                    '文件大小': f'{file_stat.st_size / 1000}kb',
                    '创建时间': datetime.fromtimestamp(file_stat.st_ctime).strftime('%Y-%m-%d %H:%M:%S'),
                    '修改时间': datetime.fromtimestamp(file_stat.st_mtime).strftime('%Y-%m-%d %H:%M:%S'),
                    '设备标识符': file_stat.st_dev,
                    'mode': oct(file_stat.st_mode & 0o777),
                    '目录': self.directory
                }
                return self.__return_json('2005', file_info, self.FILE_INFO)
        elif full_path:
            if not os.path.isfile(full_path):
                return self.__return_json('2002', full_path, self.FILE_NOT_EXISTS)
            else:
                filename = os.path.basename(full_path).split('.')[0]
                suffix = os.path.splitext(full_path)[1]
                file_stat = os.stat(full_path)
                file_info = {
                    '文件名': filename,
                    '文件扩展名': suffix,
                    '文件大小': f'{file_stat.st_size / 1000}kb',
                    '创建时间': datetime.fromtimestamp(file_stat.st_ctime).strftime('%Y-%m-%d %H:%M:%S'),
                    '修改时间': datetime.fromtimestamp(file_stat.st_mtime).strftime('%Y-%m-%d %H:%M:%S'),
                    '设备标识符': file_stat.st_dev,
                    'mode': oct(file_stat.st_mode & 0o777),
                    '目录': os.path.dirname(full_path)
                }
                return self.__return_json('2005', file_info, self.FILE_INFO)
        else:
            data = {
                '目录': self.directory, '文件': file_name, '全路径': full_path
            }
            return self.__return_json('2007', data, self.PATH_ERROR)


if __name__ == '__main__':
    file_op = FileOperation()
    print(file_op.get_file_info(full_path=r'E:\py_project\doc\11112.doc'))

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

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

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

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