Initial commit

Signed-off-by: Rouven Czerwinski <[email protected]>

Author: Emantor

labgridhelper/__init__.py

@@ -0,0 +1,36 @@
+from labgrid.driver import BareboxDriver
+
+
+def get_commands(command):
+    """Returns the available commands of a running Barebox bootloader
+    Args:
+        command (BareboxDriver): An instance of the BareboxDriver
+    Returns:
+        list: list of the available commands
+    """
+    assert isinstance(command, BareboxDriver)
+    out = command.run_check("help")
+    commands = []
+    for line in out:
+        if line and line[0] == " ":
+            for cmd in line.split(','):
+                commands.append(cmd.strip(',').strip(" "))
+    return commands
+
+def get_globals(command):
+    """Returns the global variables of a running Barebox bootloader
+    Args:
+        command (BareboxDriver): An instance of the BareboxDriver
+    Returns:
+        dict: name as key and value as key-value
+    """
+    assert isinstance(command, BareboxDriver)
+    out = command.run_check("global")
+    out = map(lambda x: x[2:], out)
+    global_variables = {}
+    for line in out:
+        sep = line.index(":")
+        key = line[:sep]
+        value = line[sep+2:]
+        global_variables[key] = value
+    return global_variables

labgridhelper/barebox.py

@@ -0,0 +1,59 @@
+from labgrid.protocol import CommandProtocol
+
+def get_systemd_status(command):
+    assert isinstance(command, CommandProtocol), "command must be a CommandProtocol"
+    # TODO: Use busctl --json if systemd>239
+    array_notation = "a(ssssssouso)"
+    out = command.run_check(
+        "busctl call --no-pager org.freedesktop.systemd1 \
+        /org/freedesktop/systemd1 org.freedesktop.systemd1.Manager ListUnits"
+    )
+
+    out = out[0]
+    if array_notation not in out:
+        raise ValueError("Systemd ListUnits output changed")
+    out = out[len(array_notation):]
+    array_length = int(out[:out.index('"')].strip(" "))
+    out = out[out.index('"')+1:-1]
+    out = out.split('\" \"')
+    data = iter(out)
+    services = {}
+    for _ in range(array_length):
+        name = next(data)
+        services[name] = {}
+        services[name]["description"] = next(data)
+        services[name]["load"] = next(data)
+        services[name]["active"] = next(data)
+        services[name]["sub"] = next(data)
+        services[name]["follow"] = next(data)
+        path_and_id = next(data)
+        pos = path_and_id.index('"')
+        services[name]["path"] = path_and_id[:pos]
+        services[name]["id"] = int(path_and_id[pos+1:-1].strip(" "))
+        services[name]["type"] = path_and_id[path_and_id.rfind('"'):]
+        services[name]["objpath"] = next(data)
+
+    return services
+
+def get_commands(command, directories=None):
+    """Returns the commands of a running linux system
+    Args:
+        command (CommandProtocol): An instance of a Driver implementing the CommandProtocol
+        directories (list): An optional list of directories to include
+    Returns:
+        list: list of commands available under linux
+    """
+    assert isinstance(command, CommandProtocol), "command must be a CommandProtocol"
+    out = command.run_check("ls /usr/bin")
+    out.extend(command.run_check("ls /usr/sbin"))
+    if directories:
+        assert isinstance(directories, list), "directories must be a list"
+        for directory in directories:
+            out.extend(command.run_check("ls {}".format(directory)))
+    commands = []
+    for line in out:
+        for cmd in line.split(" "):
+            if cmd:
+                commands.append(cmd)
+
+    return commands

labgridhelper/linux.py

@@ -0,0 +1,28 @@
+#!/usr/bin/env python3
+
+from setuptools import setup
+
+setup(
+    name='labgridhelper',
+    description='labgridhelper: helper functions for the labgrid library',
+    author='Rouven Czerwinski',
+    author_email='[email protected]',
+    license='LGPL-2.1',
+    use_scm_version=True,
+    url='https://github.com/labgrid-project',
+    python_requires='>=3.5, <3.7',
+    install_requires=[
+        'labgrid>=0.1.0',
+    ],
+    packages=[
+        'labgridhelper',
+    ],
+    # custom PyPI classifiers
+    classifiers=[
+        "Topic :: Software Development :: Testing",
+        "Framework :: Pytest",
+        "Programming Language :: Python :: 3 :: Only",
+        "Programming Language :: Python :: 3.5",
+        "Programming Language :: Python :: 3.6",
+    ],
+)