Как запретить Django изменять имя файла, когда файл с таким именем уже существует?

Question

В моем случае я разрешаю пользователю загружать изображение аватара и использовать user_id как имя файла, просто. Таким образом, будет 1.jpg, 2.jpg и т. Д.

Однако я обнаружил, что если я загружу новый аватар для некоторой учетной записи, у которой уже есть один загруженный, скажем, пользователь # 10, новый файл будет называться как " 10_1.jpg». Это нормально, но мне это не нужно, и я надеюсь, что новый файл может перезаписать старый - он также сохранит некоторое дисковое пространство.

Я искал Google и искал, но не мог найти подсказки. Я надеялся, что будет вариант для 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)