diff options
| -rw-r--r-- | users/urls.py | 1 | ||||
| -rw-r--r-- | users/views.py | 22 |
2 files changed, 11 insertions, 12 deletions
diff --git a/users/urls.py b/users/urls.py index c09d7e01..3589c7e0 100644 --- a/users/urls.py +++ b/users/urls.py @@ -4,6 +4,7 @@ from django.contrib import admin app_name = 'users' urlpatterns = [ + path('', views.home, name='home'), path('login', views.login_user, name='login'), path('logout', views.logout_user, name='logout'), path('update', views.update_user, name='update'), diff --git a/users/views.py b/users/views.py index 57a73d41..b0c2126e 100644 --- a/users/views.py +++ b/users/views.py @@ -20,16 +20,14 @@ def get_ref(request): try: referrer = request.META.get('QUERY_STRING').split('referrer=')[1] except: - referrer = None - if referrer and '?' in referrer: + # Raise a unauthorized error if the referrer is not set + return HttpResponse('Unauthorized', status=401) + if '?' in referrer: referrer = referrer.split('?')[0] return referrer -def red_(referrer): - if referrer is None: - return HttpResponse('Unauthorized', status=401) - else: - return redirect(referrer) +def home(request): + return redirect('blog:home') @csrf_exempt # Create your views here. @@ -40,7 +38,7 @@ def login_user(request): print (username, password) if username == '' or password == '': messages.error(request, 'Please fill in all fields.', extra_tags='loginError') - return red_(referrer) + return HttpResponseRedirect(referrer) else: # check if email is verified user = authenticate(request, username=username, password=password) @@ -48,18 +46,18 @@ def login_user(request): email_verified = UserProfile.objects.get(user=user.pk).email_verified if email_verified: login(request, user) - return red_(referrer) + return HttpResponseRedirect(referrer) else: messages.error(request, 'EVERR', extra_tags='loginError') - return red_(referrer + '?username=' + username) + return HttpResponseRedirect(referrer + '?username=' + username) else: messages.error(request, 'Invalid username or password.', extra_tags='loginError') - return red_(referrer + '?username=' + username) + return HttpResponseRedirect(referrer + '?username=' + username) def logout_user(request): referrer = get_ref(request) logout(request) - return red_(referrer) + return HttpResponseRedirect(referrer) def update_user(request): username = request.user |