cache root folder

Author: eric-intuitem

backend/iam/cache_builders.py

@@ -64,6 +64,7 @@ class FolderCacheState:
     children_map: Mapping[Optional[uuid.UUID], Tuple[uuid.UUID, ...]]
     depth_map: Mapping[uuid.UUID, int]
     root_ids: Tuple[uuid.UUID, ...]
+    root_folder_id: Optional[uuid.UUID]
 
 
 def build_folder_cache_state() -> FolderCacheState:
@@ -84,8 +85,14 @@ def build_folder_cache_state() -> FolderCacheState:
     }
 
     children_map: defaultdict[Optional[uuid.UUID], List[uuid.UUID]] = defaultdict(list)
+    root_folder_id: Optional[uuid.UUID] = None
     for folder in folders:
         children_map[folder.parent_folder_id].append(folder.id)
+        if (
+            root_folder_id is None
+            and folder.content_type == Folder.ContentType.ROOT
+        ):
+            root_folder_id = folder.id
 
     # Stable ordering for traversal
     for child_list in children_map.values():
@@ -112,6 +119,7 @@ def _depth(current_id: uuid.UUID) -> int:
         ),
         depth_map=MappingProxyType(depth_map),
         root_ids=tuple(children_map.get(None, ())),
+        root_folder_id=root_folder_id,
     )
 
 

backend/iam/models.py

@@ -101,14 +101,20 @@ class Folder(NameDescriptionMixin):
     """
 
     @staticmethod
-    def get_root_folder() -> Self:
+    def get_root_folder() -> Self | None:
         """class function for general use"""
+        try:
+            state = get_folder_state()
+            root_id = getattr(state, "root_folder_id", None)
+            if root_id:
+                return state.folders.get(root_id)
+        except Exception:
+            pass
         return _get_root_folder()
 
     @staticmethod
     def get_root_folder_id() -> uuid.UUID | None:
-        root = _get_root_folder()
-        return root.id if root else None
+        return getattr(Folder.get_root_folder(), "id", None)
 
     class ContentType(models.TextChoices):
         """content type for a folder"""
@@ -980,7 +986,7 @@ def is_object_readable(
         )
 
     @staticmethod
-    def get_accessible_folders(
+    def get_accessible_folder_ids(
         folder: Folder,
         user: User,
         content_type: Folder.ContentType,

backend/iam/views.py

@@ -211,7 +211,7 @@ def get(self, request) -> Response:
         user_groups_data = list(request.user.user_groups.values("name", "builtin"))
         user_groups = [(ug["name"], ug["builtin"]) for ug in user_groups_data]
 
-        accessible_domains = RoleAssignment.get_accessible_folders(
+        accessible_domains = RoleAssignment.get_accessible_folder_ids(
             Folder.get_root_folder(), request.user, Folder.ContentType.DOMAIN
         )