Django设计问题

Question

我在 django 中有一个具有布尔私有/公共属性的模型：

class TestModel(models.Model):
    name = models.CharField()
    is_public = models.BooleanField(default=False)

我希望每次在应用程序中查询该模型时，它只向普通用户返回公共实例，向超级用户返回所有可用实例。

我需要如何以及在哪里实现此类功能？

Answer 1

您可以在视图层实现该逻辑，可能使用 自定义管理器。

您的经理看起来像这样：

class PublicTestModelManager(models.Manager):
  def get_query_set(self):
    return super(PublicTestModelManager, self).get_query_set().filter(is_public = True)

您的模型看起来像这样：

class TestModel(models.Model):
    name = models.CharField()
    is_public = models.BooleanField(default=False)
    objects = models.Manager() # The default manager.
    public_objects = PublicTestModelManager()

然后您可以编写一个选择正确经理的函数：

def test_objects_for_user(user):
  if user.is_superuser:
    return TestModel.objects
  else:
    return TestModel.public_objects

然后在您看来您可以使用：

test_objects_for_user(request.user).all()