2025-12-01

scripts/addons/cache_manager/models.py

@@ -0,0 +1,85 @@
+# SPDX-FileCopyrightText: 2021 Blender Studio Tools Authors
+#
+# SPDX-License-Identifier: GPL-3.0-or-later
+
+from pathlib import Path
+from typing import Optional, List, Tuple
+
+from .logger import LoggerFactory
+
+logger = LoggerFactory.getLogger(__name__)
+
+
+class FolderListModel:
+    def __init__(self):
+        self.__root_path: Optional[Path] = None
+        self.__folders: List[str] = []
+        self.__appended: List[str] = []
+        self.__combined: List[str] = []
+
+    def rowCount(self) -> int:
+        return len(self.__combined)
+
+    def data(self, row: int) -> Optional[str]:
+        if len(self.__combined) > 0:
+            return self.__combined[row]
+
+        return None
+
+    @property
+    def root_path(self) -> Optional[Path]:
+        return self.__root_path
+
+    @root_path.setter
+    def root_path(self, path: Path) -> None:
+
+        if not path or not path.absolute().exists():
+            logger.debug("Invalid path: %s", str(path))
+            self.reset()
+        else:
+            self.__root_path = path
+            logger.debug("FolderListModel root path  was set to %s", path.as_posix())
+            self.__load_dir(self.__root_path)
+
+    def reset(self) -> None:
+        self.__root_path = None
+        self.__folders.clear()
+        self.__appended.clear()
+        self.__update_combined()
+
+    def reload(self) -> None:
+        self.__folders.clear()
+        self.__appended.clear()
+        self.root_path = self.__root_path
+
+    def __load_dir(self, path: Path) -> None:
+        self.__folders = self.__detect_folders(path)
+        self.__appended.clear()
+        self.__update_combined()
+
+    def __detect_folders(self, path: Path) -> List[str]:
+        if path.exists() and path.is_dir():
+            # Iterate through directory and return all pathes that are dirs, only return their name.
+            return sorted(
+                [str(x.name) for x in path.iterdir() if x.is_dir()], reverse=True
+            )
+        else:
+            return []
+
+    def append_item(self, item: str) -> None:
+        self.__appended.append(item)
+        self.__update_combined()
+
+    def __update_combined(self) -> None:
+        self.__combined.clear()
+        self.__combined.extend(
+            sorted(list(set(self.__folders + self.__appended)), reverse=True)
+        )
+
+    @property
+    def items(self) -> List[str]:
+        return self.__combined
+
+    @property
+    def items_as_enum_list(self) -> List[Tuple[str, str, str]]:
+        return [(item, item, "") for item in self.__combined]