user_passes_test 用于 Django 中除 User 之外的模型

Question

如果 ifAdmin（uProject 模型中的对象）== True，我希望允许用户编辑项目页面。我目前正在尝试使用 @user_passes_test 呈现视图以更新项目，但遇到了困难。我正在从 main.models 获取 uProjects，并且正在使用 projects.views。这是我的代码。

def admin_check(uProjects):
    return uProjects.ifAdmin = True

@user_passes_test(admin_check)
def update(request): etc, etc, etc

models.py

class uProjects(models.Model):
   
    user = models.ForeignKey(User, on_delete=models.CASCADE)
    project = models.ForeignKey(Project, on_delete=models.CASCADE)
    ifAccepted = models.BooleanField(null = True, blank=False, default=False)
    #ifLeader = models.BooleanField(null = False, blank=False)
    ifAdmin = models.BooleanField(null = True, blank=False, default=False)
    title = models.CharField(max_length=100, null=False, blank=False)
    def __str__(self):
        return self.user.username + ',' + self.project.name

困难在于我试图让 admin_check 与 uProjects 一起工作，而不是 User 模型。我也在尝试查看 ifAdmin == True，但我上面所做的方式会导致错误。错误是“用户对象没有属性ifAdmin”

Answer 1

你可以这样使用：

def admin_check(user):
    return user.uprojects_set.filter(ifAdmin=True).exists()

@user_passes_test(admin_check)
def update(request): etc, etc, etc

但这适用于在至少一个 uProject 实例中拥有 ifAdmin 权限的任何用户。无法通过uProject信息进行区分。

为此，您可以像这样编写自定义析取器：

from functools import wraps
from django.http import HttpResponseRedirect

def admin_check(function):
  @wraps(function)
  def wrap(request, *args, **kwargs):
        user = request.user
        name = kwargs.get('name')  # assuming you are getting the project id here via url: "/<project_id:int>"
        if uProjects.objects.filter(title=name, user=user, ifAdmin=True).exists():
             return function(request, *args, **kwargs)
        else:
            return HttpResponseRedirect('/')

  return wrap