decorators.py

from functools import wraps

from django.shortcuts import redirect


def verificar_funcionario(redirect_to='set_password'):
    def decorator(view_func):
        @wraps(view_func)
        def _wrapped_view(request, *args, **kwargs):
            if request.user.funcionario.mudou_senha_padrao:
                return view_func(request, *args, **kwargs)
            
            return redirect(redirect_to)
        return _wrapped_view
    return decorator