fischertechnik-factory-twin/simulator/devs_models/simple_collector.py

from pypdevs.DEVS import AtomicDEVS
from loguru import logger

class SimpleCollector(AtomicDEVS):
    """A simple collector which holds every received item in a list in its state"""
    def __init__(self, name):
        super(SimpleCollector, self).__init__(name)
        self.inp = self.addInPort("inp")

        # keep track of nr of customers obtained
        self.state = {
            "items": []
        }

    def extTransition(self, inputs):
        if self.inp in inputs:
            new_item = inputs[self.inp]
            self.state["items"].append(new_item)
            logger.trace(f"{type(self).__name__} '{self.name}' received: {new_item}")
        return self.state