自定义管理器中与Django访问相关的模型状态

时间:2019-02-24 14:30:58

标签: django django-models django-orm

我在Django中使用model_utils库。

我有两个模型,如下所示

class Book(TimeStampedModel):
    STATUS_CHOICES = Choices(
        (0, 'public', _('public')),
        (1, 'private', _('private')),
    )

    status = models.IntegerField(
        verbose_name=_('status'),
        choices=STATUS_CHOICES,
        default=STATUS_CHOICES.public,
        db_index=True,
    )

class Page(MPTTModel, AbstractPage):
    STATUS_CHOICES = Choices(
        (0, 'draft', _('draft')),
        (1, 'public', _('public')),
        (2, 'private', _('private')),
    )

    status = models.IntegerField(
            verbose_name=_('status'),
            choices=STATUS_CHOICES,
            default=STATUS_CHOICES.public,
            db_index=True,
        )

    book = models.ForeignKey(
            'book.Book',
            verbose_name=_('book'),
            related_name='pages',
            db_index=True,
            on_delete=models.CASCADE,
        )

BookPage模型都具有status字段。

我在managers.py中有两个自定义查询集类。

class BookQuerySet(models.QuerySet):
    def public(self):
        return self.filter(status=self.model.STATUS_CHOICES.public)

class PageQuerySet(models.QuerySet):    
    def public(self):
        return self.filter(book__status=0,
                           status=self.model.STATUS_CHOICES.public)

如上所述,book__status=0确实可以工作,但是我有点沮丧,因为我想使用类似self.book.model.STATUS_CHOICES.public的代码。

请告诉我如何访问相关的模型对象属性。

如果我导入from .models import Book,将导致“循环导入”。

谢谢。

1 个答案:

答案 0 :(得分:0)

  

如果我import from .models import Book,将导致“圆形导入”。

这是正确的,但是我们可以绕开它,可以将导入推迟到该方法中需要的时候,例如:

# app/managers.py

class BookQuerySet(models.QuerySet):

    def public(self):
        return self.filter(status=self.model.STATUS_CHOICES.public)

class PageQuerySet(models.QuerySet):

    def public(self):
        from .models import Book  # no circular imports
        return self.filter(
            book__status=Book.STATUS_CHOICES.public,
            status=self.model.STATUS_CHOICES.public
        )