Method connect

dispatcharr/consumers.py:9–44

(self)

class MyWebSocketConsumer(AsyncWebsocketConsumer):
    async def connect(self):
        self.room_name = "updates"

        user = self.scope["user"]
        if not user.is_authenticated:
            await self.close()
            return

        try:
            await self.accept()
            await self.channel_layer.group_add(self.room_name, self.channel_name)
            await self.send(text_data=json.dumps({
                'type': 'connection_established',
                'data': {
                    'success': True,
                    'message': 'WebSocket connection established successfully'
                }
            }))
            # If the IP lookup already completed before the client connected,
            # push the cached result immediately so the Skeleton resolves.
            try:
                from django.core.cache import cache
                from core.api_views import _IP_CACHE_KEY
                cached_ip = await sync_to_async(cache.get)(_IP_CACHE_KEY)
                if cached_ip:
                    await self.send(text_data=json.dumps({
                        'data': {'type': 'ip_lookup_complete', **cached_ip}
                    }))
            except Exception as e:
                logger.warning(f"Could not push cached IP result on connect: {e}")
        except Exception as e:
            logger.error(f"Error in WebSocket connect: {str(e)}")
            try:
                await self.close(code=1011)
            except:
                pass

    async def disconnect(self, close_code):
        try: