poky/bitbake/bin/toaster-eventreplay
Alexander Kanavin 2b399a01b5 bitbake: bitbake: enable python warnings at the first opportunity
We really do want to see those, as they tend to turn into
hard errors eventually, as what happened with collections
vs collections.abc in python 3.10.

(Bitbake rev: bc43fbb86361a21dc2d5deb910810c5a77fdabe8)

Signed-off-by: Alexander Kanavin <alex@linutronix.de>
Signed-off-by: Richard Purdie <richard.purdie@linuxfoundation.org>
2021-09-19 11:33:14 +01:00

3.5 KiB
Executable File

#!/usr/bin/env python3

Copyright (C) 2014 Alex Damian

SPDX-License-Identifier: GPL-2.0-only

This file re-uses code spread throughout other Bitbake source files.

As such, all other copyrights belong to their own right holders.

""" This command takes a filename as a single parameter. The filename is read as a build eventlog, and the ToasterUI is used to process events in the file and log data in the database """

import os import sys import json import pickle import codecs import warnings warnings.simplefilter("default")

from collections import namedtuple

mangle syspath to allow easy import of modules

from os.path import join, dirname, abspath sys.path.insert(0, join(dirname(dirname(abspath(file))), 'lib'))

import bb.cooker from bb.ui import toasterui

class EventPlayer: """Emulate a connection to a bitbake server."""

def __init__(self, eventfile, variables):
    self.eventfile = eventfile
    self.variables = variables
    self.eventmask = []

def waitEvent(self, _timeout):
    """Read event from the file."""
    line = self.eventfile.readline().strip()
    if not line:
        return
    try:
        event_str = json.loads(line)['vars'].encode('utf-8')
        event = pickle.loads(codecs.decode(event_str, 'base64'))
        event_name = "%s.%s" % (event.__module__, event.__class__.__name__)
        if event_name not in self.eventmask:
            return
        return event
    except ValueError as err:
        print("Failed loading ", line)
        raise err

def runCommand(self, command_line):
    """Emulate running a command on the server."""
    name = command_line[0]

    if name == "getVariable":
        var_name = command_line[1]
        variable = self.variables.get(var_name)
        if variable:
            return variable['v'], None
        return None, "Missing variable %s" % var_name

    elif name == "getAllKeysWithFlags":
        dump = {}
        flaglist = command_line[1]
        for key, val in self.variables.items():
            try:
                if not key.startswith("__"):
                    dump[key] = {
                        'v': val['v'],
                        'history' : val['history'],
                    }
                    for flag in flaglist:
                        dump[key][flag] = val[flag]
            except Exception as err:
                print(err)
        return (dump, None)

    elif name == 'setEventMask':
        self.eventmask = command_line[-1]
        return True, None

    else:
        raise Exception("Command %s not implemented" % command_line[0])

def getEventHandle(self):
    """
    This method is called by toasterui.
    The return value is passed to self.runCommand but not used there.
    """
    pass

def main(argv): with open(argv[-1]) as eventfile: # load variables from the first line variables = json.loads(eventfile.readline().strip())['allvariables']

    params = namedtuple('ConfigParams', ['observe_only'])(True)
    player = EventPlayer(eventfile, variables)

    return toasterui.main(player, player, params)

run toaster ui on our mock bitbake class

if name == "main": if len(sys.argv) != 2: print("Usage: %s " % os.path.basename(sys.argv[0])) sys.exit(1)

sys.exit(main(sys.argv))