如何防止Django在同名文件已经存在的情况下更改文件名？

Question

在我的例子中，我允许用户简单地上传头像图片并使用user_id作为文件名。所以会有1.jpg，2.jpg，等等。

然而，我发现如果我为某个已经上传了头像的账号上传了一个新头像，比如说用户#10，那么这个新文件将被命名为"10_1.jpg“。这是OKay，但是我不需要它，我希望新文件可以覆盖旧文件-它也节省了一些磁盘空间。

我用谷歌搜索了一下，但没有找到任何线索。我希望有一个ImageField或FileField的选择，但它不在那里。

感谢您的帮助！

Answer 1

您应该定义自己存储，从FileSystemStorage继承它，并覆盖其中的get_available_name函数。将此存储用于您的图像场。如下所示：

class OverwriteStorage(FileSystemStorage):

    def get_available_name(self, name):
        if self.exists(name):
            os.remove(os.path.join(SOME_PATH, name))
        return name

fs = OverwriteStorage(location=SOME_PATH)

class YourModel(models.Model):
    image_file = models.ImageField(storage=fs)

Answer 2

Michael Gendin上面的解决方案在Django 2.1上工作得很好(来自2018的Hello！)。只需要在"get_available_name“方法中添加"max_length”属性：

def get_available_name(self, name, max_length=None):