diff gpp/comments/models.py @ 1:dbd703f7d63a

Initial import of sg101 stuff from private repository.
author gremmie
date Mon, 06 Apr 2009 02:43:12 +0000 (2009-04-06)
parents
children 7b6540b185d9
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/gpp/comments/models.py	Mon Apr 06 02:43:12 2009 +0000
@@ -0,0 +1,73 @@
+"""
+Models for the comments application.
+"""
+from django.db import models
+from django.conf import settings
+from django.contrib.contenttypes.models import ContentType
+from django.contrib.contenttypes import generic
+from django.contrib.auth.models import User
+from django.template.loader import render_to_string
+
+
+COMMENT_MAX_LENGTH = getattr(settings, 'COMMENT_MAX_LENGTH', 3000)
+
+class CommentManager(models.Manager):
+    """Manager for the Comment model class."""
+
+    def for_object(self, obj, filter_public=True):
+        """QuerySet for all comments for a particular model instance."""
+        ct = ContentType.objects.get_for_model(obj)
+        qs = self.get_query_set().filter(content_type__pk=ct.id,
+                object_id=obj.id)
+        if filter_public:
+            qs = qs.filter(is_public=True)
+        return qs
+
+
+class Comment(models.Model):
+    """My own version of a Comment class that can attach comments to any other model."""
+    content_type = models.ForeignKey(ContentType)
+    object_id = models.PositiveIntegerField()
+    content_object = generic.GenericForeignKey('content_type', 'object_id')
+    user = models.ForeignKey(User)
+    comment = models.TextField(max_length=COMMENT_MAX_LENGTH)
+    html = models.TextField(blank=True)
+    creation_date = models.DateTimeField(auto_now_add=True)
+    ip_address = models.IPAddressField('IP Address')
+    is_public = models.BooleanField(default=True, 
+            help_text='Uncheck this field to make the comment invisible.')
+    is_removed = models.BooleanField(default=False,
+            help_text='Check this field to replace the comment with a ' \
+                    '"This comment has been removed" message')
+
+    # Attach manager
+    objects = CommentManager()
+
+    def __unicode__(self):
+        return u'%s: %s...' % (self.user.username, self.comment[:50])
+
+    class Meta:
+        ordering = ('creation_date', )
+
+    def save(self, force_insert=False, force_update=False):
+        html = render_to_string('comments/markdown.html', {'data': self.comment})
+        self.html = html.strip()
+        super(Comment, self).save(force_insert, force_update)
+
+
+class CommentFlag(models.Model):
+    """This model represents a user flagging a comment as inappropriate."""
+    user = models.ForeignKey(User)
+    comment = models.ForeignKey(Comment)
+    flag_date = models.DateTimeField(auto_now_add=True)
+
+    def __unicode__(self):
+        return u'Comment ID %s flagged by %s' % (self.comment_id, self.user.username)
+
+    class Meta:
+        ordering = ('flag_date', )
+
+    def get_comment_url(self):
+        return '<a href="/admin/comments/comment/%s">Comment</a>' % self.comment.id
+    get_comment_url.allow_tags = True
+