SI_InteligentnyWozekWidlowy/visualization/DisplayOrderList.py

from collections import Counter
from typing import List

from mesa.visualization.modules import TextElement

from data.Order import Order
from data.enum.ItemType import ItemType


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)

        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