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
9 from ee.influences.models import Character
10 from ee.influences.models import Influence
11 from ee.tools.forms import ContextModelForm
15 class InfluenceForm(ContextModelForm):
18 fields = ('character', 'contact', 'summary', 'description')
20 class CharacterForm(ContextModelForm):
26 def add(request, character_id=None):
28 # Get the current user's characters
29 chars = request.user.character_set.all()
32 initial['character'] = character_id
33 elif (chars.count() == 1):
34 initial['character'] = chars[0].id
36 f = InfluenceForm(request=request, initial=initial)
39 return HttpResponseRedirect(reverse('influences_influence_detail', args=[influence.id]))
41 # Only allow characters of the current user
42 f.fields['character']._set_queryset(chars)
44 return render_to_response('influences/add.html', {'form' : f}, RequestContext(request))
47 def addCharacter(request):
48 f = CharacterForm(request=request)
50 character = f.save(commit=False)
51 character.player = request.user
53 return HttpResponseRedirect(reverse('influences_character_detail', args=[character.id]))
55 return render_to_response('influences/add_character.html', {'form' : f}, RequestContext(request))
59 characters = request.user.character_set.all()
60 influences = Influence.objects.filter(character__in=characters)
61 return render_to_response('influences/index.html', {'characters' : characters, 'influences' : influences}, RequestContext(request))