博客

使用Django实现把两个模型类的数据聚合在一起

  • Post category:Python

使用Django实现把两个模型类的数据聚合在一起，主要有以下三种实现方式：

1. 使用ORM ForeignKey

在Django ORM中，如果存在两个模型类A和B，其中A中有一个字段B类的对象，那么我们可以通过ForeignKey来实现这种关联。在模板中使用select_related或者prefetch_related来获取聚合数据。

示例代码：

class Author(models.Model):
    name = models.CharField(max_length=100)

class Book(models.Model):
    title = models.CharField(max_length=100)
    author = models.ForeignKey(Author, on_delete=models.CASCADE, related_name='books')

# 获取作者及其对应的所有书籍
authors = Author.objects.prefetch_related('books').all()
for author in authors:
    print(f'{author.name}的书籍：')
    for book in author.books.all():
        print(book.title)

2. 使用ORM ManyToManyField

在Django ORM中，如果存在两个模型类A和B，这两个模型之间存在多对多的关联，那么我们可以通过ManyToManyField来实现这种关联。同样的，在模板中使用select_related或者prefetch_related来获取聚合数据。

示例代码：

class Author(models.Model):
    name = models.CharField(max_length=100)

class Book(models.Model):
    title = models.CharField(max_length=100)
    authors = models.ManyToManyField(Author, related_name='books')

# 获取书籍及其对应的所有作者
books = Book.objects.prefetch_related('authors').all()
for book in books:
    print(f'{book.title}的作者：')
    for author in book.authors.all():
        print(author.name)

3. 使用Django的聚合函数

在一些场景下，我们需要将两个模型之间的某一个字段进行聚合，例如统计某个作者的所有书籍的总价格。这个时候可以使用Django的聚合函数来实现。

示例代码：

from django.db.models import Sum

class Author(models.Model):
    name = models.CharField(max_length=100)

class Book(models.Model):
    title = models.CharField(max_length=100)
    author = models.ForeignKey(Author, on_delete=models.CASCADE, related_name='books')
    price = models.DecimalField(max_digits=8, decimal_places=2)

# 统计每个作者的所有书籍的价格总和
authors = Author.objects.annotate(total_price=Sum('books__price')).all()
for author in authors:
    print(f'{author.name}的书籍总价：{author.total_price}')

以上三种方式都可以实现将两个模型的数据聚合在一起，具体选择哪一种方式，需要根据具体的需求情况来决定。

Tags: ,