from django.db import models
from django.contrib.auth.models import User
from django.utils.translation import ugettext_lazy as _
+from threadedcomments.models import ThreadedComment
# Create your models here.
class Character(models.Model):
def __str__(self):
return self.summary
+ def get_comments(self, private):
+ def quote_reply(comment):
+ return "\n".join(["> " + l for l in comment.comment.split("\n")])
+
+ # Import here to prevent dependency loop, since forms depends on
+ # models as well
+ from forms import get_influence_comment_form
+
+ if private:
+ comments = ThreadedComment.objects.get_tree(self)
+ else:
+ comments = ThreadedComment.public.get_tree(self)
+
+ # Annotate each comment with a proper reply form
+ for comment in comments:
+ initial = { 'comment' : quote_reply(comment) }
+ comment.reply_form = get_influence_comment_form(private, comment)(initial=initial)
+
+ return comments
+
class Meta:
verbose_name = _("Influence")
verbose_name_plural = _("Influences")