1. 程式人生 > >(二)Django進階之路 自定義管理器

(二)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()