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())