码迷,mamicode.com
首页 > 其他好文 > 详细

Django之模型的高级用法

时间:2019-11-12 09:19:15      阅读:69      评论:0      收藏:0      [点我收藏+]

标签:cti   stat   赋值   多个   raw   完全   ati   ini   elf   

from django.db import models
class Publisher(models.Model):
    name = models.CharField(max_length=30)
    address = models.CharField(max_length=50)
    city = models.CharField(max_length=60)
    state_province = models.CharField(max_length=30)
    country = models.CharField(max_length=50)
    website = models.URLField()
    def __str__(self):
        return self.name


class Author(models.Model):
    first_name = models.CharField(max_length=30)
    last_name = models.CharField(max_length=40)
    email = models.EmailField()
    def __str__(self):
        return %s %s % (self.first_name, self.last_name)


class Book(models.Model):
    title = models.CharField(max_length=100)
    authors = models.ManyToManyField(Author)
    publisher = models.ForeignKey(Publisher)
    publication_date = models.DateField()
    def __str__(self):
        return self.title

01-访问外键值

访问 ForeignKey 类型的字段时,得到的是相关的模型对象。

>>> b = Book.objects.get(id=50)
>>> b.publisher
<Publisher: Apress Publishing>
>>> b.publisher.website
http://www.apress.com/

ForeignKey 字段也能反向使用,若想获取指定出版社出版的所有 图书,要使用 publisher.book_set.all()。

>>> p = Publisher.objects.get(name=Apress Publishing)
>>> p.book_set.all()
[<Book: The Django Book>, <Book: Dive Into Python>, ...]

其实,book_set 就是一个 QuerySet 对象,可以过滤和切片。

>>> p = Publisher.objects.get(name=Apress Publishing)
>>> p.book_set.filter(title__icontains=django)
[<Book: The Django Book>, <Book: Pro Django>]

book_set 属性是生成的:把模型名的小写形式与 _set 连在一起。

02-访问多对多值

查看一本的书作者要:

>>> b = Book.objects.get(id=50)
>>> b.authors.all()
[<Author: Adrian Holovaty>, <Author: Jacob Kaplan-Moss>]
>>> b.authors.filter(first_name=Adrian)
[<Author: Adrian Holovaty>]
>>> b.authors.filter(first_name=Adam)
[]

如果想查看一位作者撰写的所有图书,使用 author.book_set,

>>> a = Author.objects.get(first_name=Adrian,last_name=Holovaty)
>>> a.book_set.all()
[<Book: The Django Book>, <Book: Adrians Other Book>]

03-管理器

objects 是个特殊的属性,这是 模型的管理器(manager)。

3.1 添加额外的管理器方法

是为模型添加数据表层功能的首选方式。

下面我们为 Book 模型添加一个管理器方法 title_count(),它的参数是一个关键字,返回书名中 包含关键字的图书数量。

# models.py
from django.db import models
# ... Author 和 Publisher 模型省略了 ...
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 __str__(self):
        return self.title
注意:
    1、我们定义的 BookManager 类扩展 django.db.models.Manager。类中只有一个方法,title_count(),做相
关的计算。注意,这个方法使用了 self.filter(),其中 self 指代管理器自身。
    2、我们把 BookManager() 赋值给模型的 objects 属性。这么做的效果是替换模型的“默认”管理器,即未指
定管理器时自动创建的 objects。我们仍把它叫做 objects,以便与自动创建的管理器保持一致。

创建好管理器之后,可以像下面这样使用:
Book.objects.title_count(django)

为的是封装经常执行的查询,以免代码重复。

3.2 修改管理器返回的查询集合

管理器的基本查询集合返回系统中的所有对象。如:Book.objects.all() 返回数据库中的所有图书。

若想 覆盖管理器的基本查询集合,覆盖 Manager.get_queryset() 方法。get_queryset() 方法应该返回一个 QuerySet,包含所需的属性。

下述模型有两个管理器,一个返回所有对象,另一个只返回 Joy 写的书。

from django.db import models
# 首先,定义 Manager 子类
class JoyBookManager(models.Manager):
    def get_queryset(self):
        return super(JoyBookManager, self).get_queryset().filter(author=Joy)

# 然后,放入 Book 模型
class Book(models.Model):
    title = models.CharField(max_length=100)
    author = models.CharField(max_length=50)
    # ...
    objects = models.Manager() # 默认的管理器
    joy_objects = JoyBookManager() # 专门查询 Dahl 的管理器

对这个示例模型来说,Book.objects.all() 返回数据库中的所有图书,而 Book.joy_objects.all() 只返回 Joy写的书。

这个示例还指出了另一个有用的技术: 在同一个模型上使用多个管理器。只要需要,可以为模型添加任意个 Manager() 实例。这么做,可以轻易为模型定义常用的“过滤器”。

class MaleManager(models.Manager):
    def get_queryset(self):
        return super(MaleManager, self).get_queryset().filter(sex=M)


class FemaleManager(models.Manager):
    def get_queryset(self):
        return super(FemaleManager, self).get_queryset().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)
                           ))class MaleManager(models.Manager):
        def get_queryset(self):
            return super(MaleManager, self).get_queryset().filter(sex=M)
    class FemaleManager(models.Manager):
        def get_queryset(self):
            return super(FemaleManager, self).get_queryset().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 遇到的第一个管理器。

Django 把它解释的第一个管理器定义为“默认的”管理器,而且 Django 在很多地方(管理后台不在此 列)只使用那个管理器。

鉴于此,通常最好小心选择默认的管理器,以防把 get_queryset() 返回的结果覆盖掉,无法检索所需的对象。

04-模型方法

管理器的作用是执行数据表层的操作,而模型方法处理的是具体的模型实例。

from django.db import models
    class Person(models.Model):
        first_name = models.CharField(max_length=50)
        last_name = models.CharField(max_length=50)
        birth_date = models.DateField()

    def baby_boomer_status(self):
        # 返回一个人的出生日期与婴儿潮的关系
        import datetime
        if self.birth_date < datetime.date(1945, 8, 1):
            return "Pre-boomer"
        elif self.birth_date < datetime.date(1965, 1, 1):
            return "Baby boomer"
        else:
            return "Post-boomer"
        
    def _get_full_name(self):
        # 返回一个人的全名
        return %s %s % (self.first_name, self.last_name)
    full_name = property(_get_full_name)
各个模型自动具有的方法,其中几个最常定义:
1、__str__()。这是 Python 的一个“魔法方法”,返回对象的 Unicode 表示形式。
2、get_absolute_url()。这个方法告诉 Django 如何计算一个对象的 URL。Django 在管理后台和需要生成 对象的 URL 时调用这个方法。具有唯一标识的 URL 的对象都要定义这个方法。

4.1 覆盖预定义的模型方法

一系列封装数据库行为的模型方法有时也需要自定义。尤其是 save() 和 delete(),经常需要修改它们的运作方式。

如:

from django.db import models
class Blog(models.Model):
    name = models.CharField(max_length=100)
    tagline = models.TextField()
    def save(self, *args, **kwargs):
        do_something()
        super(Blog, self).save(*args, **kwargs) # 调用“真正的”save () 方法 
        do_something_else()

一定要记得调用超类中的方法,即 super(Blog, self).save(*args, **kwargs),确保把对象保存到数据库中。

05-执行原始 SQL

模型的查询 API 不够用时,可以编写原始 SQL。Django 为执行原始 SQL 查询提供了两种方式:使用 Manag- er.raw() 执行,返回模型实例集合;或者完全不用模型层,直接执行自定义的 SQL。

注意:编写原始 SQL 时要非常小心。一定要正确转义通过 params 传入的参数,以防 SQL 注入攻击。

06-执行原始查询

管理器的 raw() 方法用于执行原始的 SQL 查询,其返回结果是模型实例集合:

Manager.raw(raw_query, params=None, translations=None)

这个方法的参数是一个原始的 SQL 查询,执行后返回一个 django.db.models.query.RawQuerySet 实例。

实例可以像常规的 QuerySet 对象一样迭代,获取里面的模型对象。

for p in Person.objects.raw(SELECT * FROM myapp_person):
    print(p)

6.1 模型对应的表名

上例中的 Person 表名(myapp_person)是怎么来的呢?Django 默认把“应用标注”(manage.py startapp 命令指定的名称)与类 名使用下划线联结在一起得到数据库表名。在上述示例中,我们假设 Person 模型在 myapp 应用中,因此对应 的表是 myapp_person。

6.2 把查询中的字段映射到模型字段上

1、raw() 自动把查询中的字段映射到模型字段上。查询中的字段顺序无关紧要。
Person.objects.raw(SELECT id, first_name, last_name, birth_date FROM myapp_person)

2、只要名称匹配就能正确创建模型实例。
此外,还可以使用 raw() 方法的 translations 参数把查询中的字段映射到模型字段上。这个参数的值是一个字典,把查询中的字段名称映射到模型字段的名称上。
>>> name_map = {first: first_name, last: last_name, bd: birth_date, pk: id}
>>> Person.objects.raw(SELECT * FROM some_other_table, translations=name_map)

3、索引查找
raw() 支持索引,因此如果只想获得第一个结果,可以这样写:
>>> first_person = Person.objects.raw(SELECT * FROM myapp_person)[0]
然而,索引和切片不是在数据库层执行的。如果数据库中的 Person 对象很多,最好在 SQL 查询中限制数量:
>>> first_person = Person.objects.raw(SELECT * FROM myapp_person LIMIT 1)[0]

4、 延期模型字段
还可以把字段排除在外:
>>> people = Person.objects.raw(SELECT id, first_name FROM myapp_person)
这个查询返回的是延期的 Person 对象。这意味着,查询排除的字段将按需加载。如:
for p in Person.objects.raw(SELECT id, first_name FROM myapp_person):
    print(p.first_name, # 这个属性由查询取回
            p.last_name) # 这个属性按需取回
从表面看,好像这个查询把名字和姓都取回了。然而,这个示例其实发起了三个查询。raw() 执行的查询只 取回名字,两人的姓在打印时按需取回。
只有一个字段是不能排除的——主键字段。Django 使用主键标识模型实例,因此必须始终包含在原始查询 中。忘记主键时,抛出 InvalidQuery 异常。

5、添加注解
执行的查询还可以包含模型中没有定义的字段。例如,可以使用 PostgreSQL 的 age() 函数让数据库计算所得诸人的年龄:
>>> people = Person.objects.raw(SELECT *, age(birth_date) AS age FROM myapp_person)
>>> for p in people:
    ...
    print("%s is %s." % (p.first_name, p.age))
John is 37.
Jane is 42.
...

6、为 raw() 传递参数
如果想执行参数化查询,可以把 params 参数传给 raw():
>>> lname = Doe
>>> Person.objects.raw(SELECT * FROM myapp_person WHERE last_name = %s, [lname])
params 的值是一个参数列表或字典。使用列表时,查询字符串中的占位符是 %s;使用字典时,占位符是 %(key)s
注意:原始查询中不要使用字符串格式化!
>>> query = SELECT * FROM myapp_person WHERE last_name = %s % lname
 Person.objects.raw(query)
千万别这么写!

07-直接执行自定义的 SQL

django.db.connection 对象表示默认的 数据库连接。若想使用这个数据库连接,调用 connection.cursor(),获取一个游标对象。

然后,调用 cur- sor.execute(sql, [params]) 执行 SQL,再调用 cursor.fetchone() 或 cursor.fetchall() 返回所得的行。

from django.db import connection
    def my_custom_sql(self):
        cursor = connection.cursor()
        cursor.execute("UPDATE bar SET foo = 1 WHERE baz = %s", [self.baz])
        cursor.execute("SELECT foo FROM bar WHERE baz = %s", [self.baz])
        row = cursor.fetchone()
        return row

注意,传入参数时,如果查询中有百分号,应该编写两个百分号:
cursor.execute("SELECT foo FROM bar WHERE baz = ‘30%‘")
cursor.execute("SELECT foo FROM bar WHERE baz = ‘30%%‘ AND id = %s", [self.id])

使用多个数据库时,可以使用 django.db.connections 获取指定数据库的连接(和游标)。django.db.connec-
tions 是一个类似字典的对象,可以使用别名取回指定连接:
from django.db import connections
cursor = connections[my_db_alias].cursor()
# 其他代码...

7.1 连接和游标

要注意,cursor.execute() 中的 SQL 语句使用占位符 %s,而不直接把参数添加到 SQL 查询中。

使用占位符时,底层数据库库会自动转义参数。还要注意,Django 使用的占位符是 %s,而不是 ?。

7.2 添加额外的管理器方法

添加额外的管理器方法是为模型添加数据表层功能的首选方式。自定义的管理器方法可以返回任何需要的内容,不一定是 QuerySet。

例如,下面这个自定义的管理器提供了 with_counts() 方法,它返回所有 OpinionPoll 对象,每个对象都有额 外的 num_responses 属性,其值为聚合查询的结果:

from django.db import models

class PollManager(models.Manager):
    def with_counts(self):
        from django.db import connection

    cursor = connection.cursor()
    cursor.execute("""
        SELECT p.id, p.question, p.poll_date, COUNT(*)
        FROM polls_opinionpoll p, polls_response r
        WHERE p.id = r.poll_id
        GROUP BY p.id, p.question, p.poll_date
        ORDER BY p.poll_date DESC""")
    result_list = []
    for row in cursor.fetchall():
        p = self.model(id=row[0], question=row[1], poll_date=row[2])
        p.num_responses = row[3]
        result_list.append(p)
    return result_list


class OpinionPoll(models.Model):
    question = models.CharField(max_length=200)
    poll_date = models.DateField()
    objects = PollManager()


class Response(models.Model):
    poll = models.ForeignKey(OpinionPoll)
    person_name = models.CharField(max_length=50)
    response = models.TextField()

对这个示例来说,要使用 OpinionPoll.objects.with_counts() 获取具有 num_responses 属性的 OpinionPoll 对 象列表。还有一点要注意:管理器方法可以访问 self.model,获取所依附的模型类。

 

Django之模型的高级用法

标签:cti   stat   赋值   多个   raw   完全   ati   ini   elf   

原文地址:https://www.cnblogs.com/pgxpython/p/11698218.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!