Add dms handler
[pti/o2.git] / o2ims / service / watcher / pserver_watcher.py
index c4ac72c..befc2f6 100644 (file)
@@ -12,6 +12,7 @@
 #  See the License for the specific language governing permissions and\r
 #  limitations under the License.\r
 \r
+from o2ims.domain.stx_object import StxGenericModel\r
 from o2ims.service.client.base_client import BaseClient\r
 # from o2ims.service.unit_of_work import AbstractUnitOfWork\r
 from o2ims.service.watcher.resource_watcher import ResourceWatcher\r
@@ -30,10 +31,8 @@ class PServerWatcher(ResourceWatcher):
     def _targetname(self):\r
         return "pserver"\r
 \r
-    def _probe(self, parent: object = None):\r
-        resourcepoolid = parent.id if parent else None\r
+    def _probe(self, parent: StxGenericModel):\r
+        resourcepoolid = parent.id\r
         newmodels = self._client.list(resourcepoolid=resourcepoolid)\r
-        # for newmodel in newmodels:\r
-        #     super()._compare_and_update(newmodel)\r
-        # return newmodels\r
-        return [commands.UpdatePserverCpu(m) for m in newmodels]\r
+        return [commands.UpdatePserverCpu(data=m, parentid=resourcepoolid)\r
+                for m in newmodels]\r