Madra_smieciarka/QOLfunc.py

# Tutaj wrzucam jakieś funkcje różne, co by nie zaśmiecać tym maina jak nie trzeba
from classes.Household import *

colors = {
    "white": (255, 255, 255),
    "black": (0, 0, 0),
    "red": (255, 0, 0)
}


def extractPredicts(houses: list) -> set:
    preds = set()
    for house in houses:
        temp = [house, house.getGarbage().getPredict()]
        preds.add(temp)
    return preds


def allLocations(*lista):
    final = set()
    for _ in lista:
        for item in _:
            final.add(item)
    return final