From a5826ae33667c67d424376a45edb59ca80c31c6f Mon Sep 17 00:00:00 2001
From: Jakub Trllo <jakub.trllo@gmail.com>
Date: Fri, 22 Apr 2022 18:27:39 +0200
Subject: [PATCH] reorganized code a little bit

---
 openpype/plugins/publish/collect_scene_loaded_versions.py | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/openpype/plugins/publish/collect_scene_loaded_versions.py b/openpype/plugins/publish/collect_scene_loaded_versions.py
index 7b44aa7963..ffdd532df2 100644
--- a/openpype/plugins/publish/collect_scene_loaded_versions.py
+++ b/openpype/plugins/publish/collect_scene_loaded_versions.py
@@ -38,9 +38,13 @@ class CollectSceneLoadedVersions(pyblish.api.ContextPlugin):
         loaded_versions = []
         _containers = list(host.ls())
         _repr_ids = [ObjectId(c["representation"]) for c in _containers]
+        repre_docs = io.find(
+            {"_id": {"$in": _repr_ids}},
+            projection={"_id": 1, "parent": 1}
+        )
         version_by_repr = {
-            str(doc["_id"]): doc["parent"] for doc in
-            io.find({"_id": {"$in": _repr_ids}}, projection={"parent": 1})
+            str(doc["_id"]): doc["parent"]
+            for doc in repre_docs
         }
 
         # QUESTION should we add same representation id when loaded multiple