《贈汪倫》
唐·李白
李白乘舟將欲行，忽聞岸上踏歌聲。桃花潭水深千尺，不及汪倫送我情。

上一篇：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'))

如果感覺本文對您有幫助可以點個贊哦

本文為學習筆記，轉載請標明出處

本文僅供交流學習，請勿用于非法途徑

僅是個人意見，如有想法，歡迎留言