a a*j!@sxddlmZddlZddlmZmZmZddlmZddl m Z ddl m Z m Z mZmZerdddlZGdddeZdS) ) annotationsN) TYPE_CHECKINGTypecast) RedisBridge)ConfigurationError)AsyncRedisClientCallableOptionalUnionc@seZdZUddiZded<eddddZd d d d d dddZd d dddZd d dddZ ded<ded<ded<ded<ded<ded<ded<dJd d d!d"d#Z d dd$d%Z dKd'd(d d(d(d)d*d+Z d'd(d,d-d.Z d'd d,d/d0Zd1dd2d3Zd'd(d(d4d5d6d7Zd'd'd(d8d9d:d;ZdLd'd(d(d(d d<d=d>ZdMd'd'd(d(d(d d?d@dAZd'dBd,dCdDZd ddEdFZd1ddGdHZdIS)N RedispyBridgeZmax_connectionsz"dict[str, Union[float, str, bool]]DEFAULT_CLUSTER_OPTIONSz3Union[Type[Exception], tuple[Type[Exception], ...]])returncCs |jjfSN) dependencyZ RedisErrorselfr\/var/www/html/assistant/venv/lib/python3.9/site-packages/limits/aio/storage/redis/redispy.pybase_exceptionsszRedispyBridge.base_exceptionsz Optional[str]boolz,Optional[dict[str, Union[str, float, bool]]]zUnion[str, float, bool]None) service_name use_replicassentinel_kwargsoptionsrc  sg}|}jjdd}jj|ddD]$}|d\} } || t| fq4jjrrjjddn|}|durt dj j j |fdij |piiij |_j|_j|_fd d _dS) N@,:/z'service_name' not providedrcs|rrjSjSr)storage_replicastoragereadonlyrrrr8sz,RedispyBridge.use_sentinel..)copy parsed_urinetlocfindsplitappendintpathreplacerrasyncioZSentinel parsed_authsentinelZ master_forr$Z slave_forr#connection_getter) rrrrrZsentinel_configurationZconnection_optionsseplochostportrr'r use_sentinels, zRedispyBridge.use_sentinel)rrc sZ|dd}r,jjjfd|i|_njjjjjfi|_fdd_dS)Nconnection_poolcsjSrr$_rrrr(Dz)RedispyBridge.use_basic..)poprr2ZRedisr$Zfrom_urlurir5)rrr;rrr use_basic<s  zRedispyBridge.use_basicc sjjdd}g}jj|ddD]0}|d\}}|jjjj|t |dq,jjj fd|iij j |_ fdd_dS) Nrrrr )r8r9Z startup_nodescsjSrr<r=rrrr(Tr?z+RedispyBridge.use_cluster..)r*r+r,r-r.rr2clusterZ ClusterNoder/ RedisClusterrr3r$r5)rrr6Z cluster_hostsr7r8r9rrr use_clusterFs zRedispyBridge.use_clusterz'redis.commands.core.Script'lua_moving_windowlua_acquire_moving_windowlua_sliding_windowlua_acquire_sliding_windowlua_clear_keyslua_incr_expirez"Callable[[bool], AsyncRedisClient]r5Fr)r&rcCs ||Sr)r5)rr&rrrget_connection^szRedispyBridge.get_connectioncCsp||j|_||j|_||j|_||j|_ ||j |_ ||j |_ dSr)rLZregister_scriptZSCRIPT_MOVING_WINDOWrFZSCRIPT_ACQUIRE_MOVING_WINDOWrGZSCRIPT_CLEAR_KEYSrJZSCRIPT_INCR_EXPIRErKZSCRIPT_SLIDING_WINDOWrHZSCRIPT_ACQUIRE_SLIDING_WINDOWrIrrrrregister_scriptsas$zRedispyBridge.register_scriptsrstrr/)keyexpiryelastic_expiryamountrcs`||}|r>|||IdH}|||IdH|Stt||g||gIdHSdS)z increments the counter for a given rate limit key :param key: the key to increment :param expiry: amount in seconds for the key to expire in :param amount: the number to increment by N) prefixed_keyrLZincrbyZexpirerr/rK)rrOrPrQrRvaluerrrincrvs  zRedispyBridge.incr)rOrcs*||}t|jdd|IdHp&dS)zC :param key: the key to get the counter value for Tr%Nr)rSr/rLgetrrOrrrrVs zRedispyBridge.getcs"||}||IdHdS)z? :param key: the key to clear rate limits for N)rSrLdeleterWrrrclears zRedispyBridge.clearz Optional[int]cstt||dgIdHS)N*)rr/rJrSrrrr lua_resetszRedispyBridge.lua_resetztuple[float, int])rOlimitrPrcsN||}t}||g|||gIdH}|rFt|d|dfS|dfS)z returns the starting point and the number of entries in the moving window :param key: rate limit key :param expiry: expiry of entry :return: (previous count, previous TTL, current count, current TTL) Nrr)rStimerFfloat)rrOr\rP timestampwindowrrrget_moving_windows zRedispyBridge.get_moving_windowztuple[int, float, int, float]) previous_key current_keyrPrcs||||||g|gIdH}rxt|dp4dtdt|dpFddt|dpZdtdt|dplddfSdS)Nrrr )rrrf)rHrSr/maxr^)rrbrcrPr`rrrget_sliding_windowsz RedispyBridge.get_sliding_window)rOr\rPrRrcs6||}t}||g||||gIdH}t|S)z :param key: rate limit key to acquire an entry in :param limit: amount of entries allowed :param expiry: expiry of the entry N)rSr]rGr)rrOr\rPrRr_acquiredrrr acquire_entrys  zRedispyBridge.acquire_entry)rbrcr\rPrRrcs8||}||}|||g|||gIdH}t|Sr)rSrIr)rrbrcr\rPrRrirrracquire_sliding_window_entrys    z*RedispyBridge.acquire_sliding_window_entryr^cs,||}t||IdHdtS)z; :param key: the key to get the expiry for Nr)rSrgrLttlr]rWrrr get_expirys zRedispyBridge.get_expirycs,z|IdHWdSYdS0dS)z- check if storage is healthy NTF)rLZpingrrrrchecks zRedispyBridge.checkcsR|d}|jj||jjjjjdIdH}d}|D]}||j|IdH7}q2|S)NrZ)Z target_nodesr) rSr$keysrr2rCrDZ ALL_NODESrX)rprefixrocountrOrrrresets  zRedispyBridge.resetN)F)Fr)r)r)__name__ __module__ __qualname__r__annotations__propertyrr:rBrErLrMrUrVrYr[rarhrjrkrmrnrrrrrrr s>  $   r ) __future__rr]typingrrrZlimits.aio.storage.redis.bridgerZ limits.errorsrZ limits.typingrr r r Zredis.commandsZredisr rrrrs