我的模型上有一个private布尔标志,还有一个自定义管理器,它用过滤器覆盖get_query_set方法,删除private=True:
class myManager(models.Manager):
    def get_query_set(self):
        qs = super(myManager, self).get_query_set()
        qs = qs.filter(private=False)
        return qs
class myModel(models.Model):
    private = models.BooleanField(default=False)
    owner = models.ForeignKey('Profile', related_name="owned")
    #...etc...
    objects = myManager()我希望默认查询集将私有模型排除为默认的安全措施,从而防止意外使用显示私有模型的模型。
然而,有时我会想要展示私人模型,所以我有以下关于经理的信息:
def for_user(self, user):
    if user and not user.is_authenticated():
        return self.get_query_set()
    qs = super(myManager, self).get_query_set()
    qs = qs.filter(Q(owner=user, private=True) | Q(private=False))
    return qs这是很好的工作,但有一个限制,我不能连锁过滤器。当我有一个fk指向myModel并使用otherModel.mymodel_set时,这就成了一个问题。OtherModel.mymodel_set.for_user(用户)无法工作,因为mymodel_set返回一个QuerySet对象,而不是管理器。
现在,真正的问题开始了,因为我看不到使for_user()方法在QuerySet子类上工作的方法,因为我无法访问来自queryset子类的完整、未经过滤的查询集(基本上是覆盖get_query_set),就像在管理器中那样(使用super()来获取基本的查询集)。
解决这个问题的最好方法是什么?
我没有绑定到任何特定的界面,但我希望它尽可能地变得单调/干燥。显然,我可以放弃安全性,只需调用一个方法来筛选出每个调用上的私有任务,但我真的不想这样做。
更新
manji下面的答案非常接近,但是当我想要的queryset不是默认queryset的子集时,它就失败了。我想这里真正的问题是如何从链式查询中删除特定的筛选器?
发布于 2011-06-24 18:21:53
如果您需要“重置”QuerySet,您可以访问查询集的模型并再次调用原始管理器(完全重置)。但是,这对您可能不是很有用,除非您正在跟踪前面的筛选器/排除etc语句,并且可以在重置查询集上再次重放它们。有了一些计划,这实际上不会太难做,但可能有点野蛮的力量。
总的来说,曼吉的回答肯定是正确的。
因此,修改manji的答案,您需要用("model"."owner_id" = 2 AND "model"."private" = True ) OR "model"."private" = False )替换现有的("model"."owner_id" = 2 AND "model"."private" = True ) OR "model"."private" = False )。要做到这一点,您需要遍历queryset的where对象上的query对象,以找到要删除的相关位。查询对象有一个WhereNode对象,它表示where子句的树,每个节点都有多个子节点。您必须调用节点上的as_sql来确定它是否是您想要的:
from django.db import connection
qn = connection.ops.quote_name
q = myModel.objects.all()
print q.query.where.children[0].as_sql(qn, connection)这应该会给你这样的东西:
('"model"."private" = ?', [False])然而,尝试这样做可能比它的价值要大得多,而且它正在钻研一些可能不是API稳定的Django。
我的建议是使用两个经理。一个可以访问所有东西(排序的转义口),另一个应用了默认的过滤。默认的管理器是第一个,因此您需要根据您需要做的事情来处理顺序。然后重构您的代码,知道使用哪一个-这样您就不会有问题,在那里已经有额外的private=False子句。
https://stackoverflow.com/questions/6383860
复制相似问题