1234567891011121314151617181920212223242526272829303132333435 |
- """
- Adding hooks before/after saving and deleting
- To execute arbitrary code around ``save()`` and ``delete()``, just subclass
- the methods.
- """
- from __future__ import unicode_literals
- from django.db import models
- from django.utils.encoding import python_2_unicode_compatible
- @python_2_unicode_compatible
- class Person(models.Model):
- first_name = models.CharField(max_length=20)
- last_name = models.CharField(max_length=20)
- def __init__(self, *args, **kwargs):
- super(Person, self).__init__(*args, **kwargs)
- self.data = []
- def __str__(self):
- return "%s %s" % (self.first_name, self.last_name)
- def save(self, *args, **kwargs):
- self.data.append("Before save")
- # Call the "real" save() method
- super(Person, self).save(*args, **kwargs)
- self.data.append("After save")
- def delete(self):
- self.data.append("Before deletion")
- # Call the "real" delete() method
- super(Person, self).delete()
- self.data.append("After deletion")
|