bgneal@75: """
bgneal@920: Signal handlers for the forums application.
bgneal@469: 
bgneal@75: """
bgneal@75: from django.db.models.signals import post_save
bgneal@75: from django.db.models.signals import post_delete
bgneal@181: 
bgneal@348: from forums.models import Forum, Topic, Post
bgneal@920: import forums.latest
bgneal@75: 
bgneal@75: 
bgneal@75: def on_topic_save(sender, **kwargs):
bgneal@75:     if kwargs['created']:
bgneal@75:         topic = kwargs['instance']
bgneal@75:         topic.forum.topic_count_update()
bgneal@75:         topic.forum.save()
bgneal@75: 
bgneal@75: 
bgneal@75: def on_topic_delete(sender, **kwargs):
bgneal@75:     topic = kwargs['instance']
bgneal@75:     topic.forum.topic_count_update()
bgneal@75:     topic.forum.save()
bgneal@522:     forums.latest.notify_topic_delete(topic)
bgneal@75: 
bgneal@75: 
bgneal@75: def on_post_save(sender, **kwargs):
bgneal@75:     if kwargs['created']:
bgneal@75:         post = kwargs['instance']
bgneal@75: 
bgneal@75:         # update the topic
bgneal@75:         post.topic.post_count_update()
bgneal@75:         post.topic.save()
bgneal@75: 
bgneal@75:         # update the forum
bgneal@75:         post.topic.forum.post_count_update()
bgneal@75:         post.topic.forum.save()
bgneal@75: 
bgneal@75: 
bgneal@75: def on_post_delete(sender, **kwargs):
bgneal@75:     post = kwargs['instance']
bgneal@75: 
bgneal@75:     # update the topic
bgneal@348:     try:
bgneal@348:         post.topic.post_count_update()
bgneal@348:         post.topic.save()
bgneal@348:     except Topic.DoesNotExist:
bgneal@348:         pass
bgneal@348:     else:
bgneal@348:         # update the forum
bgneal@348:         try:
bgneal@348:             post.topic.forum.post_count_update()
bgneal@348:             post.topic.forum.save()
bgneal@348:         except Forum.DoesNotExist:
bgneal@348:             pass
bgneal@75: 
bgneal@75: 
bgneal@920: post_save.connect(on_topic_save, sender=Topic, dispatch_uid='forums.receivers')
bgneal@920: post_delete.connect(on_topic_delete, sender=Topic, dispatch_uid='forums.receivers')
bgneal@920: post_save.connect(on_post_save, sender=Post, dispatch_uid='forums.receivers')
bgneal@920: post_delete.connect(on_post_delete, sender=Post, dispatch_uid='forums.receivers')