U ^~Hj #@sddlZddlmZddlmZmZddlmZddlm Z m Z m Z m Z GdddeZ Gd d d e ZGd d d eZGd ddeZdS)N) async_to_sync) AsyncConsumer SyncConsumer)aclose_old_connections)AcceptConnectionDenyConnectionInvalidChannelLayerError StopConsumercs|eZdZdZdZddZddZddZdfd d Zd d Z dd dZ dfdd Z dfdd Z ddZ ddZZS)WebsocketConsumerz Base WebSocket consumer. Provides a general encapsulation for the WebSocket handling model that other applications can build on. NcOs|jdkrg|_dSNgroupsselfargskwargsr>/tmp/pip-unpacked-wheel-a46u9ts2/channels/generic/websocket.py__init__s zWebsocketConsumer.__init__cCsz$|jD]}t|jj||jqWntk r@tdYnXz |Wn6tk rj| Ynt k r| YnXdS)? Called when a WebSocket connection is opened. 1BACKEND is unconfigured or doesn't support groupsN) rr channel_layer group_add channel_nameAttributeErrorr connectracceptrclosermessagegrouprrrwebsocket_connects    z#WebsocketConsumer.websocket_connectcCs |dSr rrrrrr-szWebsocketConsumer.connectcs*d|d}|rt||d<t|dSz, Accepts an incoming socket zwebsocket.accept)type subprotocolheadersNlistsupersendrr'r(r  __class__rrr0s  zWebsocketConsumer.acceptcCs4|ddk r |j|ddn|j|dddSzk Called when a WebSocket frame is received. Decodes it and passes it to receive(). textN) text_databytes) bytes_datagetreceiverr rrrwebsocket_receive:sz#WebsocketConsumer.websocket_receivecCsdSz8 Called with a decoded WebSocket frame. Nrrr2r4rrrr7DszWebsocketConsumer.receiveFcsR|dk rtd|dn$|dk r8td|dntd|rN||dSz7 Sends a reply back down the WebSocket Nzwebsocket.send)r&r1)r&r3z,You must pass one of bytes_data or text_datar+r, ValueErrorrrr2r4rr.rrr,JszWebsocketConsumer.sendcs<ddi}|dk r |dk r ||d<|r,||d<t|dSz: Closes the WebSocket from the server end r&zwebsocket.closeNTcodereasonr+r,rrArBr r.rrrWs zWebsocketConsumer.closecCsZz$|jD]}t|jj||jqWntk r@tdYnX||dtdS) Called when a WebSocket connection is closed. Base level so you don't need to call super() all the time. rrAN) rrr group_discardrrr disconnectr rrrrwebsocket_disconnectbs    z&WebsocketConsumer.websocket_disconnectcCsdSz? Called when a WebSocket connection is closed. NrrrArrrrGsszWebsocketConsumer.disconnect)NN)NN)NNF)NN__name__ __module__ __qualname____doc__rrr"rrr9r7r,rrHrG __classcell__rrr.rr s    r csLeZdZdZdddZddZdfdd Zed d Zed d Z Z S)JsonWebsocketConsumerz Variant of WebsocketConsumer that automatically JSON-encodes and decodes messages as they come in and go out. Expects everything to be text; will error on binary data. NcKs&|r|j||f|ntddSNz-No text section for incoming WebSocket frame! receive_json decode_jsonr>rr2r4rrrrr7szJsonWebsocketConsumer.receivecKsdSz3 Called with decoded JSON content. NrrcontentrrrrrTsz"JsonWebsocketConsumer.receive_jsonFcstj|||ddS)M Encode the given content as JSON and send it to the client. r2rNr+r, encode_jsonrrYrr.rr send_jsonszJsonWebsocketConsumer.send_jsoncCs t|Sr jsonloadsclsr2rrrrUsz!JsonWebsocketConsumer.decode_jsoncCs t|Sr radumpsrdrYrrrr]sz!JsonWebsocketConsumer.encode_json)NN)F rLrMrNrOr7rTr_ classmethodrUr]rPrrr.rrQzs  rQcs|eZdZdZdZddZddZddZdfd d Zd d Z dd dZ dfdd Z dfdd Z ddZ ddZZS)AsyncWebsocketConsumerz Base WebSocket consumer, async version. Provides a general encapsulation for the WebSocket handling model that other applications can build on. NcOs|jdkrg|_dSr r rrrrrs zAsyncWebsocketConsumer.__init__csz&|jD]}|j||jIdHqWntk rBtdYnXz|IdHWnBtk rx|IdHYn"t k r| IdHYnXdS)rNr) rrrrrr rrrrrrrrrr"s  z(AsyncWebsocketConsumer.websocket_connectcs|IdHdSr r#r$rrrrszAsyncWebsocketConsumer.connectcs0d|d}|rt||d<t|IdHdSr%r)r-r.rrrs  zAsyncWebsocketConsumer.acceptcs@|ddk r&|j|ddIdHn|j|ddIdHdSr0r5r8rrrr9sz(AsyncWebsocketConsumer.websocket_receivecsdSr:rr;rrrr7szAsyncWebsocketConsumer.receiveFcsd|dk r"td|dIdHn*|dk rDtd|dIdHntd|r`||IdHdSr<r=r?r.rrr,szAsyncWebsocketConsumer.sendcsBddi}|dk r |dk r ||d<|r,||d<t|IdHdSr@rCrDr.rrrs zAsyncWebsocketConsumer.closecsnz&|jD]}|j||jIdHqWntk rBtdYnX||dIdHtIdHtdS)rENrrA) rrrFrrr rGrr rrrrrHs   z+AsyncWebsocketConsumer.websocket_disconnectcsdSrIrrJrrrrGsz!AsyncWebsocketConsumer.disconnect)NN)NN)NNF)NNrKrrr.rrjs    rjcsLeZdZdZdddZddZdfdd Zed d Zed d Z Z S)AsyncJsonWebsocketConsumerz Variant of AsyncWebsocketConsumer that automatically JSON-encodes and decodes messages as they come in and go out. Expects everything to be text; will error on binary data. Ncs2|r&|j||IdHf|IdHntddSrRrSrVrrrr7 s"z"AsyncJsonWebsocketConsumer.receivecsdSrWrrXrrrrTsz'AsyncJsonWebsocketConsumer.receive_jsonFcs&tj||IdH|dIdHdS)rZNr[r\r^r.rrr_sz$AsyncJsonWebsocketConsumer.send_jsoncs t|Sr r`rcrrrrUsz&AsyncJsonWebsocketConsumer.decode_jsoncs t|Sr rergrrrr]"sz&AsyncJsonWebsocketConsumer.encode_json)NN)Frhrrr.rrks  rk)raZ asgiref.syncrZconsumerrrdbr exceptionsrrr r r rQrjrkrrrrs  k"i