core: kraken: keep track of version id for extensions

bluerobotics · Sep 25, 2023 · 3f96de9 · 3f96de9
1 parent 73982ef
commit 3f96de9
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 5 deletions.
diff --git a/core/services/kraken/kraken.py b/core/services/kraken/kraken.py
@@ -118,6 +118,7 @@ async def install_extension(self, extension: Any) -> AsyncGenerator[bytes, None]
         except Exception as e:
             # this will fail if the container is not installed, we don't mind it
             logger.info(e)
+
         new_extension = Extension(
             identifier=extension.identifier,
             name=extension.name,
@@ -126,7 +127,7 @@ async def install_extension(self, extension: Any) -> AsyncGenerator[bytes, None]
             permissions=extension.permissions,
             enabled=extension.enabled,
             user_permissions=extension.user_permissions,
-            uuid=extension.uuid,
+            id=extension.id,
         )
         self.settings.extensions.append(new_extension)
         self.manager.save()
@@ -198,7 +199,7 @@ async def update_extension_to_version(self, identifier: str, version: str) -> As
             enabled=True,
             # TODO: handle user permissions on updates
             user_permissions="",
-            uuid=extension.uuid,
+            id=extension.id,
         )
 
         # Remove older entry if it exists

diff --git a/core/services/kraken/main.py b/core/services/kraken/main.py
@@ -25,7 +25,7 @@ class Extension(BaseModel):
     enabled: bool
     identifier: str
     user_permissions: str
-    uuid: Optional[str] = None
+    id: Optional[str] = None
 
     def is_valid(self) -> bool:
         return all([self.name, self.docker, self.tag, any([self.permissions, self.user_permissions]), self.identifier])
@@ -67,7 +67,7 @@ async def get_installed_extensions() -> Any:
             permissions=extension.permissions,
             enabled=extension.enabled,
             user_permissions=extension.user_permissions,
-            uuid=extension.uuid,
+            id=extension.id,
         )
         for extension in extensions
     ]

diff --git a/core/services/kraken/settings.py b/core/services/kraken/settings.py
@@ -17,7 +17,7 @@ class Extension(JsonObject):
     permissions = StringField()
     enabled = BooleanField()
     user_permissions = StringField()
-    uuid = StringField(null=True)
+    id = StringField(null=True)
 
     def settings(self) -> Any:
         if self.user_permissions: