extensions/python/webui_ws_connect/_10_state_sync.py

from helpers.extension import Extension
from helpers.print_style import PrintStyle
from helpers.state_monitor import get_state_monitor, _ws_debug_enabled


class StateSync(Extension):
    async def execute(self, instance=None, sid: str = "", **kwargs):
        if instance is None:
            return

        monitor = get_state_monitor()
        monitor.bind_manager(instance.manager, handler_id=instance.identifier)
        monitor.register_sid(instance.namespace, sid)
        if _ws_debug_enabled():
            PrintStyle.debug(f"[WebuiHandler] connect sid={sid}")