1 from django.shortcuts import render_to_response
2 from django.shortcuts import get_object_or_404
3 from django.utils.translation import ugettext as _
4 from ee.influences.models import Character
5 from ee.influences.models import Influence
6 from ee.tools.forms import ContextModelForm
8 class InfluenceForm(ContextModelForm):
11 fields = ('character', 'contact', 'description')
13 class CharacterForm(ContextModelForm):
18 def add(request, character_id):
23 initial['character'] = character_id
25 f = InfluenceForm(request=request, initial=initial)
28 message = _("Influence added.")
32 # Only allow characters of the current user
33 f.fields['character']._set_queryset(request.user.character_set.all())
35 return render_to_response('influences/add.html', {'message' : message, 'form' : f})
37 def addCharacter(request):
39 f = CharacterForm(request=request)
41 c = f.save(commit=False)
42 c.player = request.user
44 message = _("Character added.")
47 return render_to_response('influences/add_character.html', {'message' : message, 'form' : f})