from nicegui import ui


class Registry:

    def __init__(self):
        self.data = {}

    def __iter__(self):
        return iter(self.data.values())

    def add(self, widget):
        client = ui.context.client
        self.data[client.id] = widget
        client.on_disconnect(lambda: self.data.pop(client.id, None))