Làm thế nào để ngăn chặn Django thay đổi tên tệp khi một tệp có tên đó đã tồn tại?

Question

Trong trường hợp của tôi, tôi cho phép người dùng tải lên hình đại diện và chỉ sử dụng user_id làm tên tệp. Vì vậy, sẽ có 1.jpg, 2.jpg, v.v.

Tuy nhiên tôi thấy nếu tôi tải lên hình đại diện mới cho một số tài khoản đã tải lên, giả sử người dùng # 10, tệp mới sẽ được đặt tên là " 10_1.jpg ". Đây là OKay, tuy nhiên tôi không cần nó và tôi hy vọng tập tin mới có thể ghi đè lên một cũ - nó cũng tiết kiệm một số không gian đĩa anyway.

Tôi đã tìm kiếm và tìm kiếm nhưng không thể tìm thấy manh mối. Tôi đã hy vọng sẽ có một lựa chọn cho ImageField hoặc FileField nhưng nó không có ở đó.

Cảm ơn bạn đã trợ giúp!

Answer 1

Bạn nên xác định dung lượng lưu trữ của chính mình, kế thừa từ FileSystemStorage và ghi đè hàm get_available_name trong đó. Sử dụng bộ nhớ này cho lĩnh vực hình ảnh của bạn. Something như thế này:

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)