influences/forms.py

   1 from django.forms.fields import CharField, BooleanField
   2 from django.forms.widgets import Textarea
   3 from threadedcomments.forms import ThreadedCommentForm
   4 from xerxes.tools.forms import ContextModelForm
   5 from models import Influence, Character
   6
   7 #
   8 # A few comment form classes, to handle the various cases (staff/no staff,
   9 # reply to public/private post)
  10 #
  11 # It is probably possible to reduce this mess a bit using metaclasses, but I
  12 # didn't get this to work yet.
  13 #
  14 class InfluenceCommentForm(ThreadedCommentForm):
  15     # Force the textare to 80 columns. This is really a hack, we should
  16     # rather create a template tag to do this at the template level.
  17     comment = CharField(widget=Textarea(attrs={'cols' : 80}))
  18     def __init__(self, *args, **kwargs):
  19         super(InfluenceCommentForm, self).__init__(*args, **kwargs)
  20
  21     class Meta(ThreadedCommentForm.Meta):
  22         exclude = ('markup', )
  23
  24 class AdminInfluenceCommentForm(ThreadedCommentForm):
  25     comment = CharField(widget=Textarea(attrs={'cols' : 80}))
  26     is_public = BooleanField(required=False, initial=False)
  27     def __init__(self, *args, **kwargs):
  28         super(AdminInfluenceCommentForm, self).__init__(*args, **kwargs)
  29
  30     class Meta(ThreadedCommentForm.Meta):
  31         fields = ThreadedCommentForm.Meta.fields + ('is_public',)
  32         exclude = ('markup', )
  33
  34 class AdminPrivateInfluenceCommentForm(ThreadedCommentForm):
  35     comment = CharField(widget=Textarea(attrs={'cols' : 80}))
  36     def __init__(self, *args, **kwargs):
  37         super(AdminPrivateInfluenceCommentForm, self).__init__(*args, **kwargs)
  38         self.instance.is_public = False
  39
  40     class Meta(ThreadedCommentForm.Meta):
  41         exclude = ('markup', )
  42
  43 def get_influence_comment_form(is_staff, reply_to):
  44     """ Gets the form class that a user can use to reply to the given comment.
  45     reply_to can be None, in which case the form class for a new comment is
  46     returned. """
  47     allow_markup = False
  48     allow_private = is_staff
  49     if reply_to:
  50         allow_public = reply_to.is_public
  51     else:
  52         allow_public = True
  53     return _get_influence_comment_form(allow_markup, allow_public, allow_private)
  54
  55 def _get_influence_comment_form(allow_markup, allow_public, allow_private):
  56     """ Internal wrapper that selects the right form class depending on the
  57     given options. Should not be called directly, call
  58     get_influence_comment_form instead. """
  59     if not allow_markup and allow_public and allow_private:
  60         return AdminInfluenceCommentForm
  61     elif not allow_markup and not allow_public and allow_private:
  62         return AdminPrivateInfluenceCommentForm
  63     elif not allow_markup and allow_public and not allow_private:
  64         return InfluenceCommentForm
  65     else:
  66         raise Exception("Unsupported configuration")
  67
  68 class InfluenceForm(ContextModelForm):
  69     class Meta:
  70         model = Influence
  71         fields = ('character', 'contact', 'summary', 'description')
  72
  73 class CharacterForm(ContextModelForm):
  74     class Meta:
  75         model = Character
  76         fields = ('name')
  77