From 8422469ca6afd29d0bda9e95899e54d492ff543c Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Cem=20G=C3=B6kmen?=
 <1408354+cgokmen@users.noreply.github.com>
Date: Tue, 2 May 2023 12:37:31 -0700
Subject: [PATCH] Make prefetch happen only once for all the scene data

---
 data/models.py | 13 +++++++++----
 1 file changed, 9 insertions(+), 4 deletions(-)

diff --git a/data/models.py b/data/models.py
index 1bca52e7..90a7fcc6 100644
--- a/data/models.py
+++ b/data/models.py
@@ -49,14 +49,19 @@
 ]
 
 
+class SceneManager(models.Manager):
+    def get_queryset(self):
+        return super().get_queryset().prefetch_related("room_set__roomobject_set__object__category__synset")
+
+    
 class Scene(models.Model):
     name = models.CharField(max_length=64, primary_key=True)
+    objects_with_data = SceneManager()
 
     def __str__(self):
         return self.name 
     
 
-
 class Category(models.Model):
     name = models.CharField(max_length=64, primary_key=True)
     # the synset that the category belongs to
@@ -200,11 +205,11 @@ def synset_state(self) -> str:
     @cached_property
     def problem_synsets(self):
         return self.illegal_synsets | self.synsets.filter(state=STATE_UNMATCHED)
-    
+       
     @cached_property
     def scene_matching_dict(self) -> Dict[str, Dict[str, str]]:
         ret = {status: {} for status in ["matched", "planned", "unmatched"]}
-        for scene in Scene.objects.prefetch_related("room_set__roomobject_set__object__category__synset").all():
+        for scene in Scene.objects_with_data.all():
             # first check whether it can be matched to the task in the future
             result = self.matching_scene(scene=scene, ready=False)
             # if it is matched, check whether it can be matched to the task it its current state
@@ -325,4 +330,4 @@ class Meta:
         unique_together = ('room', 'object')
 
     def __str__(self):
-        return f"{str(self.room)}_{self.object.name}"   
\ No newline at end of file
+        return f"{str(self.room)}_{self.object.name}"