Trying to construct modelspace

modelspace/ModelPackageSubRepository.py

@@ -0,0 +1,95 @@
+import uuid
+from pathlib import Path
+
+from modelspace.Essentials import SetsDict
+from modelspace.ModelPackage import ModelPackage
+
+
+class ModelPackageSubRepository:
+    def __init__(self, path, seed):
+        path.mkdir(exist_ok=True)
+        self.path = Path(path)
+        self.seed = seed
+        self.packages: dict[str, ModelPackage] | None = None
+        self.resources: SetsDict | None = None
+        self.reload()
+
+    #   Completed
+    def _reload_packages(self):
+        self.packages = dict()
+        try:
+            dirs = [item.name for item in self.path.iterdir() if item.is_dir()]
+        except OSError as e:
+            print(f"Ошибка доступа к директории: {e}")
+            dirs = []
+
+        for d in dirs:
+            package = ModelPackage.load(str(self.path / d))
+            self.packages[package.uuid] = package
+
+    #   Completed
+    def _reload_resources(self):
+        self.resources = SetsDict()
+
+        for pkg_id, package in self.packages.items():
+            for resource in package.provides:
+                self.resources.add(resource, pkg_id)
+
+    def reload(self):
+        self._reload_packages()
+        self._reload_resources()
+
+
+    # debugged
+    def resources_from_pkg_list(self, uuids: list[str]):
+        selected_packages = []
+        for pkg_id in uuids:
+            package = self.packages.get(pkg_id, None)
+            selected_packages.append(package)
+
+        resources = SetsDict()
+        for package in selected_packages:
+            for resource in package.provides:
+                resources.add(resource, package.uuid)
+
+        return resources
+
+    @staticmethod
+    def deps_from_pkg_list(packages: list[ModelPackage]) -> set[str]:
+        res = set()
+        for package in packages: res = res | set(package.dependencies)
+        return res
+
+    # debugged
+    def packages_by_resource(self, resource):
+        packages_ids = self.resources.by_key(resource)
+
+        if not packages_ids or len(packages_ids) == 0:
+            raise RuntimeError(f"{resource}: There are no packages in the repository that provide this resource")
+        else:
+            packages_ids = list(packages_ids)
+
+        packages: set[ModelPackage] = set()
+        for pkg_id in packages_ids: packages.add(self.package_by_id(pkg_id))
+        return packages
+
+    # debugged
+    def package_by_id(self, pkg_id):
+        package = self.packages.get(pkg_id, None)
+        if not package: raise RuntimeError(f"{pkg_id}: Something went wrong while reading package info")
+        return package
+
+
+    def add_package_interactive(self) -> ModelPackage:
+        """Добавляет новый пакет модели интерактивно"""
+        # Генерируем новый UUID
+        package_uuid = str(uuid.uuid4())
+
+        # Создаем путь к новому пакету
+        package_path = self.path / package_uuid
+
+        # Вызываем интерактивное создание пакета
+        package = ModelPackage.interactive(str(package_path), package_uuid)
+        loaded_package = ModelPackage.load(str(package_path))
+        self.packages[loaded_package.uuid] = loaded_package
+        return package