Django: Как получить доступ к исходному экземпляру из RelatedManager?

Я хотел бы получить доступ к Foo пример foo в моем методе менеджера baz:

foo.bar_set.baz()

baz будет обычно принимать аргумент Foo тип:

BarManager(models.Manager):
    def baz(self, foo=None):
        if foo is None:
            # assume this call originates from
            # a RelatedManager and set `foo`.
            # Otherwise raise an exception
        # do something cool with foo

Таким образом, как первый запрос выше, так и следующий работает одинаково:

Bar.objects.baz(foo)

Bar будет иметь ForeignKey для Foo:

class Bar(models.Model):
    foo = models.ForeignKey(Foo)
    objects = BarManager()

2 ответа

Решение

Если я правильно понимаю, что вы хотите, вам нужно сделать это:

 BarManager(models.Manager):
      use_for_related_fields = True

Редактировать: по- видимому, мне удалось полностью упустить момент. Вы можете использовать что-то вроде этого (может быть, немного "волшебно" на мой вкус, ну да ладно):

class BarManager(models.Manager):
    use_for_related_fields = True

    def bar(self, foo=None):
        if foo == None:
            qs = Foo.objects.all()
            for field_name, field_val in self.core_filters.items():
                field_name = field_name.split('__')[1]
                qs = qs.filter(**{ field_name: field_val })
            foo = qs.get()
        # do k00l stuff with foo

По крайней мере, в Django 2.1 вы можете просто:

foo.bar_set.instance

Другие вопросы по тегам