Django：限制模型数据

Question

我正在寻找一种方式来限制我可以通过模型获得的查询集。

假设我有以下模型（具有依赖项）：

Company
 |- Section
 |   |- Employee
 |   |- Task
 |   `- more models...
 |- Customer
 |   |- Contract
 |   |- Accounts
 |   `- other great models ...
 `- some more models...

应当指出，我的真实模型要深入得多，并且与业务无关。

使用上下文处理器，我添加了一个公司实例来请求：

def magic_view(request):
    request.company # is a instance of Company model

现在，我的问题是将对Company的子模型的访问限制为company的请求实例的最佳方法是什么？

我可以使它像task = Task.objects.get(pk=4,section__task=task) ，但是如果我的模型结构越来越深入，这是一种不好的方法。

编辑：我可以给对方模型公司的外键，但这是存储关系冗余的好习惯吗？ 编辑2：不，不是。 请参阅使用冗余关系是否不好？ 。

Answer 1

我通过以下方式解决了它：

首先，我创建了一个CurrentCompanyManager。

class CurrentCompanyManager(models.Manager):
    def __init__(self,field,*args,**kwargs):
        super(CurrentCompanyManager,self).__init__(*args,**kwargs)
        self.__field_name = field

    def on(self,company):
        return self.filter( **{  self.__field_name + '__id__exact':company.id } )

然后，我将管理器添加到需要的所有模型中。

class Employee(models.Model):
    # some fields and relationships
    objects = CurrentCompanyManager("section__company")

class Accounts(models.Model):
    # some fields and relationships
    objects = CurrentCompanyManager("customer__company")

现在，我可以轻松地在视图中限制模型数据。

def magic_view(request):
    Employee.objects.on(request.company).all()

它应该是不言自明的。 如果没有，那就问我。

Answer 2

将您的层次结构视为图形。 让所有模型扩展Node类：

class Node(models.Model):
parent = models.ForeignKey("Node", blah blah...)
def get_root(self):
    n = self
    while ((n = n.parent) != None): pass
    return n

然后，您可以像这样限制查询集：

qset = Task.objects.filter(blah...)
result = []
for row in qset:
    if row.get_root() == request.company:
        result += row

这是sloooowwww，但这是我可以在2:00 AM提出的所有内容