# pylint: disable=no-member
import attr
import subprocess
import os.path
from ..factory import target_factory
from ..protocol import BootstrapProtocol
from ..resource.remote import NetworkMXSUSBLoader, NetworkIMXUSBLoader
from ..resource.udev import MXSUSBLoader, IMXUSBLoader
from ..step import step
from .common import Driver, check_file
from .exception import ExecutionError
@target_factory.reg_driver
@attr.s
[docs]class MXSUSBDriver(Driver, BootstrapProtocol):
bindings = {
"loader": {MXSUSBLoader, NetworkMXSUSBLoader},
}
image = attr.ib(default=None)
def __attrs_post_init__(self):
super().__attrs_post_init__()
# FIXME make sure we always have an environment or config
if self.target.env:
self.tool = self.target.env.config.get_tool('mxs-usb-loader') or 'mxs-usb-loader'
else:
self.tool = 'mxs-usb-loader'
[docs] def on_activate(self):
pass
[docs] def on_deactivate(self):
pass
@Driver.check_active
@step(args=['filename'])
[docs] def load(self, filename=None):
if filename is None and self.image is not None:
filename = self.target.env.config.get_image_path(self.image)
filename = os.path.abspath(filename)
check_file(filename, command_prefix=self.loader.command_prefix)
subprocess.check_call(
self.loader.command_prefix+[self.tool, "0", filename]
)
@target_factory.reg_driver
@attr.s
[docs]class IMXUSBDriver(Driver, BootstrapProtocol):
bindings = {
"loader": {IMXUSBLoader, NetworkIMXUSBLoader},
}
image = attr.ib(default=None)
def __attrs_post_init__(self):
super().__attrs_post_init__()
# FIXME make sure we always have an environment or config
if self.target.env:
self.tool = self.target.env.config.get_tool('imx-usb-loader') or 'imx-usb-loader'
else:
self.tool = 'imx-usb-loader'
[docs] def on_activate(self):
pass
[docs] def on_deactivate(self):
pass
@Driver.check_active
@step(args=['filename'])
[docs] def load(self, filename=None):
if filename is None and self.image is not None:
filename = self.target.env.config.get_image_path(self.image)
filename = os.path.abspath(filename)
check_file(filename, command_prefix=self.loader.command_prefix)
subprocess.check_call(
self.loader.command_prefix+[self.tool, "-p", str(self.loader.path), "-c", filename]
)