* Remove old addInfluence view and replace it by a new add view (which allows you...

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

diff --git a/influences/views.py b/influences/views.py
index 4450efadc8e863fe825e9e5871d9cb0faf42fca8..1b988dd43227baae8d1b731169ff6e9f21d25ecd 100644 (file)
--- a/influences/views.py
+++ b/influences/views.py
@@ -4,20 +4,25 @@ from ee.influences.models import Character
 from ee.influences.models import Influence
 from django import newforms as forms
 
-class AddInfluenceForm(forms.Form):
-       contact     = forms.CharField(max_length=255, help_text='Which contact will you ask?')
-       description = forms.CharField(help_text='What do you want to achieve')
+class InfluenceForm(forms.ModelForm):
+       class Meta:
+               model = Influence
+               fields = ('character', 'contact', 'description')
 
-def addinfluence(request, character_id):
-       character = get_object_or_404(Character, id=character_id)
+def add(request, character_id):
        message = None
+
        if (request.method == "POST"):
-               f = AddInfluenceForm(request.POST)
+               f = InfluenceForm(request.POST)
                if (f.is_valid()):
-                       inf = character.influence_set.create(contact = f.clean_data['contact'], description = f.clean_data['description'])
+                       f.save()
                        message = "Influence added"
                        f = None
        else:
-               f = AddInfluenceForm()
+               f = InfluenceForm()
+       
+       if f:
+               # Only allow characters of the current user
+               f.fields['character']._set_queryset(request.user.character_set.all())
 
-       return render_to_response('influences/character_addinfluence.html', {'character' : character, 'message' : message, 'form' : f})
+       return render_to_response('influences/character_addinfluence.html', {'message' : message, 'form' : f})