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 

Теперь мой вопрос, что это лучший способ, чтобы ограничить доступ к дочерним моделям компании к экземпляру запроса компании ?

Я мог бы сделать это как task = Task.objects.get(pk=4,section__task=task), но это плохо, если моя структура модели становится все глубже.

Редактировать: Я мог бы дать друг другу модель иностранного ключа компании, но разве это хорошая практика хранения отношений избыточным? Редактировать 2: Нет, это не так. См. Is it bad to use redundant relationships?.

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 

Тогда вы можете ограничить свой QuerySet так:

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

Это sloooowwww, но это все, что я могу придумать, чтобы в 2:00 AM