Use the rewrap function for wrapping comments.

[matthijs/projects/xerxes.git] / influences / models.py

diff --git a/influences/models.py b/influences/models.py
index ed28de0c0b86ecb09dc4bf55c63e6765001e6b18..78daa15992304ee33882f4da9922ee38a930f3c7 100644 (file)
--- a/influences/models.py
+++ b/influences/models.py
@@ -1,36 +1,91 @@
 from django.db import models
+from django.core.urlresolvers import reverse
 from django.contrib.auth.models import User
+from django.utils.text import normalize_newlines
+from django.utils.translation import ugettext_lazy as _
+from threadedcomments.models import ThreadedComment
+from xerxes.tools.text import rewrap
 
 # Create your models here.
 class Character(models.Model):
-       created     = models.DateField(auto_now_add=1)
-       modified    = models.DateField(auto_now=1)
-       name        = models.CharField(max_length=255)
-       player      = models.ForeignKey(User)
+    STATUS_CHOICES = (
+        ('N', _('New')),
+        ('A', _('Approved')),
+    )
+    created     = models.DateField(auto_now_add=1, verbose_name = _("Creation time"))
+    modified    = models.DateField(auto_now=1, verbose_name = _("Modification time"))
+    name        = models.CharField(max_length=255, verbose_name = _("Name"))
+    status      = models.CharField(max_length=2, choices=STATUS_CHOICES, default='N', verbose_name = _("Status"))
+    player      = models.ForeignKey(User, verbose_name = _("Player"))
 
-       def __str__(self):
-               return self.name
+    def __str__(self):
+        return self.name
 
-       class Admin:
-               pass
+    def get_absolute_url(self):
+        return reverse('influences_influence_detail', kwargs={'object_id' : self.pk})
+
+    class Meta:
+        verbose_name = _("Character")
+        verbose_name_plural = _("Characters")
 
 class Influence(models.Model):
-       STATUS_CHOICES = (
-               ('N', 'New'),
-               ('P', 'Processing'),
-               ('D', 'Done'),
-       )
-       created     = models.DateField(auto_now_add=1)
-       modified    = models.DateField(auto_now=1)
-       
-       character   = models.ForeignKey(Character, edit_inline=models.TABULAR, num_in_admin=3, core=True)
-       contact     = models.CharField(max_length=255)
-       description = models.TextField()
-       status      = models.TextField(max_length=1, choices=STATUS_CHOICES, default='N')
-       longterm    = models.BooleanField(default=False)
-
-       def __str__(self):
-               return self.description[0:10]
-
-       class Admin:
-               pass
+    STATUS_CHOICES = (
+        ('N', _('New')),
+        ('U', _('Under discussion')),
+        ('P', _('Processing')),
+        ('D', _('Done')),
+    )
+    created     = models.DateField(auto_now_add=1, verbose_name = _("Creation time"))
+    modified    = models.DateField(auto_now=1, verbose_name = _("Modification time"))
+    
+    character   = models.ForeignKey(Character, verbose_name = _("Character"))
+    contact     = models.CharField(max_length=255, verbose_name = _("Contact Name"))
+    summary     = models.CharField(max_length=255, verbose_name = _("Summary"))
+    description = models.TextField(verbose_name = _("Description"))
+    status      = models.TextField(max_length=1, choices=STATUS_CHOICES, default='N', verbose_name = _("Status"))
+    longterm    = models.BooleanField(default=False, verbose_name = _("Long term"))
+
+    result      = models.TextField(blank=True,verbose_name = _("Result"))
+
+    def __str__(self):
+        return self.summary
+
+    def get_absolute_url(self):
+        return reverse('influences_influence_detail', kwargs={'object_id' : self.pk})
+
+    def get_comments(self, private):
+        """
+        Gets the comments that have been made on this Influence. Each
+        comment gets its reply_form attribute set to a Form appropriate
+        for replying to the comment.
+        
+        If private is True, private comments are included in this list.
+        """
+        def quote_reply(comment):
+            regex = "^([ >]*)(.*)$"
+            text = rewrap(normalize_newlines(comment.comment), 72, regex)
+            return "\n".join(["> " + l for l in text.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) }
+            prefix = "reply-to-%s" % (comment.pk)
+            FormClass = get_influence_comment_form(private, comment)
+            comment.reply_form = FormClass(initial=initial,
+                                           prefix=prefix)
+        return comments
+
+    class Meta:
+        verbose_name = _("Influence")
+        verbose_name_plural = _("Influences")
+
+# vim: set sts=4 sw=4 expandtab: