from mesa.visualization.modules import TextElement class DisplayPictureElement(TextElement): def __init__(self, attr_name): ''' Create a new text attribute element. Args: attr_name: The name of the attribute to extract from the model. Example return: "happy: 10" ''' self.attr_name = attr_name def render(self, model): val = getattr(model, self.attr_name) if val is not None: return self.attr_name + ': {}'.format(val, self.attr_name) else: return ""