Come impedire a Django di cambiare il nome del file quando esiste già un file con quel nome?

Question

Nel mio caso consento all'utente di caricare un'immagine avatar e utilizzare user_id come nome file, semplicemente. Quindi ci saranno 1.jpg, 2.jpg, ecc.

Tuttavia ho trovato se caricavo un nuovo avatar per un account che ne ha già uno caricato, diciamo utente # 10, il nuovo file sarà denominato come " 10_1.jpg". Va bene, comunque non ne ho bisogno e spero che il nuovo file possa sovrascrivere quello vecchio - salva comunque dello spazio su disco.

Ho cercato su Google e cercato ma non ho trovato alcun indizio. Speravo ci sarebbe stata un'opzione per ImageField o FileField ma non è lì.

Grazie per l'aiuto!

Answer 1

È necessario definire il proprio archivio, ereditarlo da FileSystemStorage e sostituire la funzione get_available_name al suo interno. Usa questa memoria per il tuo imagefield. Qualcosa di simile a questo:

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)