Source code for labgrid.driver.flashscriptdriver

import logging
import attr

from ..factory import target_factory
from ..step import step
from ..util.managedfile import ManagedFile
from .common import Driver

from ..util.helper import processwrapper

[docs] @target_factory.reg_driver @attr.s(eq=False) class FlashScriptDriver(Driver): bindings = { "device": { "USBFlashableDevice", "NetworkUSBFlashableDevice", }, } script = attr.ib( default=None, validator=attr.validators.optional(attr.validators.instance_of(str)), ) args = attr.ib( default=attr.Factory(list), validator=attr.validators.optional(attr.validators.instance_of(list)), )
[docs] def __attrs_post_init__(self): super().__attrs_post_init__() self.logger = logging.getLogger(f"{self}:{}")
[docs] def on_activate(self): pass
[docs] def on_deactivate(self): pass
[docs] @Driver.check_active @step(args=["script"]) def flash(self, script=None, args=None): """ Transfers and remotely executes the script Args: script (str): optional, path to the script to write to bound Flashable Device """ if script is None and self.script is not None: script = assert script, "flash requires a script" if args is None: args = self.args mf = ManagedFile(script, self.device) mf.sync_to_resource() cmd = [mf.get_remote_path()] + [a.format(device=self.device, file=mf) for a in args] self.logger.debug("Running command '%s'", " ".join(cmd)) processwrapper.check_output( self.device.command_prefix + cmd, print_on_silent_log=True )