influences/models.py

   1 from django.db import models
   2 from django.core.urlresolvers import reverse
   3 from django.contrib.auth.models import User
   4 from django.utils.translation import ugettext_lazy as _
   5 from threadedcomments.models import ThreadedComment
   6
   7 # Create your models here.
   8 class Character(models.Model):
   9     STATUS_CHOICES = (
  10         ('N', _('New')),
  11         ('A', _('Approved')),
  12     )
  13     created     = models.DateField(auto_now_add=1, verbose_name = _("Creation time"))
  14     modified    = models.DateField(auto_now=1, verbose_name = _("Modification time"))
  15     name        = models.CharField(max_length=255, verbose_name = _("Name"))
  16     status      = models.CharField(max_length=2, choices=STATUS_CHOICES, default='N', verbose_name = _("Status"))
  17     player      = models.ForeignKey(User, verbose_name = _("Player"))
  18
  19     def __str__(self):
  20         return self.name
  21
  22     def get_absolute_url(self):
  23         return reverse('influences_influence_detail', kwargs={'object_id' : self.pk})
  24
  25     class Meta:
  26         verbose_name = _("Character")
  27         verbose_name_plural = _("Characters")
  28
  29 class Influence(models.Model):
  30     STATUS_CHOICES = (
  31         ('N', _('New')),
  32         ('U', _('Under discussion')),
  33         ('P', _('Processing')),
  34         ('D', _('Done')),
  35     )
  36     created     = models.DateField(auto_now_add=1, verbose_name = _("Creation time"))
  37     modified    = models.DateField(auto_now=1, verbose_name = _("Modification time"))
  38
  39     character   = models.ForeignKey(Character, verbose_name = _("Character"))
  40     contact     = models.CharField(max_length=255, verbose_name = _("Contact Name"))
  41     summary     = models.CharField(max_length=255, verbose_name = _("Summary"))
  42     description = models.TextField(verbose_name = _("Description"))
  43     status      = models.TextField(max_length=1, choices=STATUS_CHOICES, default='N', verbose_name = _("Status"))
  44     longterm    = models.BooleanField(default=False, verbose_name = _("Long term"))
  45
  46     result      = models.TextField(blank=True,verbose_name = _("Result"))
  47
  48     def __str__(self):
  49         return self.summary
  50
  51     def get_absolute_url(self):
  52         return reverse('influences_influence_detail', kwargs={'object_id' : self.pk})
  53
  54     def get_comments(self, private):
  55         """
  56         Gets the comments that have been made on this Influence. Each
  57         comment gets its reply_form attribute set to a Form appropriate
  58         for replying to the comment.
  59
  60         If private is True, private comments are included in this list.
  61         """
  62         def quote_reply(comment):
  63             return "\n".join(["> " + l for l in comment.comment.split("\n")])
  64
  65         # Import here to prevent dependency loop, since forms depends on
  66         # models as well
  67         from forms import get_influence_comment_form
  68
  69         if private:
  70             comments = ThreadedComment.objects.get_tree(self)
  71         else:
  72             comments = ThreadedComment.public.get_tree(self)
  73
  74         # Annotate each comment with a proper reply form
  75         for comment in comments:
  76             initial = { 'comment' : quote_reply(comment) }
  77             prefix = "reply-to-%s" % (comment.pk)
  78             FormClass = get_influence_comment_form(private, comment)
  79             comment.reply_form = FormClass(initial=initial,
  80                                            prefix=prefix)
  81         return comments
  82
  83     class Meta:
  84         verbose_name = _("Influence")
  85         verbose_name_plural = _("Influences")
  86
  87 # vim: set sts=4 sw=4 expandtab: