X-Git-Url: https://git.stderr.nl/gitweb?a=blobdiff_plain;f=influences%2Fviews.py;h=35ee02ef921227f8faf626fb355ad031c8422440;hb=6ddbd39b9bfb49ffeb8e444b3cbe23a378cd36f2;hp=84719024b46dfc642580a4719461936253370aa4;hpb=d97a17fc5f4d7f1f7e70220171352224f64b2cf6;p=matthijs%2Fprojects%2Fxerxes.git diff --git a/influences/views.py b/influences/views.py index 8471902..35ee02e 100644 --- a/influences/views.py +++ b/influences/views.py @@ -1,47 +1,141 @@ +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 ee.influences.models import Character -from ee.influences.models import Influence -from ee.tools.forms import ContextModelForm - -class InfluenceForm(ContextModelForm): - class Meta: - model = Influence - fields = ('character', 'contact', 'description') - -class CharacterForm(ContextModelForm): - class Meta: - model = Character - fields = ('name') - -def add(request, character_id): - message = None +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.views import free_comment, _preview +from xerxes.influences.models import Character +from xerxes.influences.models import Influence +from forms import get_influence_comment_form, InfluenceForm, CharacterForm +@login_required +def add_influence(request, character_id=None): initial = {} + # Get the current user's characters + chars = request.user.character_set.all() + + # 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 + f = InfluenceForm(request=request, initial=initial) - if (f.is_valid()): - f.save() - message = _("Influence added.") - f = None - if f: - # Only allow characters of the current user - f.fields['character']._set_queryset(request.user.character_set.all()) + # 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) + + if (f.is_valid()): + # The form was submitted, let's save it. + influence = f.save() + # Redirect to the just saved influence + return HttpResponseRedirect(reverse('influences_influence_detail', args=[influence.id])) - return render_to_response('influences/add.html', {'message' : message, 'form' : f}) + return render_to_response('influences/add_influence.html', {'form' : f}, RequestContext(request)) -def addCharacter(request): - message = None +@login_required +def add_character(request): f = CharacterForm(request=request) if (f.is_valid()): - c = f.save(commit=False) - c.player = request.user - c.save() - message = _("Character added.") - f = None + character = f.save(commit=False) + character.player = request.user + character.save() + return HttpResponseRedirect(reverse('influences_character_detail', args=[character.id])) + + return render_to_response('influences/add_character.html', {'form' : f}, RequestContext(request)) + +@login_required +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) + return render_to_response('influences/index.html', {'characters' : characters, 'influences' : influences}, RequestContext(request)) + +# +# The views below are very similar to django's generic views (in fact, +# they used to be generic views before). However, since they all depend +# on the currently logged in user (for limiting the show list or +# performing access control), we won't actually use the generic views +# here. + +@login_required +def character_list(request): + # Only show this player's characters + os = request.user.character_set.all() + return render_to_response('influences/character_list.html', {'object_list' : os}, RequestContext(request)) + +@login_required +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): + 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)) + +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): + + 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' : 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, 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 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) + # 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) - return render_to_response('influences/add_character.html', {'message' : message, 'form' : f}) +# vim: set sts=4 sw=4 expandtab: