o >h5@sddlZddlZddlZddlZddlZddlZddlZddlmZddl m Z ddl m Z ddl mZddlmZddlmZmZGd d d ZGd d d ZGd ddeZefddZeZdS)N)deepcopy)settings)setting_changed) import_string)DEFAULT_CHANNEL_LAYER) ChannelFullInvalidChannelLayerErrorc@s\eZdZdZddZddZeddZdd Zd d Z d d Z ddZ ddZ ddZ dS)ChannelLayerManagerzR Takes a settings dictionary of backends and initialises them on request. cCsi|_t|jdSN)backendsrconnect_reset_backendsselfro/var/www/vedio/testing/chatpythonscript.ninositsolution.com/env/lib/python3.10/site-packages/channels/layers.py__init__szChannelLayerManager.__init__cKs|dkr i|_dSdS)zX Removes cached channel layers when the CHANNEL_LAYERS setting changes. CHANNEL_LAYERSN)r )rsettingkwargsrrrrs z#ChannelLayerManager._reset_backendscCs ttdiS)Nr)getattrrrrrrconfigs#s zChannelLayerManager.configscCs|j|di}|||S)z, Instantiate channel layer. CONFIG)rget _make_backendrnameconfigrrr make_backend(s z ChannelLayerManager.make_backendcCs:z |j|d}Wn tytd|w|||S)zB Instantiate channel layer using its test config. TEST_CONFIGzNo TEST_CONFIG specified for %s)rKeyErrorr rrrrrmake_test_backend/s    z%ChannelLayerManager.make_test_backendcCsd|j|vr td|z t|j|d}Wn ty%td|ty8td|j|d|fw|di|S)NROUTINGzBROUTING key found for %s - this is no longer needed in Channels 2.BACKENDzNo BACKEND specified for %sz)Cannot import BACKEND %r specified for %sr)rr rr! ImportError)rrr backend_classrrrr9s$   z!ChannelLayerManager._make_backendcCs$||jvr |||j|<|j|Sr )r rrkeyrrr __getitem__Ms  zChannelLayerManager.__getitem__cCs ||jvSr )rr'rrr __contains__Rs z ChannelLayerManager.__contains__cCs|j|d}||j|<|S)z Sets an alias to point to a new ChannelLayerWrapper instance, and returns the old one that it replaced. Useful for swapping out the backend during tests. N)r r)rr(layeroldrrrsetUs zChannelLayerManager.setN)__name__ __module__ __qualname____doc__rrpropertyrrr"rr)r*r-rrrrr s   r c@seZdZdZdZd-ddZddZd d Zd d Ze d Z e dZ dd edZd.ddZddZd.ddZddZddZddZdd Zd!d"Zd#d$Zd%d&Zd'd(Zd.d)d*Zd+d,ZdS)/BaseChannelLayerzf Base channel layer class that others can inherit from, with useful common functionality. d<NcCs||_||_|p i|_dSr expirycapacitychannel_capacity)rr7r8r9rrrrhszBaseChannelLayer.__init__cCsNg}|D]\}}t|dr|||fq|tt||fq|S)z Takes an input channel_capacity dict and returns the compiled list of regexes that get_capacity will look for as self.channel_capacity match)itemshasattrappendrecompilefnmatch translate)rr9resultpatternvaluerrrcompile_capacitiesms  z#BaseChannelLayer.compile_capacitiescCs(|jD] \}}||r|Sq|jS)a Gets the correct capacity for the given channel; either the default, or a matching result from channel_capacity. Returns the first matching result; if you want to control the order of matches, use an ordered dict as input. )r9r:r8)rchannelrCr8rrr get_capacity|s  zBaseChannelLayer.get_capacitycCs t|trt||jkrdSdS)NTF) isinstancestrlenMAX_NAME_LENGTHrrrrrmatch_type_and_lengthsz&BaseChannelLayer.match_type_and_lengthz!^[a-zA-Z\d\-_.]+(\![\d\w\-_.]*)?$z^[a-zA-Z\d\-_.]+$z'{} name must be a valid unicode string zwith length < {} zFcontaining only ASCII alphanumerics, hyphens, underscores, or periods.FcCs\||s t|jdt|j|st|jdd|vr,|ds,|r,tddS)NChannel!z5Specific channel names in receive() must end at the !T)rM TypeErrorinvalid_name_errorformatboolchannel_name_regexr:endswith)rrreceiverrrrequire_valid_channel_names z+BaseChannelLayer.require_valid_channel_namecCs>||s t|jdt|j|st|jddS)NGroupT)rMrPrQrRrSgroup_name_regexr:rLrrrrequire_valid_group_names z)BaseChannelLayer.require_valid_group_namecCsB|rdnd}t|t}|r|sJd|D] }|j||dqdS)NTFznames must be a non-empty list)rV)rHlistrW)rnamesrV_non_empty_list _names_typerFrrrvalid_channel_namess  z$BaseChannelLayer.valid_channel_namescCs"d|vr|d|ddS|S)z Given a channel name, returns the "non-local" part. If the channel name is a process-specific channel (contains !) this means the part up to and including the !; if it is anything else, this means the full name. rONr)findrLrrrnon_local_nameszBaseChannelLayer.non_local_namec td)Nz/send() should be implemented in a channel layerNotImplementedError)rrFmessagerrrsendzBaseChannelLayer.sendcrb)Nz2receive() should be implemented in a channel layerrc)rrFrrrrVrgzBaseChannelLayer.receivecrb)Nz6new_channel() should be implemented in a channel layerrcrrrr new_channelszBaseChannelLayer.new_channelcrb)Nz)flush() not implemented (flush extension)rcrrrrflushrgzBaseChannelLayer.flushcrb)Nz.group_add() not implemented (groups extension)rcrgrouprFrrr group_addrgzBaseChannelLayer.group_addcrb)Nz2group_discard() not implemented (groups extension)rcrjrrr group_discardrgzBaseChannelLayer.group_discardcrb)Nz/group_send() not implemented (groups extension)rc)rrkrerrr group_sendrgzBaseChannelLayer.group_sendcCtjdtdd||S)zE Deprecated: Use require_valid_channel_name instead. zIvalid_channel_name is deprecated, use require_valid_channel_name instead. stacklevel)warningswarnDeprecationWarningrW)r channel_namerVrrrvalid_channel_name  z#BaseChannelLayer.valid_channel_namecCro)zD Deprecated: Use require_valid_group_name instead.. zEvalid_group_name is deprecated, use require_valid_group_name instead.rprq)rsrtrurZ)r group_namerrrvalid_group_namerxz!BaseChannelLayer.valid_group_name)r5r4N)F)r.r/r0r1rKrrErGrMr>r?rTrYrRrQrWrZr_rarfrVrhrirlrmrnrwrzrrrrr3`s8        r3cseZdZdZ    dfdd Zdd gZd d Zd d Zd ddZddZ ddZ ddZ ddZ ddZ ddZddZZS)!InMemoryChannelLayerz0 In-memory channel layer implementation r5Qr4Nc s0tjd|||d|i|_i|_||_dS)Nr6r)superrchannelsgroups group_expiry)rr7rr8r9r __class__rrrs zInMemoryChannelLayer.__init__rricst|ts Jd||d|vsJ|j|tj||d}z|t |j t |fWdStj j yAt|w)zF Send a message onto a (general or specific) channel. zmessage is not a dict__asgi_channel__maxsizeN)rHdictrWr~ setdefaultasyncioQueuerG put_nowaittimer7rqueues QueueFullr)rrFrequeuerrrrfs  "zInMemoryChannelLayer.sendc sz||||j|tj||d}z|IdH\}}W|r.|j |d|S|r<|j |dww)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. rN) rW_clean_expiredr~rrrrGremptypop)rrFr_rerrrrVs zInMemoryChannelLayer.receive specific.cs"d|dddtdDfS)zx Returns a new channel name that can be used by something in our process as a specific channel. z%s.inmemory!%scss|] }ttjVqdSr )randomchoicestring ascii_letters).0irrr 5sz3InMemoryChannelLayer.new_channel.. )joinrange)rprefixrrrrh.s z InMemoryChannelLayer.new_channelcCst|jD]6\}}|s=|jddtkr=||||r.|j|d|s=|jddtksqt t|j }|j D]}t|D]\}}|rd||krd||dqTqLdS)z Goes through all messages and groups and removes those that are expired. Any channel with an expired message is removed from all groups. rN) r[r~r;r_queuer get_nowait_remove_from_groupsrintrrvalues)rrFrtimeoutr~r timestamprrrr:s     z#InMemoryChannelLayer._clean_expiredcsi|_i|_dSr )r~rrrrrriVs zInMemoryChannelLayer.flushcsdSr rrrrrcloseZszInMemoryChannelLayer.closecCs |jD]}||dqdS)zW Removes a channel from all groups. Used when a message on it expires. N)rrr)rrFr~rrrr^sz(InMemoryChannelLayer._remove_from_groupscs:|||||j|it|j||<dS)z3 Adds the channel name to a group. N)rZrWrrrrjrrrrlgs   zInMemoryChannelLayer.group_addcsR|||||j|d}|r%||d|s'|j|ddSdSdSr )rWrZrrr)rrkrFgroup_channelsrrrrmrs   z"InMemoryChannelLayer.group_discardc st|ts Jd|||g}||jvr0|j|D]}|t| ||q!t |D]}z|IdHWq5t yGYq5wdS)NzMessage is not a dict) rHrrZrrkeysr=r create_taskrf as_completedr)rrkreopsrF send_resultrrrrns   zInMemoryChannelLayer.group_send)r5r|r4N)r)r.r/r0r1r extensionsrfrVrhrrirrrlrmrn __classcell__rrrrr{s$     r{cCs zt|WStyYdSw)zL Returns a channel layer by alias, or None if it is not configured. N)channel_layersr!)aliasrrrget_channel_layers   r)rr@rr>rrrscopyr django.confrdjango.core.signalsrdjango.utils.module_loadingrr~r exceptionsrr r r3r{rrrrrrs(     M  *