57 lines
2.4 KiB
Python
57 lines
2.4 KiB
Python
|
from employee.models import Employee
|
||
|
|
from recruitment.models import Recruitment
|
||
|
|
from onboarding.models import *
|
||
|
|
from django.contrib import messages
|
||
|
|
from django.http import HttpResponseRedirect
|
||
|
|
|
||
|
|
decorator_with_arguments = lambda decorator: lambda *args, **kwargs: lambda func: decorator(func, *args, **kwargs)
|
||
|
|
|
||
|
|
@decorator_with_arguments
|
||
|
|
def all_manager_can_enter(function, perm):
|
||
|
|
"""
|
||
|
|
This method is used to check permission of employee for enter to the function
|
||
|
|
"""
|
||
|
|
def _function(request, *args, **kwargs):
|
||
|
|
user = request.user
|
||
|
|
employee = Employee.objects.filter(employee_user_id=user).first()
|
||
|
|
is_manager = OnboardingTask.objects.filter(employee_id=employee).exists() or OnboardingStage.objects.filter(employee_id=employee) or Recruitment.objects.filter(recruitment_managers=employee).exists()
|
||
|
|
if user.has_perm(perm) or is_manager:
|
||
|
|
return function(request, *args, **kwargs)
|
||
|
|
messages.info(request,'You dont have permission.')
|
||
|
|
return HttpResponseRedirect(request. META. get('HTTP_REFERER', '/'))
|
||
|
|
|
||
|
|
return _function
|
||
|
|
|
||
|
|
|
||
|
|
@decorator_with_arguments
|
||
|
|
def stage_manager_can_enter(function, perm):
|
||
|
|
"""
|
||
|
|
This method is used to check permission of employee for enter to the function
|
||
|
|
"""
|
||
|
|
def _function(request, *args, **kwargs):
|
||
|
|
user = request.user
|
||
|
|
employee = Employee.objects.filter(employee_user_id=user).first()
|
||
|
|
is_manager = OnboardingStage.objects.filter(employee_id=employee) or Recruitment.objects.filter(recruitment_managers=employee).exists()
|
||
|
|
if user.has_perm(perm) or is_manager:
|
||
|
|
return function(request, *args, **kwargs)
|
||
|
|
messages.info(request,'You dont have permission.')
|
||
|
|
return HttpResponseRedirect(request. META. get('HTTP_REFERER', '/'))
|
||
|
|
|
||
|
|
return _function
|
||
|
|
|
||
|
|
|
||
|
|
@decorator_with_arguments
|
||
|
|
def recruitment_manager_can_enter(function, perm):
|
||
|
|
"""
|
||
|
|
This method is used to check permission of employee for enter to the function
|
||
|
|
"""
|
||
|
|
def _function(request, *args, **kwargs):
|
||
|
|
user = request.user
|
||
|
|
employee = Employee.objects.filter(employee_user_id=user).first()
|
||
|
|
is_manager = Recruitment.objects.filter(recruitment_managers=employee).exists()
|
||
|
|
if user.has_perm(perm) or is_manager:
|
||
|
|
return function(request, *args, **kwargs)
|
||
|
|
messages.info(request,'You dont have permission.')
|
||
|
|
return HttpResponseRedirect(request. META. get('HTTP_REFERER', '/'))
|
||
|
|
|
||
|
|
return _function
|