Question

我有一个用于显示表单的FormView，代码依旧如下：

class AddProject(FormView):
    template_name = "project/add_project.html"

    @method_decorator(check_user_type)
    def dispatch(self,request, *args, **kwargs):
        return super(AddProject,self).dispatch(request, *args, **kwargs)

    def get_form_class(self):
        return AddProjectForm 

    def form_valid(self,form):
        #do validation, return response

装饰器check_user_type是这样的：

def check_user_type(func):
    def wrapped_func(request, *args, **kwargs):
        kwargs['invalid_user'] = True
        return func(request,*args, **kwargs)
    return wrapped_func

在我的装饰器中我想要确保只有特定类型的用户才能看到表单，即如果request.user.Iam =='Architect'或request.user.Iam =='Interior Designer'只看到表单和其他人看到一条消息“只有架构师/室内设计师才能上传照片”。为此我想插入一个变量'invalid_user'来传递，这取决于我显示表单或消息。问题是我无法传递变量:(随之而来......并且有疑问......如果我已经正确地设计了这样做的想法......？

Answer 1

如果我理解正确，你想把这个论点传递给check_user_type装饰师，对吧？然后你需要在装饰器中嵌套另一个函数，使用闭包在其中设置变量。像

这样的东西

def check_user_type(parameter):
    def check_user_type_inner(func):
        def wrapped_func(request, *args, **kwargs):
            ... # parameter here depends on argument passed to most outer function
            return func(request,*args, **kwargs)
        return wrapped_func
    return check_user_type_inner

然后参数在两个内部函数的范围内都可用。

基于类的视图和装饰器

1 个答案: