（二）Django进阶之路 自定义管理器

在 Test 模型中构造管理器子类, 并同步如下数据库

from django.db import models
class Test(model.Model):
    test_id = models.AutoField(primary_key=True)
    test_content = models.CharField(max_length=20)
    test_status = models.SmallIntegerField()

    custom_objects = models.Manager()

查询数据库中test表中既有数据（需建表后预先添加 test_status 1 表示无效 0 表示有效）

Test.custom_objects.all() 
此时再用 Test.objects.all() 会报错

自定义管理器子类（可定义多个）

from django.db import models


class Test(model.Model):
    test_id = models.AutoField(primary_key=True)
    test_content = models.CharField(max_length=20)
    test_status = models.SmallIntegerField()

    custom_objects = models.Manager()
    valid_objects = ValidManager()


class ValidManager(models.Manager):
    """
    重构父类get_queryset方法，获取有效数据
    """
    def get_queryset(self):
        return super(Valid_objects, self).get_queryset().filter(test_status=0)

使用自定义管理器

Test.valid_objects.all()

在自定义管理器中自定义方法

class ValidManager(models.Manager):
    """
    自定义方法，获取有效数据
    """
    def get_valid_test(self):
        return self.filter(test_status=0)

class Test(model.Model):
    test_id = models.AutoField(primary_key=True)
    test_content = models.CharField(max_length=20)
    test_status = models.SmallIntegerField()

    objects = ValidManager()

使用自定义管理器方法

Test.valid_objects.get_valid_test()

自定义查询集

class ValidQuerySet(models.QuerySet):
    """
    获取有效数据
    """
    def get_valid_test(self):
        return self.filter(test_status=0)


class ValidManager(models.Manager):
    """
    自定义方法，获取有效数据
    """
    def get_queryset(self):
        return ValidQuerySet(self.model, using=self._db)    


class Test(model.Model):
    test_id = models.AutoField(primary_key=True)
    test_content = models.CharField(max_length=20)
    test_status = models.SmallIntegerField()
    # 方法一：
    # objects = ValidManager()
    # 方法二:
    # objects = ValidQuerySet.as_manager()
    objects = ValidQuerySet.as_manager()

使用自定义查询集（使用方法）

Test.valid_objects.all().get_valid_test()

省略 Test.valid_objects.all().get_valid_test() 的 all()

class ValidQuerySet(models.QuerySet):
    """
    获取有效数据
    """
    def get_valid_test(self):
        return self.filter(test_status=0)


class ValidManager(models.Manager):
    """
    自定义方法，获取有效数据
    """
    def get_queryset(self):
        return ValidQuerySet(self.model, using=self._db)

    def get_valid_test(self):
        return self.get_queryset().get_valid_test() 


class Test(model.Model):
    test_id = models.AutoField(primary_key=True)
    test_content = models.CharField(max_length=20)
    test_status = models.SmallIntegerField()
    # 方法一：
    # objects = ValidManager()
    # 方法二:
    # objects = ValidQuerySet.as_manager()
    objects = ValidQuerySet.as_manager()

省略 all() 的直接使用自定义查询集的使用方法

Test.valid_objects.get_valid_test()