SI_InteligentnyWozekWidlowy/visualization/DisplayOrderList.py

from collections import Counter
from typing import List

from mesa.visualization.modules import TextElement

from data.Order import Order


class DisplayOrderList(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)
        res = ""
        if val is not None:
            orderList: List[Order] = val

            res = self.attr_name + ": <ol>"

            for o in orderList:
                if o is None:
                    continue

                itemList = map(lambda x: x.real_type, o.items)
                itemCounter = Counter(itemList)

                item_str = "<ul>"
                for e in itemCounter:
                    key = e
                    val = itemCounter[key]

                    # key_str = ""
                    # if key == ItemType.DOOR:
                    #     key_str = "Door"
                    # elif key == ItemType.SHELF:
                    #     key_str = "Shelf"

                    item_str += f"<li>{str(key)}:{str(val)}</li>"

                item_str += "</ul>"

                res += f"<li> items: {item_str} priority: {o.priority} <br> Client: {vars(o.client_params)} </li>"

            res += "</ol>"

        return res