o >h21@sddlZddlZddlZddlZddlmZddlmZddlm Z m Z m Z m Z m Z eeZddZGdd d ZGd d d ZGd d d ZdS)N)asyncio)registry) _close_redis_consistent_hash _wrap_close create_pool decode_hostscs$|}t|||i|IdHSN) _get_layergetattr)objnameargskwargslayerru/var/www/vedio/testing/chatpythonscript.ninositsolution.com/env/lib/python3.10/site-packages/channels_redis/pubsub.py _async_proxysrc@s@eZdZddd dddZddZd d Zd d Zd dZdS)RedisPubSubChannelLayerNmsgpack)symmetric_encryption_keysserializer_formatreturncOs&||_||_i|_tj||d|_dS)N)r)_args_kwargs_layersrget_serializer _serializer)selfrrrrrrr__init__s z RedisPubSubChannelLayer.__init__cCs$|dvr tt||St||S)N) new_channelsendreceive group_add group_discard group_sendflush) functoolspartialrr r )rrrrr __getattr__,s z#RedisPubSubChannelLayer.__getattr__cC |j|S)z6 Serializes message to a byte string. )r serializermessagerrrr,: z!RedisPubSubChannelLayer.serializecCr+)z2 Deserializes from a byte string. )r deserializer-rrrr0@r/z#RedisPubSubChannelLayer.deserializecCs\t}z|j|}W|Sty-t|ji|jd|i}||j|<t||Y|Sw)N channel_layer)rget_running_looprKeyErrorRedisPubSubLoopLayerrrr)rlooprrrrr Fs     z"RedisPubSubChannelLayer._get_layer)rN)__name__ __module__ __qualname__r r*r,r0r rrrrrs  rc@s~eZdZdZ     dddZddZdd Zd d Zd d gZddZ dddZ ddZ ddZ ddZ ddZddZdS)r4z@ Channel Layer that uses Redis's pub/sub functionality. Nasgic s@|_|_|_|_i_i_fddt|D_dS)Ncsg|]}t|qSr)RedisSingleShardConnection).0hostrrr ts z1RedisPubSubLoopLayer.__init__..)prefix on_disconnect on_reconnectr1channelsgroupsr _shards)rhostsr?r@rAr1rrr=rr \s   zRedisPubSubLoopLayer.__init__cCs|jt|t|jS)zV Return the shard that is used exclusively for this channel or group. )rDrlen)rchannel_or_group_namerrr _get_shardxszRedisPubSubLoopLayer._get_shardcCs|jd|S)a\ Return the channel name used by a group. Includes '__group__' in the returned string so that these names are distinguished from those returned by `new_channel()`. Technically collisions are possible, but it takes what I believe is intentional abuse in order to have colliding names. __group__)r?)rgrouprrr_get_group_channel_name~s z,RedisPubSubLoopLayer._get_group_channel_namecs.t|j|<||}||IdHdSr )rQueuerBrH subscribe)rchannelshardrrr_subscribe_to_channels z*RedisPubSubLoopLayer._subscribe_to_channelrCr'cs*||}|||j|IdHdS)zF Send a message onto a (general or specific) channel. N)rHpublishr1r,)rrNr.rOrrrr"s zRedisPubSubLoopLayer.send specific.cs.|j|tj}||IdH|S)zy Returns a new channel name that can be used by a consumer in our process as a specific channel. N)r?uuiduuid4hexrP)rr?rNrrrr!sz RedisPubSubLoopLayer.new_channelc s||jvr||IdH|j|}z |IdH}Wn4tjtjtfyP||jvrO|j|=z||}||IdHWt yNt dYww|j |S)z Receive the first message that arrives on the channel. If more than one coroutine waits on the same channel, a random one of the waiting coroutines will get the result. Nz/Unexpected exception while cleaning-up channel:)rBrPgetrCancelledError TimeoutError GeneratorExitrH unsubscribe BaseExceptionlogger exceptionr1r0)rrNqr.rOrrrr#s(      zRedisPubSubLoopLayer.receivecsz||jvrtdt|d||}||jvr t|j|<|j|}||vr.||||}||IdHdS)z3 Adds the channel name to a group. zYou can only call group_add() on channels that exist in-process. Consumers are encouraged to use the common pattern: self.channel_layer.group_add(z, self.channel_name)N) rB RuntimeErrorreprrKrCsetaddrHrMrrJrN group_channelgroup_channelsrOrrrr$s       zRedisPubSubLoopLayer.group_addcsh||}|j|t}||vrdS||t|dkr2|j|=||}||IdHdSdS)zy Removes the channel from a group if it is in the group; does nothing otherwise (does not error) Nr)rKrCrVraremoverFrHrZrcrrrr%s    z"RedisPubSubLoopLayer.group_discardcs4||}||}|||j|IdHdS)zC Send the message to all subscribers of the group. N)rKrHrQr1r,)rrJr.rdrOrrrr&s  zRedisPubSubLoopLayer.group_sendcs,i|_i|_|jD] }|IdHq dS)z Flush the layer, making it like new. It can continue to be used as if it was just created. This also closes connections, serving as a clean-up method; connections will be re-opened if you continue using this layer. N)rBrCrDr')rrOrrrr's  zRedisPubSubLoopLayer.flush)Nr9NNN)rR)r6r7r8__doc__r rHrKrP extensionsr"r!r#r$r%r&r'rrrrr4Ws&    % r4c@sTeZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ dS)r:cCs4||_||_t|_t|_d|_d|_d|_ dSr ) r<r1ra_subscribed_torLock_lock_redis_pubsub _receive_task)rr<r1rrrr s  z#RedisSingleShardConnection.__init__c s^|j4IdH||j||IdHWdIdHdS1IdHs(wYdSr )rk _ensure_redisrlrQ)rrNr.rrrrQs .z"RedisSingleShardConnection.publishc s|j4IdH3||jvr0|||j|IdH|j|WdIdHdSWdIdHdS1IdHsAwYdSr )rkriro_ensure_receiverrmrMrbrrNrrrrM .z$RedisSingleShardConnection.subscribec s|j4IdH3||jvr0|||j|IdH|j|WdIdHdSWdIdHdS1IdHsAwYdSr )rkrirorprmrZrfrqrrrrZrrz&RedisSingleShardConnection.unsubscribec s|j4IdHC|jdur)|jz|jIdHWn tjy%Ynwd|_|jdurs   </