AMTPowerPort: add status polling into exporter

Author: sessl3r

doc/configuration.rst

@@ -311,7 +311,8 @@ A :any:`AMTPowerPort` describes a AMT port accessible via `amtctrl`.
 Arguments:
  - host (str): hostname or ip the AMT interface of the PC is reachable
  - password (str): password to use for AMT login
- - timeout (int): timeout to use when polling the resource
+ - timeout (int): time to wait for the command to finish
+ - polling (int): interval to poll resource in exporter (disabled if 0)
 
 Used by:
   - `AMTPowerDriver`_

labgrid/remote/exporter.py

@@ -1044,6 +1044,74 @@ def _get_params(self):
 exports["IPMIPowerPort"] = IPMIPowerPortExport
 
 
+@attr.s(eq=False)
+class AMTPowerPortExport(ResourceExport):
+    """ResourceExport for AMTPowerPort devices"""
+
+    def __attrs_post_init__(self):
+        super().__attrs_post_init__()
+        self.data["cls"] = self.cls
+        from ..resource.power import AMTPowerPort
+        self.local = AMTPowerPort(target=None, name=None, **self.local_params)
+
+        self.lastpoll = 0
+        self.job = None
+        self.local.avail = False
+        self.status = None
+
+    def poll(self):
+        now = time.monotonic()
+        if self.job:
+            self.lastpoll = now
+            if self.job.poll() is None:
+                return super().poll()
+            self.status = self._amt_parse(self.job.returncode)
+
+            if self.job.returncode != 0:
+                self.local.avail = False
+            else:
+                self.local.avail = True
+            self.job = None
+            return super().poll()
+
+        if now - self.lastpoll < self.local.polling:
+            return super().poll()
+        self.lastpoll = now
+
+        if self.local.polling > 0:
+            self.job = self._amt_power('status')
+
+        return super().poll()
+
+    def _amt_power(self, cmd):
+        runstr = f"amtctrl {self.local.host} {cmd} -p"
+        p = subprocess.Popen(runstr.split(' '), text=True,
+                             stdout=subprocess.PIPE, stderr=subprocess.PIPE,
+                             stdin=subprocess.PIPE)
+        p.stdin.write(self.local.password)
+        p.stdin.close()
+        return p
+
+    def _amt_parse(self, ret):
+        out = self.job.stdout.read() + self.job.stderr.read()
+        if ret != 0:
+            out = out.split(']')[-1]
+            out = ''.join(x for x in out if not x in ['(',')','"', "'"])
+        return out.strip().upper()
+
+    def _get_params(self):
+        """Helper function to return parameters"""
+        return {
+            **self.local_params,
+            "extra": {
+                "status": self.status,
+            }
+        }
+
+
+exports["AMTPowerPort"] = AMTPowerPortExport
+
+
 class ExporterSession(ApplicationSession):
     def onConnect(self):
         """Set up internal datastructures on successful connection:

labgrid/resource/power.py

@@ -43,11 +43,13 @@ class AMTPowerPort(Resource):
     Args:
         host (str): hostname or ip the AMT interface of the PC is reachable
         password (str): password to use for AMT login
-        timeout (int): timeout to use when polling the resource
+        timeout (int): time to wait for command to finish
+        polling (int): interval to poll resource in exporter (disabled if 0)
     """
     host = attr.ib(validator=attr.validators.instance_of(str))
     password = attr.ib(validator=attr.validators.instance_of(str))
     timeout = attr.ib(default=30, validator=attr.validators.instance_of(int))
+    polling = attr.ib(default=0, validator=attr.validators.instance_of(int))
 
 
 @target_factory.reg_resource