influences/views.py

   1 from django.shortcuts import render_to_response
   2 from django.shortcuts import get_object_or_404
   3 from django import newforms as forms
   4 from django.utils.translation import ugettext as _
   5 from ee.influences.models import Character
   6 from ee.influences.models import Influence
   7
   8 class InfluenceForm(forms.ModelForm):
   9         class Meta:
  10                 model = Influence
  11                 fields = ('character', 'contact', 'description')
  12
  13 def add(request, character_id):
  14         message = None
  15
  16         if (request.method == "POST"):
  17                 f = InfluenceForm(request.POST)
  18                 if (f.is_valid()):
  19                         f.save()
  20                         message = _("Influence added.")
  21                         f = None
  22         else:
  23                 f = InfluenceForm()
  24
  25         if f:
  26                 # Only allow characters of the current user
  27                 f.fields['character']._set_queryset(request.user.character_set.all())
  28
  29         return render_to_response('influences/character_addinfluence.html', {'message' : message, 'form' : f})