Adding user logout log

Zambom
1 parent 027410af
Showing 3 changed files with 31 additions and 11 deletions Show diff stats
log/decorators.py
users/urls.py
users/views.py
@@ -10,17 +10,25 @@ def log_decorator(log_component = &#39;&#39;, log_action = &#39;&#39;, log_resource = &#39;&#39;):
 	def _log_decorator(view_function):
 		def _decorator(request, *args, **kwargs):
+			user = None
+
+			if request.user.is_authenticated:
+				user = request.user				
 			response = view_function(request, *args, **kwargs)
-			if request.user.is_authenticated:
-				
+			log_context = {}
+
+			if hasattr(request, 'log_context'):
+				log_context = request.log_context
+
+			if user:				
 				log = Log()
-				log.user = str(request.user)
-				log.user_id = request.user.id
-				log.user_email = request.user.email
+				log.user = str(user)
+				log.user_id = user.id
+				log.user_email = user.email
 				log.component = log_component
-				log.context = request.log_context
+				log.context = log_context
 				log.action = log_action
 				log.resource = log_resource
@@ -55,9 +63,15 @@ def log_decorator_ajax(log_component = &#39;&#39;, log_action = &#39;&#39;, log_resource = &#39;&#39;):
 					response = view_function(request, *args, **kwargs)
+					log_context = {}
+
+					if hasattr(request, 'log_context'):
+						log_context = request.log_context
+
 					log = Log.objects.latest('id')
-					log.context = request.log_context
+					log.context = log_context
 					log.save()
+					
 			elif view_action == 'close':
 				if request.user.is_authenticated:
 					log = get_object_or_404(Log, id = request.GET.get('log_id'))
@@ -5,7 +5,7 @@ from . import views
 urlpatterns = [
 	url(r'^login/$', views.login, name='login'),
-	url(r'^logout/$', auth_views.logout, {'next_page': 'users:login'}, name='logout'),
+	url(r'^logout/$', views.logout, {'next_page': 'users:login'}, name='logout'),
 	url(r'^signup/$', views.RegisterUser.as_view(), name = 'signup'),
 	url(r'^forgot_password/$', views.ForgotPassword.as_view(), name = 'forgot_pass'),
 	url(r'^reset_password_confirm/(?P<uidb64>[0-9A-Za-z]+)-(?P<token>.+)/$', views.PasswordResetConfirmView.as_view(), name = 'reset_password_confirm'), 
 from django.shortcuts import get_object_or_404, redirect, render
 from django.views import generic
 from django.contrib import messages
-from django.contrib.auth import authenticate, login as login_user
+from django.contrib.auth import authenticate, login as login_user, logout as logout_user
 from django.contrib.auth.mixins import LoginRequiredMixin
 from django.core.urlresolvers import reverse, reverse_lazy
 from django.utils.translation import ugettext_lazy as _
@@ -498,8 +498,6 @@ def login(request):
 				next_url = request.GET.get('next', None)
-				request.log_context = {}
-
 				if next_url:
 					return redirect(next_url)
@@ -514,6 +512,14 @@ def login(request):
 	return render(request, "users/login.html", context)
+@log_decorator('user', 'logout', 'system')
+def logout(request, next_page = None):
+	logout_user(request)
+
+	if next_page:
+		return redirect(next_page)
+
+	return redirect(reverse('users:login'))
 # API VIEWS
 class UserViewSet(viewsets.ModelViewSet):
	@@ -5,7 +5,7 @@ from . import views		@@ -5,7 +5,7 @@ from . import views
5		5
6	urlpatterns = [	6	urlpatterns = [
7	url(r'^login/$', views.login, name='login'),	7	url(r'^login/$', views.login, name='login'),
8	- url(r'^logout/$', auth_views.logout, {'next_page': 'users:login'}, name='logout'),	8	+ url(r'^logout/$', views.logout, {'next_page': 'users:login'}, name='logout'),
9	url(r'^signup/$', views.RegisterUser.as_view(), name = 'signup'),	9	url(r'^signup/$', views.RegisterUser.as_view(), name = 'signup'),
10	url(r'^forgot_password/$', views.ForgotPassword.as_view(), name = 'forgot_pass'),	10	url(r'^forgot_password/$', views.ForgotPassword.as_view(), name = 'forgot_pass'),
11	url(r'^reset_password_confirm/(?P<uidb64>[0-9A-Za-z]+)-(?P<token>.+)/$', views.PasswordResetConfirmView.as_view(), name = 'reset_password_confirm'),	11	url(r'^reset_password_confirm/(?P<uidb64>[0-9A-Za-z]+)-(?P<token>.+)/$', views.PasswordResetConfirmView.as_view(), name = 'reset_password_confirm'),