X-Git-Url: https://git.stderr.nl/gitweb?a=blobdiff_plain;f=influences%2Fviews.py;h=e1ec516a5ca30f024aae289ac8dd4c92dd54356e;hb=06863704f30a2025ee82fd89a41716f5ab2cce87;hp=ea07c70c03f86c668562ced31549d3673d8bd72c;hpb=6e0ad555b25c2d9f3f28726f797edb3a51783af2;p=matthijs%2Fprojects%2Fxerxes.git

diff --git a/influences/views.py b/influences/views.py
index ea07c70..e1ec516 100644
--- a/influences/views.py
+++ b/influences/views.py
@@ -1,29 +1,19 @@
+from django import forms
 from django.contrib.auth.decorators import login_required
 from django.shortcuts import render_to_response
 from django.shortcuts import get_object_or_404
 from django.template import RequestContext
 from django.utils.translation import ugettext as _
 from django.contrib.auth.models import User
+from django.contrib.contenttypes.models import ContentType
 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 free_comment
 from xerxes.influences.models import Character
 from xerxes.influences.models import Influence
-from xerxes.tools.forms import ContextModelForm
-
-
-
-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
 
 @login_required
 def add_influence(request, character_id=None):
@@ -101,15 +91,38 @@ def influence_list(request):
 
 @login_required
 def influence_detail(request, object_id):
+
     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")
+
+    # Show all comments to staff, but only public comments to other
+    # users
+    comments = o.get_comments(private=request.user.is_staff)
+    
     context  = {
         'object' : o,
-        'comments' : ThreadedComment.public.get_tree(o),
-        'comment_form' : ThreadedCommentForm(),
+        'comments' : comments,
+        '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):
+    # 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)
+    else:
+        reply_to = None
+    # Find the right form class
+    kwargs['form_class']   = get_influence_comment_form(request.user.is_staff, reply_to)
+    # 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
+    return free_comment(request, edit_id=edit_id, *args, **kwargs)
+
 # vim: set sts=4 sw=4 expandtab: