a Ti @sndZddlZddlZddlZeeZGdddZGdddZddZ d d Z d d Z d dZ ddZ dS)z WebSocket Messaging Module This module provides utilities for sending messages to WebSocket clients, including batched updates, serialization helpers, and typed message senders. Nc@sTeZdZdZddZeeddddZedd d Z edd d Z e dd dZ dS) BatchedUpdatez Accumulator for batched game state updates. Instead of sending 14+ separate messages per tick, batch all updates into a single message. cCs i|_dS)N_updatesselfr;/var/www/lichun.app/lichun/ws/server/websocket_messaging.py__init__szBatchedUpdate.__init__N)keyvaluereturncCs||j|<dS)zAdd an update to the batchNr)rr r rrraddszBatchedUpdate.add)r cCs|js dSd|jdS)z!Convert to dictionary for sendingNZ batch_update)typeZupdatesrrrrrto_dicts zBatchedUpdate.to_dictcCs"|}|durdStj|tdS)zConvert to JSON stringNdefault)rjsondumpsComplexHandler)rdatarrrto_json)szBatchedUpdate.to_jsoncCst|jdkS)zCheck if batch is emptyr)lenrrrrris_empty0szBatchedUpdate.is_empty) __name__ __module__ __qualname____doc__r stranyr dictrrboolrrrrrrs  rc@seZdZdZddZdS) serverClassz& Server state tracking class. cCs d|_dS)Nr)ticksrrrrr 9szserverClass.__init__N)rrrrr rrrrr!5sr!cCsNt|dr|St|tr$t|St|dr4|jStdt|jddS)z JSON serialization handler for complex objects. Args: Obj: Object to serialize Returns: Jsonable representation if available jsonable__dict__zObject of type z is not JSON serializableN) hasattrr# isinstancesetlistr$ TypeErrorrr)Objrrrr=s   rc sddlm}||j}|r~z||IdHWqtyz}z2td|jd|ddl}|WYd}~qd}~00ntd|jdS)z Send message to user (O(1) lookup). Args: websocket: WebSocket connection with userID attribute message: Message string to send r)USERSNzError sending to z: z,Warning: No active websocket found for user ) server.websocket_registryr+getZuserIDsend Exceptionprint traceback print_exc) websocketmessager+userer1rrr sendToUserQs   r7cs$|r t|tj|jtdIdHdS)z Send event message to websocket. Args: websocket: WebSocket connection m: Event object with __dict__ attribute rNr7rrr$r)r3mrrrsendEventMessagegsr:cs$|r t|tj|jtdIdHdS)z Send player information to websocket. Args: player: Player object with __dict__ attribute websocket: WebSocket connection rNr8)playerr3rrr sendUserInfossr<cs"|rt|tj|tdIdHdS)z Send dictionary object to websocket. Args: websocket: WebSocket connection obj: Object to serialize and send rN)r7rrr)r3objrrrsendDictsr>)rasynciorlogging getLoggerrloggerrr!rr7r:r<r>rrrrs %