cli/vms: serialize uuid already in the api

2023-09-27 11:11:30 +02:00
parent 08eab785c6
commit dea49073cb
2 changed files with 6 additions and 16 deletions
--- a/pkgs/clan-cli/clan_cli/webui/task_manager.py
+++ b/pkgs/clan-cli/clan_cli/webui/task_manager.py
@@ -82,13 +82,9 @@ class TaskPool:
        self.lock: threading.RLock = threading.RLock()
        self.pool: dict[UUID, BaseTask] = {}

-    def __getitem__(self, uuid: str | UUID) -> BaseTask:
+    def __getitem__(self, uuid: UUID) -> BaseTask:
        with self.lock:
-            if type(uuid) is UUID:
-                return self.pool[uuid]
-            else:
-                uuid = UUID(uuid)
-                return self.pool[uuid]
+            return self.pool[uuid]

    def __setitem__(self, uuid: UUID, task: BaseTask) -> None:
        with self.lock: