Add dms handler

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

diff --git a/o2ims/service/watcher/ocloud_watcher.py b/o2ims/service/watcher/ocloud_watcher.py
index bf4ce43..31e2f1b 100644 (file)
--- a/o2ims/service/watcher/ocloud_watcher.py
+++ b/o2ims/service/watcher/ocloud_watcher.py
@@ -14,7 +14,7 @@
 \r
 # from o2ims.domain.resource_type import ResourceTypeEnum\r
 from o2ims.service.client.base_client import BaseClient\r
-# from o2ims.domain.stx_object import StxGenericModel\r
+from o2ims.domain.stx_object import StxGenericModel\r
 # from o2ims.service.unit_of_work import AbstractUnitOfWork\r
 from o2ims.service.watcher.base import BaseWatcher\r
 from o2ims.domain import commands\r
@@ -73,10 +73,11 @@ class DmsWatcher(BaseWatcher):
     def _targetname(self):\r
         return "dms"\r
 \r
-    def _probe(self, parent: object = None):\r
-        ocloudid = parent.id if parent else None\r
+    def _probe(self, parent: StxGenericModel):\r
+        ocloudid = parent.id\r
         newmodels = self._client.list(ocloudid=ocloudid)\r
         # for newmodel in newmodels:\r
         #     super()._compare_and_update(newmodel)\r
         # return newmodels\r
-        return [commands.UpdateDms(m) for m in newmodels]\r
+        return [commands.UpdateDms(data=m, parentid=ocloudid)\r
+                for m in newmodels]\r