diff options
Diffstat (limited to 'tests/modeltests/save_delete_hooks')
| -rw-r--r-- | tests/modeltests/save_delete_hooks/__init__.py | 0 | ||||
| -rw-r--r-- | tests/modeltests/save_delete_hooks/models.py | 42 |
2 files changed, 42 insertions, 0 deletions
diff --git a/tests/modeltests/save_delete_hooks/__init__.py b/tests/modeltests/save_delete_hooks/__init__.py new file mode 100644 index 0000000000..e69de29bb2 --- /dev/null +++ b/tests/modeltests/save_delete_hooks/__init__.py diff --git a/tests/modeltests/save_delete_hooks/models.py b/tests/modeltests/save_delete_hooks/models.py new file mode 100644 index 0000000000..47748082da --- /dev/null +++ b/tests/modeltests/save_delete_hooks/models.py @@ -0,0 +1,42 @@ +""" +13. Adding hooks before/after saving and deleting + +To execute arbitrary code around ``save()`` and ``delete()``, just subclass +the methods. +""" + +from django.db import models + +class Person(models.Model): + first_name = models.CharField(maxlength=20) + last_name = models.CharField(maxlength=20) + + def __repr__(self): + return "%s %s" % (self.first_name, self.last_name) + + def save(self): + print "Before save" + super(Person, self).save() # Call the "real" save() method + print "After save" + + def delete(self): + print "Before deletion" + super(Person, self).delete() # Call the "real" delete() method + print "After deletion" + +API_TESTS = """ +>>> p1 = Person(first_name='John', last_name='Smith') +>>> p1.save() +Before save +After save + +>>> Person.objects.all() +[John Smith] + +>>> p1.delete() +Before deletion +After deletion + +>>> Person.objects.all() +[] +""" |
