1. 程式人生 > >詳解Python的Django框架中Manager方法的使用

詳解Python的Django框架中Manager方法的使用

在語句Book.objects.all()中,objects是一個特殊的屬性,需要通過它查詢資料庫。 在第5章,我們只是簡要地說這是模組的manager 。現在是時候深入瞭解managers是什麼和如何使用了。

總之,模組manager是一個物件,Django模組通過它進行資料庫查詢。 每個Django模組至少有一個manager,你可以建立自定義manager以定製資料庫訪問。

下面是你建立自定義manager的兩個原因: 增加額外的manager方法,和/或修manager返回的初始QuerySet。
增加額外的Manager方法

增加額外的manager方法是為模組新增表級功能的首選辦法。

例如,我們為Book模型定義了一個title_count()方法,它需要一個關鍵字,返回包含這個關鍵字的書的數量。 (這個例子有點牽強,不過它可以說明managers如何工作。)

# models.py
 
from django.db import models
 
# ... Author and Publisher models here ...
 
**class BookManager(models.Manager):**
  **def title_count(self, keyword):**
    **return self.filter(title__icontains=keyword).count()**
 
class Book(models.Model):
  title = models.CharField(max_length=100)
  authors = models.ManyToManyField(Author)
  publisher = models.ForeignKey(Publisher)
  publication_date = models.DateField()
  num_pages = models.IntegerField(blank=True, null=True)
  **objects = BookManager()**
 
  def __unicode__(self):
    return self.title

有了這個manager,我們現在可以這樣做:

>>> Book.objects.title_count('django')
4
>>> Book.objects.title_count('python')
18

下面是編碼該注意的一些地方:

  •     我們建立了一個BookManager類,它繼承了django.db.models.Manager。這個類只有一個title_count()方法,用來做統計。 注意,這個方法使用了self.filter(),此處self指manager本身。
  •     我們把BookManager()賦值給模型的objects屬性。 它將取代模型的預設manager(objects)如果我們沒有特別定義,它將會被自動建立。 我們把它命名為objects,這是為了與自動建立的manager保持一致。

為什麼我們要新增一個title_count()方法呢?是為了將經常使用的查詢進行封裝,這樣我們就不必重複編碼了。
修改初始Manager QuerySets

manager的基本QuerySet返回系統中的所有物件。 例如,`` Book.objects.all()`` 返回資料庫book中的所有書本。

我們可以通過覆蓋Manager.get_query_set()方法來重寫manager的基本QuerySet。 get_query_set()按照你的要求返回一個QuerySet。

例如,下面的模型有* 兩個* manager。一個返回所有對像,另一個只返回作者是Roald Dahl的書。

from django.db import models
 
**# First, define the Manager subclass.**
**class DahlBookManager(models.Manager):**
  **def get_query_set(self):**
    **return super(DahlBookManager, self).get_query_set().filter(author='Roald Dahl')**
 
**# Then hook it into the Book model explicitly.**
class Book(models.Model):
  title = models.CharField(max_length=100)
  author = models.CharField(max_length=50)
  # ...
 
  **objects = models.Manager() # The default manager.**
  **dahl_objects = DahlBookManager() # The Dahl-specific manager.**

在這個示例模型中,Book.objects.all()返回了資料庫中的所有書本,而Book.dahl_objects.all()只返回了一本. 注意我們明確地將objects設定成manager的例項,因為如果我們不這麼做,那麼唯一可用的manager就將是dah1_objects。

當然,由於get_query_set()返回的是一個QuerySet物件,所以我們可以使用filter(),exclude()和其他一切QuerySet的方法。 像這些語法都是正確的:

Book.dahl_objects.all()
Book.dahl_objects.filter(title='Matilda')
Book.dahl_objects.count()

這個例子也指出了其他有趣的技術: 在同一個模型中使用多個manager。 只要你願意,你可以為你的模型新增多個manager()例項。 這是一個為模型新增通用濾器的簡單方法。

例如:

class MaleManager(models.Manager):
  def get_query_set(self):
    return super(MaleManager, self).get_query_set().filter(sex='M')
 
class FemaleManager(models.Manager):
  def get_query_set(self):
    return super(FemaleManager, self).get_query_set().filter(sex='F')
 
class Person(models.Model):
  first_name = models.CharField(max_length=50)
  last_name = models.CharField(max_length=50)
  sex = models.CharField(max_length=1, choices=(('M', 'Male'), ('F', 'Female')))
  people = models.Manager()
  men = MaleManager()
  women = FemaleManager()

這個例子允許你執行`` Person.men.all()`` ,`` Person.women.all()`` ,`` Person.people.all()`` 查詢,生成你想要的結果。

如果你使用自定義的Manager物件,請注意,Django遇到的第一個Manager(以它在模型中被定義的位置為準)會有一個特殊狀態。 Django將會把第一個Manager 定義為預設Manager ,Django的許多部分(但是不包括admin應用)將會明確地為模型使用這個manager。 結論是,你應該小心地選擇你的預設manager。因為覆蓋get_query_set() 了,你可能接受到一個無用的返回對像,你必須避免這種情況。