django之重寫FileField字段的保存

wangzy2019 發(fā)布于2019-07-30 14:55 / 2136人閱讀

摘要：重寫字段的保存，以重命名為例添加配置中末尾添加配置如下，指定指向的操作文件中的類文件上傳重寫添加應(yīng)用下添加包，注意有文件，下創(chuàng)建文件，文件內(nèi)容為初始化重寫方法文件擴(kuò)展名文件目錄定義文件名，源文件名，避開系統(tǒng)定義的隨機(jī)字符串追加，所

重寫FileField字段的保存，以重命名為例

settings.py添加配置

settings.py中末尾添加配置如下，指定指向的操作文件中的類：

# 文件上傳重寫
DEFAULT_FILE_STORAGE = "app.customfilefield.storage.FileStorage"

添加FileStorage

app應(yīng)用下添加python包customfilefield，注意有init.py文件，customfilefield下創(chuàng)建py文件storage.py，文件內(nèi)容為：

storage.py

# -*-coding:utf-8 -*-
from django.core.files.storage import FileSystemStorage
from django.http import HttpResponse
from django.conf import settings
import os, time, random
from app import utils

class FileStorage(FileSystemStorage):
    def __init__(self, location=settings.MEDIA_ROOT, base_url=settings.MEDIA_URL):
        #初始化
        super(FileStorage, self).__init__(location, base_url)

    #重寫 _save方法
    def _save(self, name, content):
        #文件擴(kuò)展名
        ext = os.path.splitext(name)[1]
        #文件目錄
        d = os.path.dirname(name)
        # 定義文件名，源文件名，避開系統(tǒng)定義的隨機(jī)字符串追加，所以避開不用name字段
        end = utils.find_last(str(content), ".")
        filename = ""
        if end != -1:
            filename = str(content)[:end]
        # 定義文件名，年月日時分秒隨機(jī)數(shù)
        fn = time.strftime("%Y%m%d%H%M%S")
        fn = fn + "_%d" % random.randint(0,100)
        #重寫合成文件名
        name = os.path.join(d, filename + fn + ext)
        #調(diào)用父類方法
        return super(FileStorage, self)._save(name, content)

utils.py

# 獲取字符串中指定字符最后一次出現(xiàn)的位置
def find_last(string,str):
    last_position=-1
    while True:
        position=string.find(str,last_position+1)
        if position==-1:
            return last_position
        last_position=position

如此，最后上傳的文件名為原文件名加上年月日時分秒加上0-100的隨機(jī)數(shù)保存，效果如下：