Djangoのmodelから任意のデータを取得してリストに格納する際のパフォーマンス向上方法

前提・実現したいこと

DjangoのQuerySetからデータを1件ずつ取得してdict型要素としてlistに格納する処理を実装しているのですが、
パフォーマンスを向上させたいと考えています。

該当のソースコード

python
1# SearchModel:データ取得対象のmodel
2# result_list:取得結果格納用のlist
3# search_id:SearchModelから取得したいデータのid
4from .models import SearchModel
5result_list = []
6search_id = [1, 9, 6, 10, 2]
7
8for id in search_id:
9    result_list.append(SearchModel.objects.get(id=id))

上記のコードだとsearch_idの個数分クエリが発行されてしまうためパフォーマンスが悪いです。

試したこと

最初にmodelを全件取得してdict型として保持した後にsearch_idの要素のみ抜いてくるという方法も考えましたが、
結局search_idの個数分のループを回すことになるため、
何か他の方法はないか検討しております。
ご教示宜しくお願い致します。

補足情報（FW/ツールのバージョンなど）

python 3.7
Django 2.2

行動規範の内容に同意します

回答1件

ベストアンサー

フィールドルックアップのinを使用するのはいかがでしょうか？

python
1search_ids = [1, 9, 6, 10, 2, ]
2SearchModel.objects.get(id__in=search_ids)

投稿2020/04/22 12:31

hasami

総合スコア1277

tanana_m

2020/04/22 12:40

ご回答ありがとうございます。この場合取得データのi順番は、search_idsの要素の順番の通りになるのでしょうか。（最初はid=1のレコード、、次はid=9のレコード…）明記していなかったのですが、取得したデータの並び順はsearch_idの順番通りとする必要があります。説明不足で大変申し明けありません。