influences/views.py

   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
  12
  13
  14
  15 class InfluenceForm(ContextModelForm):
  16     class Meta:
  17         model = Influence
  18         fields = ('character', 'contact', 'summary', 'description')
  19
  20 class CharacterForm(ContextModelForm):
  21     class Meta:
  22         model = Character
  23         fields = ('name')
  24
  25 @login_required
  26 def add(request, character_id=None):
  27     initial = {}
  28     # Get the current user's characters
  29     chars = request.user.character_set.all()
  30
  31     if (character_id):
  32         initial['character'] = character_id
  33     elif (chars.count() == 1):
  34         initial['character'] = chars[0].id
  35
  36     f = InfluenceForm(request=request, initial=initial)
  37     if (f.is_valid()):
  38         influence = f.save()
  39         return HttpResponseRedirect(reverse('influences_influence_detail', args=[influence.id]))
  40
  41     # Only allow characters of the current user
  42     f.fields['character']._set_queryset(chars)
  43
  44     return render_to_response('influences/add.html', {'form' : f}, RequestContext(request))
  45
  46 @login_required
  47 def addCharacter(request):
  48     f = CharacterForm(request=request)
  49     if (f.is_valid()):
  50         character = f.save(commit=False)
  51         character.player = request.user
  52         character.save()
  53         return HttpResponseRedirect(reverse('influences_character_detail', args=[character.id]))
  54
  55     return render_to_response('influences/add_character.html', {'form' : f}, RequestContext(request))
  56
  57 @login_required
  58 def index(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))