Django, select_relatedとprefetch_relatedの組み合わせ

Question

djangoにおけるクエリの生成についての質問です。

class Group(Model):
  name = models.CharField("名称",max_length=100)

class User(Model):
  name = models.CharField("名称",max_length=100)
  group = models.ForeignKey(Group, related_name="user_group")

class Task(Model):
  name = models.CharField("名称",max_length=100)
  group = models.ForeignKey(Group, related_name="task_group")
  status = models.IntegerField(default=0)

上記のようなモデル構成だとします。
この場合、関係性は User - Group - Task = many - 1 - many
となりますが、ここでユーザー起点で見たときにタスクに辿り着かせる方法を模索しています。

まずUserとGroupを結びつけるために

users = User.objects.select_related("group").all()
```とし、次にGroupの中にTaskを含むために
```ここに言語を入力
#groupに対して付与するprefetch要素
tasks = Prefetch("task_group",
                 queryset=Task.objects.filter(status=0),
                 to_attr="tasks")
```とすることで要素としては完成できました。
しかしこの後にこの二つを組み合わせてuserから参照する方法がわからず困っています。

理想的な形としてはそれぞれのユーザーが
```python
for user in users:
  print (user.tasks)
  #または
  print (user.group.tasks)

とすることでタスクにアクセスできるようにしたいのですが...

どなたか良い解決策を知っていましたら、ご教授いただければと思います。

Comments

[gh640]

回答させていただきましたが、問題は解決しませんか？

Answer 1

Django 2.1.x と仮定して回答させていただきますね。

ForeignKey を定義するときに related_name をセットされているので、その名前で参照を逆方向に辿ることができます。

つまり、今回の場合ですと、次の形でそのユーザにひもづく Task オブジェクトのリストが取得できるものと思います。

for user in users:
  print(user.group.task_group)

公式ドキュメントの次のところあたりにその説明が載っていますので、よろしければご覧ください。

• https://docs.djangoproject.com/en/2.1/ref/models/fields/#django.db.models.ForeignKey
• https://docs.djangoproject.com/en/2.1/ref/models/fields/#django.db.models.ForeignKey.related_name