[Django]-How do I override delete() on a model and have it still work with related deletes

47πŸ‘

βœ…

I figured it out. I just put this on that Widget model:

def delete(self):
    files = WidgetFile.objects.filter(widget=self)
    if files:
        for file in files:
            file.delete()
    super(Widget, self).delete()

This triggered the necessary delete() method on each of the related objects, thus triggering my custom file deleting code. It’s more database expensive yes, but when you’re trying to delete files on a hard drive anyway, it’s not such a big expense to hit the db a few extra times.

πŸ‘€orokusaki

81πŸ‘

I’m doing the same thing and noticed a nugget in the Django docs that you should think about.

Overriding predefined model methods

Overriding Delete
Note that the delete() method for an object is not necessarily called when deleting objects in bulk using a QuerySet. To ensure customized delete logic gets executed, you can use pre_delete and/or post_delete signals.

This means your snippet will not always do what you want. Using Signals is a better option for dealing with deletions.

I went with the following:

import shutil
from django.db.models.signals import pre_delete 
from django.dispatch import receiver

@receiver(pre_delete)
def delete_repo(sender, instance, **kwargs):
    if sender == Set:
        shutil.rmtree(instance.repo)
πŸ‘€ElementalVoid

4πŸ‘

Using clear() prior to deleting, removes all objects from the related object set.

see django-following-relationships-backward

example:

group.link_set.clear() 
group.delete() 
πŸ‘€panchicore

1πŸ‘

This seems only be sense-full if one Widget is connected to one WidgetFile exactly. In that case you should use a OneToOneField

from On-to-one examples:

# Delete the restaurant; the waiter should also be removed
>>> r = Restaurant.objects.get(pk=1)
>>> r.delete()

1πŸ‘

Just to throw in a possible way around this problem: pre-delete signal. (Not in any way implying there’s no actual solution.)

πŸ‘€che

1πŸ‘

It should look like described on the django site:

class Blog(models.Model):
    name = models.CharField(max_length=100)
    tagline = models.TextField()
    def save(self, *args, **kwargs):
        do_something()
        super(Blog, self).save(*args, **kwargs) # Call the "real" save() method.
        do_something_else()

http://docs.djangoproject.com/en/dev/topics/db/models/#overriding-predefined-model-methods

you forgot to pass some arguments

0πŸ‘

Is some_widget_instance and instance of Widget or of WidgetFile ? Because if it is an instance of Widget it won’t get your custom delete() function, which is in the WidgetFile class.

πŸ‘€thornomad

0πŸ‘

I refactored this answer from the top answer above

import shutil
from django.db.models.signals import pre_delete 
from django.dispatch import receiver


"""WidgetFile is the model for the widget files.
pre_delete means delete run this function before the object is deleted from database"""
@receiver(pre_delete, sender=WidgetFile)
def delete_preo(sender, instance, **kwargs):
    shutil.rmtree(instance.repo)

Override model methods
Django signals

πŸ‘€NetrobeWebby

-2πŸ‘

From Django 1.9, if You would just define on_delete=models.CASCADE for field, it will remove all related objects on delete.

πŸ‘€CLTanuki

Leave a comment