From 1f8cdc0d9fb5d13b3d784282a9f6f58dc1d7056f Mon Sep 17 00:00:00 2001 From: Linus G Thiel Date: Fri, 6 May 2011 15:21:28 +0200 Subject: [PATCH] add signal for deleting votes on object deletion --- voting/signals.py | 10 ++++++++++ 1 file changed, 10 insertions(+) create mode 100644 voting/signals.py diff --git a/voting/signals.py b/voting/signals.py new file mode 100644 index 0000000..ddb4541 --- /dev/null +++ b/voting/signals.py @@ -0,0 +1,10 @@ +from django.contrib.contenttypes.models import ContentType +from models import Vote + +def delete_votes(sender, **kwargs): + """ + Delete all votes on the sender. Used in signals. + E.g. models.signals.post_delete(delete_votes, sender=YourModel) + """ + ctype = ContentType.objects.get_for_model(sender) + Vote.objects.filter(content_type=ctype, object_id=kwargs['instance'].id).delete()