Source code for labgrid.resource.remote

import logging
import os
import attr

from ..factory import target_factory
from .common import NetworkResource, ManagedResource, ResourceManager


[docs]@attr.s(eq=False) class RemotePlaceManager(ResourceManager):
[docs] def __attrs_post_init__(self): super().__attrs_post_init__() self.logger = logging.getLogger("{}".format(self)) self.url = None self.realm = None self.loop = None self.session = None self.ready = None self.unmanaged_resources = []
def _start(self): if self.session: return from ..remote.client import start_session try: self.session = start_session(self.url, self.realm, {'env': self.env}) except ConnectionRefusedError as e: raise ConnectionRefusedError("Could not connect to coordinator {}".format(self.url)) \ from e self.loop = self.session.loop
[docs] def on_resource_added(self, resource): if not isinstance(resource, RemotePlace): # we only need to handle new remote places here return remote_place = resource # Use the config file from the first resources we see (they should all # be the same). if not self.session: self.env = remote_place.target.env config = self.env.config self.url = config.get_option( 'crossbar_url', os.environ.get("LG_CROSSBAR", "ws://127.0.0.1:20408/ws")) self.realm = config.get_option( 'crossbar_realm', os.environ.get("LG_CROSSBAR_REALM", "realm1")) self._start() place = self.session.get_place(remote_place.name) resource_entries = self.session.get_target_resources(place) expanded = [] for resource_name, resource_entry in resource_entries.items(): new = target_factory.make_resource( remote_place.target, resource_entry.cls, resource_name, resource_entry.args) new.parent = remote_place new.avail = resource_entry.avail new.extra = resource_entry.extra new._remote_entry = resource_entry if not isinstance(new, ManagedResource): self.unmanaged_resources.append(new) expanded.append(new) self.logger.debug("expanded remote resources for place %s: %s", remote_place.name, expanded) remote_place.avail = True
[docs] def poll(self): import asyncio if not self.loop.is_running(): self.loop.run_until_complete(asyncio.sleep(0.1)) for resource in self.resources + self.unmanaged_resources: if isinstance(resource, RemotePlace): continue attrs = resource._remote_entry.args.copy() attrs['avail'] = resource._remote_entry.avail # TODO allow the resource to do the update itself? changes = [] fields = attr.fields(resource.__class__) for k, v_new in attrs.items(): # check for attr converters attrib = getattr(fields, k) if attrib.converter: v_new = attrib.converter(v_new) v_old = getattr(resource, k) setattr(resource, k, v_new) if v_old != v_new: changes.append((k, v_old, v_new)) if changes: self.logger.debug("changed attributes for %s:", resource) for k, v_old, v_new in changes: self.logger.debug(" %s: %s -> %s", k, v_old, v_new)
[docs]@target_factory.reg_resource @attr.s(eq=False) class RemotePlace(ManagedResource): manager_cls = RemotePlaceManager
[docs] def __attrs_post_init__(self): self.timeout = 10.0 super().__attrs_post_init__()
[docs]@attr.s(eq=False) class RemoteUSBResource(NetworkResource, ManagedResource): manager_cls = RemotePlaceManager busnum = attr.ib(validator=attr.validators.optional(attr.validators.instance_of(int))) devnum = attr.ib(validator=attr.validators.optional(attr.validators.instance_of(int))) path = attr.ib(validator=attr.validators.optional(attr.validators.instance_of(str))) vendor_id = attr.ib(validator=attr.validators.optional(attr.validators.instance_of(int))) model_id = attr.ib(validator=attr.validators.optional(attr.validators.instance_of(int)))
[docs]@target_factory.reg_resource @attr.s(eq=False) class NetworkAndroidFastboot(RemoteUSBResource):
[docs] def __attrs_post_init__(self): self.timeout = 10.0 super().__attrs_post_init__()
[docs]@target_factory.reg_resource @attr.s(eq=False) class NetworkIMXUSBLoader(RemoteUSBResource):
[docs] def __attrs_post_init__(self): self.timeout = 10.0 super().__attrs_post_init__()
[docs]@target_factory.reg_resource @attr.s(eq=False) class NetworkMXSUSBLoader(RemoteUSBResource):
[docs] def __attrs_post_init__(self): self.timeout = 10.0 super().__attrs_post_init__()
[docs]@target_factory.reg_resource @attr.s(eq=False) class NetworkRKUSBLoader(RemoteUSBResource):
[docs] def __attrs_post_init__(self): self.timeout = 10.0 super().__attrs_post_init__()
[docs]@target_factory.reg_resource @attr.s(eq=False) class NetworkAlteraUSBBlaster(RemoteUSBResource):
[docs] def __attrs_post_init__(self): self.timeout = 10.0 super().__attrs_post_init__()
[docs]@target_factory.reg_resource @attr.s(eq=False) class NetworkSigrokUSBDevice(RemoteUSBResource): """The NetworkSigrokUSBDevice describes a remotely accessible sigrok USB device""" driver = attr.ib( default=None, validator=attr.validators.instance_of(str) ) channels = attr.ib( default=None, validator=attr.validators.optional(attr.validators.instance_of(str)) )
[docs] def __attrs_post_init__(self): self.timeout = 10.0 super().__attrs_post_init__()
[docs]@target_factory.reg_resource @attr.s(eq=False) class NetworkSigrokUSBSerialDevice(RemoteUSBResource): """The NetworkSigrokUSBSerialDevice describes a remotely accessible sigrok USB device""" driver = attr.ib( default=None, validator=attr.validators.instance_of(str) ) channels = attr.ib( default=None, validator=attr.validators.optional(attr.validators.instance_of(str)) )
[docs] def __attrs_post_init__(self): self.timeout = 10.0 super().__attrs_post_init__()
[docs]@target_factory.reg_resource @attr.s(eq=False) class NetworkUSBMassStorage(RemoteUSBResource): """The NetworkUSBMassStorage describes a remotely accessible USB storage device"""
[docs] def __attrs_post_init__(self): self.timeout = 10.0 super().__attrs_post_init__()
[docs]@target_factory.reg_resource @attr.s(eq=False) class NetworkUSBSDMuxDevice(RemoteUSBResource): """The NetworkUSBSDMuxDevice describes a remotely accessible USBSDMux device""" control_path = attr.ib( default=None, validator=attr.validators.optional(attr.validators.instance_of(str)) )
[docs] def __attrs_post_init__(self): self.timeout = 10.0 super().__attrs_post_init__()
[docs]@target_factory.reg_resource @attr.s(eq=False) class NetworkUSBPowerPort(RemoteUSBResource): """The NetworkUSBPowerPort describes a remotely accessible USB hub port with power switching""" index = attr.ib(default=None, validator=attr.validators.instance_of(int))
[docs] def __attrs_post_init__(self): self.timeout = 10.0 super().__attrs_post_init__()
[docs]@target_factory.reg_resource @attr.s(eq=False) class NetworkUSBVideo(RemoteUSBResource): """The NetworkUSBVideo describes a remotely accessible USB video device"""
[docs] def __attrs_post_init__(self): self.timeout = 10.0 super().__attrs_post_init__()
[docs]@target_factory.reg_resource @attr.s(eq=False) class NetworkUSBTMC(RemoteUSBResource): """The NetworkUSBTMC describes a remotely accessible USB TMC device"""
[docs] def __attrs_post_init__(self): self.timeout = 10.0 super().__attrs_post_init__()
[docs]@target_factory.reg_resource @attr.s(eq=False) class NetworkDeditecRelais8(RemoteUSBResource): """The NetworkDeditecRelais8 describes a remotely accessible USB relais port""" index = attr.ib(default=None, validator=attr.validators.instance_of(int)) invert = attr.ib(default=False, validator=attr.validators.instance_of(bool))
[docs] def __attrs_post_init__(self): self.timeout = 10.0 super().__attrs_post_init__()
[docs]@target_factory.reg_resource @attr.s(eq=False) class NetworkSysfsGPIO(NetworkResource, ManagedResource): manager_cls = RemotePlaceManager """The NetworkSysfsGPIO describes a remotely accessible gpio line""" index = attr.ib(validator=attr.validators.optional(attr.validators.instance_of(int)))
[docs] def __attrs_post_init__(self): self.timeout = 10.0 super().__attrs_post_init__()