def character_add(request):
if request.method == 'POST':
form = CharacterForm(request.POST)
if form.is_valid():
character = AlbionCharacter(name=request.POST['name'],
role=request.POST['role'],
user=request.user)
character.save()
return redirect('dashboard')
else:
form = CharacterForm()
context = {}
context['form'] = form
return render(request, 'characters/character_add.html', context)
def character_update(request):
if request.method == 'POST':
form = CharacterForm(request.POST)
if form.is_valid():
character = AlbionCharacter.objects.get(user=request.user)
character.name = request.POST['name']
character.role = request.POST['role']
character.secondary = request.POST['secondary']
character.discord = request.POST['discord']
character.save()
return redirect('dashboard')
else:
character = AlbionCharacter.objects.get(user=request.user)
initial = {}
initial['name'] = character.name
initial['role'] = character.role
initial['secondary'] = character.secondary
initial['discord'] = character.discord
form = CharacterForm(initial=initial)
context = {}
context['form'] = form
return render(request, 'characters/character_update.html', context)
