1 from django.contrib.auth.decorators import login_required
2 from django.shortcuts import render_to_response
3 from django.shortcuts import get_object_or_404
4 from django.template import RequestContext
5 from django.utils.translation import ugettext as _
6 from django.contrib.auth.models import User
7 from django.core.urlresolvers import reverse
8 from django.http import HttpResponseRedirect, HttpResponseForbidden
9 from django.views.generic.list_detail import object_detail, object_list
10 from ee.influences.models import Character
11 from ee.influences.models import Influence
12 from ee.tools.forms import ContextModelForm
16 class InfluenceForm(ContextModelForm):
19 fields = ('character', 'contact', 'summary', 'description')
21 class CharacterForm(ContextModelForm):
27 def add(request, character_id=None):
29 # Get the current user's characters
30 chars = request.user.character_set.all()
33 initial['character'] = character_id
34 elif (chars.count() == 1):
35 initial['character'] = chars[0].id
37 f = InfluenceForm(request=request, initial=initial)
39 influence = f.save(commit=False)
40 if (influence.character.player == request.user):
42 return HttpResponseRedirect(reverse('influences_influence_detail', args=[influence.id]))
44 # TODO: Make this a bit more pretty. Perhaps throw an exception here and add some middleware to catch it?
45 return HttpResponseForbidden("Forbidden -- Trying to submit influence for somebody else's character")
47 # Only allow characters of the current user
48 f.fields['character']._set_queryset(chars)
50 return render_to_response('influences/add.html', {'form' : f}, RequestContext(request))
53 def addCharacter(request):
54 f = CharacterForm(request=request)
56 character = f.save(commit=False)
57 character.player = request.user
59 return HttpResponseRedirect(reverse('influences_character_detail', args=[character.id]))
61 return render_to_response('influences/add_character.html', {'form' : f}, RequestContext(request))
65 characters = request.user.character_set.all()
66 influences = Influence.objects.filter(character__in=characters)
67 return render_to_response('influences/index.html', {'characters' : characters, 'influences' : influences}, RequestContext(request))
70 def character_list(*args, **kwargs):
71 return object_list(*args, **kwargs)
74 def character_detail(*args, **kwargs):
75 return object_detail(*args, **kwargs)
78 def influence_detail(*args, **kwargs):
79 return object_detail(*args, **kwargs)