(二)Django進階之路 自定義管理器
阿新 • • 發佈:2019-01-04
在 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()