ImageField覆盖同名的镜像文件

问题描述

我有模型 UserProfile 与字段 avatar = models.ImageField（upload_to = upload_avatar）。 upload_avatar 函数名称图像文件根据 user.id （例如12.png）。但是，当用户更新化身时，新的头像名称与旧的头像名称相符，Django会为文件名添加后缀（例如，12-1.png）。有办法覆盖文件而不是创建新文件？

I have model UserProfile with field avatar = models.ImageField(upload_to=upload_avatar). upload_avatar function names image file according user.id (12.png for example). But when user updates the avatar, new avatar name coincide with old avatar name and Django adds suffix to file name (12-1.png for example). There are way to overwrite file instead of create new file?

推荐答案

是的，这已经出现了对我而言这是我所做的。

Yeah, this has come up for me, too. Here's what I've done.

型号：

from app.storage import OverwriteStorage

class Thing(models.Model):
    image = models.ImageField(max_length=SOME_CONST, storage=OverwriteStorage(), upload_to=image_path)

另外在models.py中定义：

Also defined in models.py:

def image_path(instance, filename):
    return os.path.join('some_dir', str(instance.some_identifier), 'filename.ext')

在另一个文件中，storage.py：

In a separate file, storage.py:

from django.core.files.storage import FileSystemStorage
from django.conf import settings
import os

class OverwriteStorage(FileSystemStorage):

    def get_available_name(self, name):
        """Returns a filename that's free on the target storage system, and
        available for new content to be written to.

        Found at http://djangosnippets.org/snippets/976/

        This file storage solves overwrite on upload problem. Another
        proposed solution was to override the save method on the model
        like so (from https://code.djangoproject.com/ticket/11663):

        def save(self, *args, **kwargs):
            try:
                this = MyModelName.objects.get(id=self.id)
                if this.MyImageFieldName != self.MyImageFieldName:
                    this.MyImageFieldName.delete()
            except: pass
            super(MyModelName, self).save(*args, **kwargs)
        """
        # If the filename already exists, remove it as if it was a true file system
        if self.exists(name):
            os.remove(os.path.join(settings.MEDIA_ROOT, name))
        return name

显然，这些是这里的示例值，但总体来说这对我很有用，这应该在必要时可以非常简单的进行修改。

Obviously, these are sample values here, but overall this works well for me and this should be pretty straightforward to modify as necessary.

这篇关于ImageField覆盖同名的镜像文件的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持！