app/django/contrib/comments/managers.py
author Sverre Rabbelier <srabbelier@gmail.com>
Mon, 06 Apr 2009 22:53:47 +0000
changeset 2101 d6250eac3ab0
parent 323 ff1a9aa48cfd
permissions -rw-r--r--
Fixed bug introduced by fixing a pylint error Here first was very much intentionally set to None, as later on we check if it is none in an 'first if first else exception' statement. Patch by: Sverre Rabbelier

from django.db import models
from django.dispatch import dispatcher
from django.contrib.contenttypes.models import ContentType
from django.utils.encoding import force_unicode

class CommentManager(models.Manager):

    def in_moderation(self):
        """
        QuerySet for all comments currently in the moderation queue.
        """
        return self.get_query_set().filter(is_public=False, is_removed=False)

    def for_model(self, model):
        """
        QuerySet for all comments for a particular model (either an instance or
        a class).
        """
        ct = ContentType.objects.get_for_model(model)
        qs = self.get_query_set().filter(content_type=ct)
        if isinstance(model, models.Model):
            qs = qs.filter(object_pk=force_unicode(model._get_pk_val()))
        return qs