【问题标题】:使用 Django ORM 从相关表中获取数据
【发布时间】:2021-02-26 00:19:30
【问题描述】:

使用 Django ORM,如何在不有效地对每条记录进行单独调用(或冗余非规范化数据以使其更易于访问)的情况下访问相关表中的数据?

假设我有 3 个模型:

class Tournament(models.Model):
    name = models.CharField(max_length=250) 
    active = models.BooleanField(null=True,default=1)
    
class Team(models.Model):
    name = models.CharField(max_length=250)
    coach_name = models.CharField(max_length=250)
    active = models.BooleanField(null=True,default=1)
    
class Player(models.Model):
    user = models.ForeignKey(
        settings.AUTH_USER_MODEL,
        on_delete=models.DO_NOTHING
    )
    number = models.PositiveIntegerField()
    age = models.PositiveIntegerField()
    active = models.BooleanField(null=True,default=1)

请注意,此 Player 模型在应用程序中很重要,因为它是大多数模型的主要连接 - 从注册到团队到统计数据到结果再到奖品。但是这个 Player 模型实际上并不包含人名,因为该模型包含一个用户字段,该字段是自定义 AUTH_USER_MODEL ('user') 模型的外键,该模型包含名字/姓氏信息。这允许玩家登录应用程序并执行某些操作。

除了这些基本模型,假设一个玩家可以在不同的比赛中为不同的球队效力,我还有一个连接的 ManyToMany 模型:

class PlayerToTeam(models.Model):
    player = models.ForeignKey(
        Player,
        on_delete=models.DO_NOTHING
    )
    team = models.ForeignKey(
        Team,
        on_delete=models.DO_NOTHING
    )
    tournament = models.ForeignKey(
        Tournament,
        on_delete=models.DO_NOTHING
    )       

作为我遇到的挑战之一的示例,假设我正在尝试创建一个表格,允许教练选择他们的首发阵容。所以我需要我的表格来列出特定锦标赛中特定球队的球员姓名。

鉴于锦标赛和团队 ID,我可以轻松地拉回必要的 QuerySet 来描述我感兴趣的初始记录。

playersOnTeam = PlayerToTeam.objects.filter(tournament=[Tournament_id]).filter(team=[Team_id])

这将返回球队、锦标赛和球员的 ID(但只有 ID)的 QuerySet。但是,名称数据有两个模型:

PlayerToTeam->[player_id]->Player->[user_id]->User->[first_name] [last_name]

现在,如果我只拉回一条记录,我可以这样做

onlyPlayerOnTeam = PlayerToTeam.objects.filter(tournament=[Tournament_id]).filter(team=[Team_id]).filter(player=[Player_id]).get()

onlyPlayerOnTeam.player.user.first_name

因此,如果我只需要显示名称,我相信我可以在视图返回中传递 QuerySet 并在模板中循环它并显示我需要的内容。但是当我需要将名称显示为表单的一部分时,我不知道您是否可以做类似的事情。

为了填充表单,我相信我可以遍历初始 QuerySet 并构建一个新的数据结构:

playersOnTeam = PlayerToTeam.objects.filter(tournament=[Tournament_id]).filter(team=[Team_id])

allPlayersData= []
for nextPlayer in playersOnTeam:
    playerDetails= {
        "player_id": nextPlayer.player.id,
        "first_name": nextPlayer.player.user.first_name, 
        "last_name": nextPlayer.player.user.last_name,
    }
    allPlayersData.append(playerDetails)

form = StartingLineupForm(allPlayersData)

但是,我担心这会导致每个玩家/用户都调用单独的数据库!

虽然这对于 6 到 10 名玩家来说可能是可以容忍的,但对于更大的数据集,这似乎不太理想。循环为每个用户执行查询似乎是完全错误的。

此外,令人沮丧的是,使用直接的 SQL 查询就足够简单了:

SELECT User.first_name, User.last_name 
FROM PlayerToTeam 
INNER JOIN Player ON PlayerToTeam.player_id = Player.id 
INNER JOIN User ON Player.user_id = User.id 
WHERE PlayerToTeam.tournament_id=[tourney_id] AND PlayerToTeam.team_id=[team_id]

但我正在尝试尽可能地坚持 Django ORM 最佳实践,并避免在无法立即解决问题时直接使用 SQL 查询,而且我几乎可以肯定这不是情况如此复杂,如果不使用直接 SQL 查询,我就无法完成。

我开始关注select_relatedprefetch_related,但我很难理解它们是如何用于关系的,而不仅仅是一个表连接。好像我可以使用预取访问Player.age 数据,但我不知道如何从中获取User.first_name

任何帮助将不胜感激。

【问题讨论】:

    标签: python django django-models


    【解决方案1】:
    1. 我建议两种方法:

    A) 选择相关(一个数据库查询):

    objects = PlayerToTeam.objects.filter(
        ...
    ).select_related(
        'player__user',
    ).only('player__user__name')
    
    name = objects.first().user.name
    

    B) 注释(一个数据库查询):

    objects = PlayerToTeam.objects.filter(
        ...
    ).annotate(
        player_name=F('player__user__name'),
    )
    
    name = objects.first().player_name
    
    1. 为确保您只有一个特定的playerteamtournament 对象,我建议添加unique_together
    class PlayerToTeam(models.Model):
        ...
    
        class Meta:
            unique_together = ('player', 'team', 'tournament', )
    

    【讨论】:

    • 感谢您的回复。我确实有 unique_together 设置。所以我很感谢你确认这一点。那么我只需要遍历 QuerySet 并构建数据结构以传递给表单类,对吗?并且通过使用 select_related,它不会为每条记录执行单独的查询来获取名称吗?如果我可以再问一个(稍微相关的)问题,是否有一种相对简单的方法来确定每个请求调用了多少实际查询?
    • 我建议使用with self.assertNumQueries(10):为它编写一个单元测试,当这个测试失败时,它会打印整个数据库查询列表。
    • 我发现这篇文章有一个装饰器,似乎可以输出时间和查询次数:medium.com/better-programming/… 现在我只需要弄清楚如何使用装饰器。 :)
    猜你喜欢
    • 1970-01-01
    • 2020-06-14
    • 2018-09-25
    • 1970-01-01
    • 2014-11-05
    • 1970-01-01
    • 2018-12-22
    • 1970-01-01
    • 1970-01-01
    相关资源
    最近更新 更多