Refactor watchers

[pti/o2.git] / o2ims / service / watcher / resourcepool_watcher.py

diff --git a/o2ims/service/watcher/resourcepool_watcher.py b/o2ims/service/watcher/resourcepool_watcher.py
index 6343292..dc7745a 100644 (file)
--- a/o2ims/service/watcher/resourcepool_watcher.py
+++ b/o2ims/service/watcher/resourcepool_watcher.py
@@ -13,8 +13,10 @@
 #  limitations under the License.\r
 \r
 from o2ims.service.client.base_client import BaseClient\r
-from o2ims.service.unit_of_work import AbstractUnitOfWork\r
+# from o2ims.service.unit_of_work import AbstractUnitOfWork\r
 from o2ims.service.watcher.base import BaseWatcher\r
+from o2ims.domain import commands\r
+from o2ims.service.messagebus import MessageBus\r
 \r
 from o2common.helper import o2logging\r
 logger = o2logging.get_logger(__name__)\r
@@ -22,8 +24,8 @@ logger = o2logging.get_logger(__name__)
 \r
 class ResourcePoolWatcher(BaseWatcher):\r
     def __init__(self, client: BaseClient,\r
-                 uow: AbstractUnitOfWork) -> None:\r
-        super().__init__(client, uow)\r
+                 bus: MessageBus) -> None:\r
+        super().__init__(client, bus)\r
 \r
     def _targetname(self):\r
         return "resourcepool"\r
@@ -31,7 +33,8 @@ class ResourcePoolWatcher(BaseWatcher):
     def _probe(self, parent: object = None):\r
         ocloudid = parent.id if parent else None\r
         newmodels = self._client.list(ocloudid=ocloudid)\r
-        for newmodel in newmodels:\r
-            logger.info("detect ocloudmodel:" + newmodel.name)\r
-            super()._compare_and_update(newmodel)\r
-        return newmodels\r
+        # for newmodel in newmodels:\r
+        #     logger.info("detect ocloudmodel:" + newmodel.name)\r
+        #     super()._compare_and_update(newmodel)\r
+        # return newmodels\r
+        return [commands.UpdateResourcePool(m) for m in newmodels]\r