Only allow users to add NPC or Player characters.
[matthijs/projects/xerxes.git] / influences / views.py
index ca1a1e1e01baffe85c9d2766af6c6bc1b20affff..cf6429f0755dfa98ca21e73225089cc54fce8d88 100644 (file)
@@ -10,92 +10,44 @@ from django.core.urlresolvers import reverse
 from django.http import HttpResponseRedirect, HttpResponseForbidden
 from django.views.generic.list_detail import object_detail, object_list
 from threadedcomments.models import ThreadedComment
-from threadedcomments.forms import ThreadedCommentForm
-from threadedcomments.views import comment
+from threadedcomments.views import free_comment, _preview
 from xerxes.influences.models import Character
 from xerxes.influences.models import Influence
-from xerxes.tools.forms import ContextModelForm
-
-#
-# A few comment form classes, to handle the various cases (staff/no staff,
-# reply to public/private post)
-# 
-# It is probably possible to reduce this mess a bit using metaclasses, but I
-# didn't get this to work yet.
-# 
-class InfluenceCommentForm(ThreadedCommentForm):
-    def __init__(self, *args, **kwargs):
-        super(InfluenceCommentForm, self).__init__(*args, **kwargs)
-
-    class Meta(ThreadedCommentForm.Meta):
-        exclude = ('markup', )
-
-class AdminInfluenceCommentForm(ThreadedCommentForm):
-    def __init__(self, *args, **kwargs):
-        super(AdminInfluenceCommentForm, self).__init__(*args, **kwargs)
-
-    class Meta(ThreadedCommentForm.Meta):
-        fields = ThreadedCommentForm.Meta.fields + ('is_public',)
-
-class AdminPrivateInfluenceCommentForm(ThreadedCommentForm):
-    def __init__(self, *args, **kwargs):
-        super(AdminPrivateInfluenceCommentForm, self).__init__(*args, **kwargs)
-        self.instance.is_public = False
-
-def get_influence_comment_form(is_staff, reply_to):
-    """ Gets the form class that a user can use to reply to the given comment.
-    reply_to can be None, in which case the form class for a new comment is
-    returned. """
-    allow_markup = allow_private = is_staff
-    if reply_to:
-        allow_public = reply_to.is_public
-    else:
-        allow_public = True
-    return _get_influence_comment_form(allow_markup, allow_public, allow_private)
-    
-def _get_influence_comment_form(allow_markup, allow_public, allow_private):
-    """ Internal wrapper that selects the right form class depending on the
-    given options. Should not be called directly, call
-    get_influence_comment_form instead. """
-    if allow_markup and allow_public and allow_private:
-        return AdminInfluenceCommentForm
-    elif allow_markup and not allow_public and allow_private:
-        return AdminPrivateInfluenceCommentForm
-    elif not allow_markup and allow_public and not allow_private:
-        return InfluenceCommentForm
-    else:
-        raise Exception("Unsupported configuration")
-
-class InfluenceForm(ContextModelForm):
-    class Meta:
-        model = Influence
-        fields = ('character', 'contact', 'summary', 'description')
-
-class CharacterForm(ContextModelForm):
-    class Meta:
-        model = Character
-        fields = ('name')
+from forms import get_influence_comment_form, InfluenceForm, CharacterForm
+from xerxes.tools.misc import make_choices, filter_choices
 
 @login_required
 def add_influence(request, character_id=None):
     initial = {}
     # Get the current user's characters
-    chars = request.user.character_set.all()
+    my_chars = request.user.character_set.all().filter(type__in=[Character.PLAYER, Character.NPC])
+    # Get all chars
+    all_chars = Character.objects.all().filter(type__in=[Character.PLAYER, Character.NPC])
 
     # If a character_id was specified in the url, or there is only one
     # character, preselect it.
     if (character_id):
-        initial['character'] = character_id
-    elif (chars.count() == 1):
-        initial['character'] = chars[0].id
-
+        initial['initiator'] = character_id
+    elif (my_chars.count() == 1):
+        initial['initiator'] = my_chars[0].id
 
     f = InfluenceForm(request=request, initial=initial)
 
     # Only allow characters of the current user. Putting this here also
     # ensures that a form will not validate when any other choice was
     # selected (perhaps through URL crafting).
-    f.fields['character']._set_queryset(chars)
+    f.fields['initiator']._set_queryset(my_chars)
+    
+    # List the contacts of each of the current users characters, as well
+    # as all other (non-contact) characters as choices for the
+    # other_characters field.
+    char_choices = [
+        ("Contacts of %s" % c, make_choices(c.contacts.all()))
+        for c in my_chars
+        if c.contacts.all()
+    ]
+    char_choices.append(('All player characters', make_choices(all_chars)))
+    f.fields['other_characters'].choices = char_choices
 
     if (f.is_valid()):
         # The form was submitted, let's save it.
@@ -108,6 +60,10 @@ def add_influence(request, character_id=None):
 @login_required
 def add_character(request):
     f = CharacterForm(request=request)
+    f.fields['type'].choices = filter_choices(
+        f.fields['type'].choices,
+        [Character.PLAYER, Character.NPC]
+    )
     if (f.is_valid()):
         character = f.save(commit=False)
         character.player = request.user
@@ -120,7 +76,7 @@ def add_character(request):
 def index(request):
     # Only show this player's characters and influences
     characters = request.user.character_set.all()
-    influences = Influence.objects.filter(character__player=request.user)
+    influences = Influence.objects.filter(initiator__player=request.user)
     return render_to_response('influences/index.html', {'characters' : characters, 'influences' : influences}, RequestContext(request))
 
 #
@@ -140,57 +96,63 @@ def character_list(request):
 def character_detail(request, object_id):
     o = Character.objects.get(pk=object_id)
     # Don't show other player's characters
-    if (o.player != request.user):
+    if (not request.user.is_staff and o.player != request.user):
         return HttpResponseForbidden("Forbidden -- Trying to view somebody else's character")
     return render_to_response('influences/character_detail.html', {'object' : o}, RequestContext(request))
 
 @login_required
 def influence_list(request):
-    # Only show this player's influences
-    os = Influence.objects.filter(character__player=request.user)
-    return render_to_response('influences/influence_list.html', {'object_list' : os}, RequestContext(request))
+    # Only show the influences related to this player's characters
+    characters = request.user.character_set.all()
+    return render_to_response('influences/influence_list.html', {'characters' : characters}, RequestContext(request))
+
+def influence_comment_preview(request, context_processors, extra_context, **kwargs):
+    # Use a custom template
+    kwargs['template'] = 'influences/influence_comment_preview.html'
+    # The object to be show in the influence detail
+    extra_context['object'] = get_object_or_404(Influence, pk=kwargs['object_id'])
+    return _preview(request, context_processors, extra_context, **kwargs)
 
 @login_required
 def influence_detail(request, object_id):
 
-    def quote_reply(comment):
-        return "\n".join(["> " + l for l in comment.comment.split("\n")])
-
     o = Influence.objects.get(pk=object_id)
     # Don't show other player's influences
-    if (o.character.player != request.user):
-        return HttpResponseForbidden("Forbidden -- Trying to view influences of somebody else's character")
+    if (not request.user.is_staff and not request.user in o.related_players):
+        return HttpResponseForbidden("Forbidden -- Trying to view influences you are not involved in.")
 
     # Show all comments to staff, but only public comments to other
     # users
-    if request.user.is_staff:
-        comments = ThreadedComment.objects.get_tree(o)
-    else:
-        comments = ThreadedComment.public.get_tree(o)
-
-    # Annotate each comment with a proper reply form
-    for comment in comments:
-        initial = { 'comment' : quote_reply(comment) }
-        comment.reply_form = get_influence_comment_form(request.user.is_staff, comment)(initial=initial)
+    comments = o.get_comments(private=request.user.is_staff)
     
     context  = {
         'object' : o,
         'comments' : comments,
-        'comment_form' : get_influence_comment_form(request.user.is_staff, None)
+        'comment_form' : get_influence_comment_form(request.user.is_staff, None)()
     }
     return render_to_response('influences/influence_detail.html', context, RequestContext(request))
 
 @login_required
-def influence_comment(request, edit_id=None, *args, **kwargs):
+def influence_comment(request, object_id, parent_id=None):
+    kwargs = {}
     # Add the content_type, since we don't put in in the url explicitly
     kwargs['content_type'] = ContentType.objects.get_for_model(Influence).id
     # Find the comment to which we're replying, so we can get the right form for it.
-    if edit_id:
-        reply_to = get_object_or_404(ThreadedComment, id=edit_id)
+    if parent_id:
+        reply_to = get_object_or_404(ThreadedComment, id=parent_id)
     else:
         reply_to = None
+
     # Find the right form class
     kwargs['form_class']   = get_influence_comment_form(request.user.is_staff, reply_to)
-    return comment(request, edit_id=edit_id, *args, **kwargs)
+    # Override the model, so we don't get a free comment, but a normal
+    # one. We can't use threadedcomments' comment view for that, since
+    # that hardcodes the form_class.
+    kwargs['model'] = ThreadedComment
+    # Set a custom preview view
+    kwargs['preview'] = influence_comment_preview
+    if parent_id:
+        kwargs['prefix'] = "reply-to-%s" % (parent_id)
+    return free_comment(request, object_id=object_id, parent_id=parent_id, **kwargs)
 
 # vim: set sts=4 sw=4 expandtab: