NetworkPowerPort in exporter yaml config #1515

import attr
from labgrid.factory import target_factory
from labgrid.step import step
from labgrid.driver import Driver
from labgrid.driver.exception import ExecutionError
from labgrid.driver.powerdriver import PowerResetMixin
from labgrid.protocol import PowerProtocol
from labgrid.resource.power import NetworkPowerPort
from labgrid.util.helper import processwrapper
from labgrid.util.proxy import proxymanager
OID = ".1.3.6.1.4.1.318.1.1.4.4.2.1.3"
@target_factory.reg_driver
@attr.s(eq=False)
class ProxySMTPPowerDriver(Driver, PowerResetMixin, PowerProtocol):
 bindings = {
 "port": NetworkPowerPort,
 }
 delay = attr.ib(default=2.0, validator=attr.validators.instance_of(float))
 def on_activate(self):
 self._host = self.port.host
 self._index = self.port.index
 self._proxy = proxymanager._force_proxy
 @Driver.check_active
 @step()
 def on(self):
 self._set(self._host, self._index, True)
 @Driver.check_active
 @step()
 def off(self):
 self._set(self._host, self._index, False)
 @Driver.check_active
 @step()
 def cycle(self):
 self.off()
 time.sleep(self.delay)
 self.on()
 @Driver.check_active
 def get(self):
 return self._get(
 self._host,
 self._index,
 )
 def _set(self, host: str, index: int, value: bool):
 value = 1 if value else 2
 try:
 processwapper.check_output(
 f"ssh {self._proxy} snmpset -v1 -c private -Onqe {host} {OID}.{index} {value}".split()
 )
 except Exception as e:
 raise ExecutionError("failed to set snmp value") from e
 def _get(self, host: str, index: int) -> bool:
 oid = f"{OID}.{index}"
 out = processwrapper.check_output(
 f"ssh {self._proxy} snmpget -v1 -c private -Onqe {host} {oid}".split()
 ).decode("ascii")
 out_oid, value = out.strip().split(" ", 1)
 assert oid == out_oid
 match value:
 case "1":
 return True
 case "2":
 return False
 case _:
 raise ExecutionError("failed to get snmp value")