* Add an addCharacter view to the influences app.

[matthijs/projects/xerxes.git] / influences / views.py

diff --git a/influences/views.py b/influences/views.py
index 1c7edb44753aa0dec03275fc869373a3883930c5..50ce60001996f26b82db32c2f8372aab70db8f5a 100644 (file)
--- a/influences/views.py
+++ b/influences/views.py
@@ -1,15 +1,20 @@
 from django.shortcuts import render_to_response
 from django.shortcuts import get_object_or_404
-from django import newforms as forms
 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(forms.ModelForm):
+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
 
@@ -27,3 +32,15 @@ def add(request, character_id):
                f.fields['character']._set_queryset(request.user.character_set.all())
 
        return render_to_response('influences/character_addinfluence.html', {'message' : message, 'form' : f})
+
+def addCharacter(request):
+    message = None
+    f = CharacterForm(request=request)
+    if (f.is_valid()):
+        c = f.save(commit=False)
+        c.player = request.user
+        c.save()
+        message = _("Character added.")
+        f = None
+
+    return render_to_response('influences/add_character.html', {'message' : message, 'form' : f})