def profile_edit(request, slug):
me = UserProfile.objects.get(user=request.user)
if me.display_name != slug:
return redirect('/dashboard/')
if request.method == 'POST':
form = UserProfileForm(request.POST, instance=me)
if form.is_valid():
form.save()
return redirect('/dashboard/')
else:
form = UserProfileForm(instance=me)
return render(request, 'profile_edit_or_create.html', locals())
def register(request):
try:
me = UserProfile.objects.get(user=request.user)
except UserProfile.DoesNotExist:
if request.method == 'POST':
me = User.objects.get(username=request.user)
form = UserProfileForm(request.POST)
if form.is_valid():
f = form.save(commit=False)
f.user = me
f.email = me.email
form.save()
return redirect('/dashboard/')
else:
form = UserProfileForm()
return render(request, 'profile_edit_or_create.html', locals())