influences/models.py

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