from django.http import HttpResponseRedirect
from django.shortcuts import redirect
from django.urls import reverse

def checklogindriver(function):
	def  wrap(request, *args, **kwargs):
		print('checklogindriver decorator')
		if 'userrole' in request.session:
			userrole = request.session['userrole']
			print('userrole is',userrole)
			if userrole != 'driver':
				print('not driver')
				# return HttpResponseRedirect(reverse('customer-login'))
				return redirect('/login/')
		else:
			# return HttpResponseRedirect(reverse('customer-login'))
			return redirect('/login/')
		return function(request, *args, **kwargs)
	wrap.__doc__ = function.__doc__
	return wrap