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 django.views.generic.list_detail import object_detail, object_list
  10 from ee.influences.models import Character
  11 from ee.influences.models import Influence
  12 from ee.tools.forms import ContextModelForm
  13
  14
  15
  16 class InfluenceForm(ContextModelForm):
  17     class Meta:
  18         model = Influence
  19         fields = ('character', 'contact', 'summary', 'description')
  20
  21 class CharacterForm(ContextModelForm):
  22     class Meta:
  23         model = Character
  24         fields = ('name')
  25
  26 @login_required
  27 def add(request, character_id=None):
  28     initial = {}
  29     # Get the current user's characters
  30     chars = request.user.character_set.all()
  31
  32     if (character_id):
  33         initial['character'] = character_id
  34     elif (chars.count() == 1):
  35         initial['character'] = chars[0].id
  36
  37     f = InfluenceForm(request=request, initial=initial)
  38     if (f.is_valid()):
  39         influence = f.save()
  40         return HttpResponseRedirect(reverse('influences_influence_detail', args=[influence.id]))
  41
  42     # Only allow characters of the current user
  43     f.fields['character']._set_queryset(chars)
  44
  45     return render_to_response('influences/add.html', {'form' : f}, RequestContext(request))
  46
  47 @login_required
  48 def addCharacter(request):
  49     f = CharacterForm(request=request)
  50     if (f.is_valid()):
  51         character = f.save(commit=False)
  52         character.player = request.user
  53         character.save()
  54         return HttpResponseRedirect(reverse('influences_character_detail', args=[character.id]))
  55
  56     return render_to_response('influences/add_character.html', {'form' : f}, RequestContext(request))
  57
  58 @login_required
  59 def index(request):
  60     characters = request.user.character_set.all()
  61     influences = Influence.objects.filter(character__in=characters)
  62     return render_to_response('influences/index.html', {'characters' : characters, 'influences' : influences}, RequestContext(request))
  63
  64 @login_required
  65 def character_list(*args, **kwargs):
  66     return object_list(*args, **kwargs)
  67
  68 @login_required
  69 def character_detail(*args, **kwargs):
  70     return object_detail(*args, **kwargs)
  71
  72 @login_required
  73 def influence_detail(*args, **kwargs):
  74     return object_detail(*args, **kwargs)