U QVgM3@sddlZddlZddlZddlZddlZddlZddlmZddlm Z ddl m Z ddl m Z ddlmZddlmZmZGd 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_backendsselfr3/tmp/pip-unpacked-wheel-lkk2pq7o/channels/layers.py__init__szChannelLayerManager.__init__cKs|dkri|_dS)zX Removes cached channel layers when the CHANNEL_LAYERS setting changes. CHANNEL_LAYERSN)r )rZsettingkwargsrrrrsz#ChannelLayerManager._reset_backendscCs ttdiS)Nr)getattrrrrrrconfigs"szChannelLayerManager.configscCs|j|di}|||S)z, Instantiate channel layer. ZCONFIG)rget _make_backendrnameconfigrrr make_backend'sz ChannelLayerManager.make_backendcCs@z|j|d}Wn tk r2td|YnX|||S)zB Instantiate channel layer using its test config. Z TEST_CONFIGzNo TEST_CONFIG specified for %s)rKeyErrorr rrrrrmake_test_backend.s z%ChannelLayerManager.make_test_backendcCsd|j|krtd|zt|j|d}WnLtk rPtd|Yn.tk r|td|j|d|fYnX|f|S)NZROUTINGzBROUTING key found for %s - this is no longer needed in Channels 2.ZBACKENDzNo BACKEND specified for %sz)Cannot import BACKEND %r specified for %s)rr rr ImportError)rrrZ backend_classrrrr8s" z!ChannelLayerManager._make_backendcCs$||jkr|||j|<|j|Sr )r rrkeyrrr __getitem__Ls zChannelLayerManager.__getitem__cCs ||jkSr )rr!rrr __contains__Qsz 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"ZlayeroldrrrsetTs zChannelLayerManager.setN)__name__ __module__ __qualname____doc__rrpropertyrrrrr#r$r&rrrrr s  r c@seZdZdZdZd*ddZddZd d Zd d Ze d Z e dZ dd eddZd+ddZddZd,ddZddZddZddZd d!Zd"d#Zd$d%Zd&d'Zd(d)ZdS)-BaseChannelLayerzf Base channel layer class that others can inherit from, with useful common functionality. d<NcCs||_||_|pi|_dSr expirycapacitychannel_capacity)rr0r1r2rrrrgszBaseChannelLayer.__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)rr2resultpatternvaluerrrcompile_capacitiesls  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. )r2r3r1)rchannelr<r1rrr 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 < {} zGcontaining only ASCII alphanumerics, hyphens, underscores, or periods, znot {}FcCsR||rsz7BaseChannelLayer.valid_channel_names..)rAlistAssertionErrorall)rnamesrNZ_non_empty_listZ _names_typerrTrvalid_channel_namess   z$BaseChannelLayer.valid_channel_namescCs&d|kr|d|ddS|SdS)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. rGNr)findrErrrnon_local_nameszBaseChannelLayer.non_local_namecs tddS)Nz/send() should be implemented in a channel layerNotImplementedError)rr?messagerrrsendszBaseChannelLayer.sendcs tddS)Nz2receive() should be implemented in a channel layerr])rr?rrrrNszBaseChannelLayer.receivecs tddS)Nz6new_channel() should be implemented in a channel layerr]rrrr new_channelszBaseChannelLayer.new_channelcs tddS)Nz)flush() not implemented (flush extension)r]rrrrflushszBaseChannelLayer.flushcs tddS)Nz.group_add() not implemented (groups extension)r]rgroupr?rrr group_addszBaseChannelLayer.group_addcs tddS)Nz2group_discard() not implemented (groups extension)r]rcrrr group_discardszBaseChannelLayer.group_discardcs tddS)Nz/group_send() not implemented (groups extension)r])rrdr_rrr group_sendszBaseChannelLayer.group_send)r.r-N)F)F)r'r(r)r*rDrr>r@rFr7r8rIrQrMrLrOrRrZr\r`rNrarbrerfrgrrrrr,_s6       r,cs|eZdZdZdfdd Zdd gZd d Zd d Zd ddZddZ ddZ ddZ ddZ ddZ ddZddZZS)!InMemoryChannelLayerz0 In-memory channel layer implementation r.Qr-Nc s0tjf|||d|i|_i|_||_dS)Nr/)superrchannelsgroups group_expiry)rr0rmr1r2r __class__rrrszInMemoryChannelLayer.__init__rlrbcst|tstd||s$tdd|ks0t|j|tj||d}z | t |j t |fWn tj jk rt|YnXdS)zF Send a message onto a (general or specific) channel. zmessage is not a dictChannel name not validZ__asgi_channel__maxsizeN)rAdictrWrOrk setdefaultasyncioQueuer@ put_nowaittimer0rZqueuesZ QueueFullr)rr?r_queuerrrr`s  zInMemoryChannelLayer.sendc sf||st||j|tj||d}z| IdH\}}W5|r`|j |dX|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. rqN) rOrW_clean_expiredrkrtrurvr@emptypopr)rr?ry_r_rrrrNszInMemoryChannelLayer.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)rSirrrrU!sz3InMemoryChannelLayer.new_channel.. )joinrange)rprefixrrrrasz InMemoryChannelLayer.new_channelcCst|jD]P\}}|s|jddtkr||||r|j|dqqt t|j }|j D]2}t|D] \}}|r||kr||dqq|dS)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) rVrkr4r{_queuerx get_nowait_remove_from_groupsr|intrmrlvalues)rr?rytimeoutrkr timestamprrrrz&s  z#InMemoryChannelLayer._clean_expiredcsi|_i|_dSr )rkrlrrrrrbBszInMemoryChannelLayer.flushcsdSr rrrrrcloseFszInMemoryChannelLayer.closecCs |jD]}||dq dS)zW Removes a channel from all groups. Used when a message on it expires. N)rlrr|)rr?rkrrrrJsz(InMemoryChannelLayer._remove_from_groupscsH||std||s$td|j|it|j||<dS)z3 Adds the channel name to a group. zGroup name not validrpN)rRrWrOrlrtrxrcrrrreSszInMemoryChannelLayer.group_addcsX||std||s$td|j|d}|rT||d|sT|j|ddS)NzInvalid channel nameInvalid group name)rOrWrRrlrr|)rrdr?Zgroup_channelsrrrrf^s z"InMemoryChannelLayer.group_discardc st|tstd||s$td|g}||jkrf|j|D]}|t | ||qHt |D](}z|IdHWqpt k rYqpXqpdS)NzMessage is not a dictr) rArsrWrRrzrlkeysr6ruZ create_taskr`Z as_completedr)rrdr_opsr?Z send_resultrrrrgks zInMemoryChannelLayer.group_send)r.rir-N)r~)r'r(r)r*r extensionsr`rNrarzrbrrrerfrg __classcell__rrrnrrhs"    rhcCs&z t|WStk r YdSXdS)zL Returns a channel layer by alias, or None if it is not configured. N)channel_layersr)aliasrrrget_channel_layer~s r)rur9rr7rrxcopyrZ django.confrZdjango.core.signalsrZdjango.utils.module_loadingrrkr exceptionsrr r r,rhrrrrrrs"     Mv*