def follow(request): username = request.user.username following = request.POST.get('following', 'empty') try: User.objects.get(username__exact=following) except (KeyError, User.DoesNotExist): return render(request, 'follow.html', { 'error_message': "Not a valid user!"}) else: follow = Follow(username=username, following=following) follow.save() return HttpResponseRedirect(reverse('index'))
def register(request): username = request.POST.get('username', 'empty') password = request.POST.get('password', 'empty') email = request.POST.get('email', 'empty') try: user = User.objects.create_user(username, email, password) user.save() except(IntegrityError): return render(request, 'register.html', { 'error_message': "Username taken!"}) else: user = authenticate(username=username, password=password) follow = Follow(username=username, following=username) follow.save() login(request, user) return HttpResponseRedirect(reverse('index'))