o KDiv @slddlmZddlZddlmZmZddlmZddlm Z ddl m Z m Z er,ddl ZGdddeZdS) ) annotationsN) TYPE_CHECKINGcast) RedisBridge)ConfigurationError)AsyncRedisClientCallablec@seZdZUddiZded< edLddZdMddZdNddZdNddZ ded<ded<ded<ded<ded<ded<ded <dOdPd$d%Z dQd&d'Z (dRdSd.d/Z dTd0d1Z dUd2d3ZdVd5d6ZdWd9d:ZdXd>d?Z (dRdYd@dAZ (dRdZdBdCZd[dEdFZd\dGdHZdVdIdJZdKS)] RedispyBridgemax_connectionszdict[str, float | str | bool]DEFAULT_CLUSTER_OPTIONSreturn-type[Exception] | tuple[type[Exception], ...]cCs |jjfSN) dependency RedisErrorselfrY/var/www/Datamplify/venv/lib/python3.10/site-packages/limits/aio/storage/redis/redispy.pybase_exceptionss zRedispyBridge.base_exceptions service_name str | None use_replicasboolsentinel_kwargs$dict[str, str | float | bool] | Noneoptionsstr | float | boolNonec  sg}|}jjdd}jj|ddD]}|d\} } || t| fqjjr9jjddn|}|durCt dj j j |fdij |pQiiij |_j|_j|_fd d _dS) N@,:/z'service_name' not providedrcs|rrjSjSr)storage_replicastoragereadonlyrrrr8sz,RedispyBridge.use_sentinel..)copy parsed_urinetlocfindsplitappendintpathreplacerrasyncioSentinel parsed_authsentinel master_forr' slave_forr&connection_getter) rrrrrsentinel_configurationconnection_optionsseplochostportrr*r use_sentinels, zRedispyBridge.use_sentinelc sZ|dd}rjjjdd|i|_njjjjjfi|_fdd_dS)Nconnection_poolcjSrr'_rrrr+Dz)RedispyBridge.use_basic..r)poprr5Redisr'from_urlurir;)rrrCrrr use_basic<s  zRedispyBridge.use_basicc sjjdd}g}jj|ddD]}|d\}}|jjjj|t |dqjjj d d|iij j |_ fdd_dS) Nr r!r"r#)r@rA startup_nodescrDrrErFrrrr+TrHz+RedispyBridge.use_cluster..r)r-r.r/r0r1rr5cluster ClusterNoder2 RedisClusterr r7r'r;)rrr> cluster_hostsr?r@rArrr use_clusterFs zRedispyBridge.use_clusterzredis.commands.core.Scriptlua_moving_windowlua_acquire_moving_windowlua_sliding_windowlua_acquire_sliding_windowlua_clear_keyslua_incr_expirez"Callable[[bool], AsyncRedisClient]r;Fr)rcCs ||Sr)r;)rr)rrrget_connection^s zRedispyBridge.get_connectioncCsp||j|_||j|_||j|_||j|_ ||j |_ ||j |_ dSr)rZregister_scriptSCRIPT_MOVING_WINDOWrTSCRIPT_ACQUIRE_MOVING_WINDOWrUSCRIPT_CLEAR_KEYSrXSCRIPT_INCR_EXPIRErYSCRIPT_SLIDING_WINDOWrVSCRIPT_ACQUIRE_SLIDING_WINDOWrWrrrrregister_scriptsas$ zRedispyBridge.register_scriptsr!keystrexpiryr2amountcs*||}tt||g||gIdHS)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_keyrr2rY)rrcrerfrrrincrvs zRedispyBridge.incrcs,||}t|jdd|IdHpdS)zC :param key: the key to get the counter value for Tr(Nr)rgr2rZgetrrcrrrris  zRedispyBridge.getcs$||}||IdHdS)z? :param key: the key to clear rate limits for N)rgrZdeleterjrrrclears zRedispyBridge.clear int | Nonecs tt||dgIdHS)N*)rr2rXrgrrrr lua_resetszRedispyBridge.lua_resetlimittuple[float, int]csP||}t}||g|||gIdH}|r$t|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!)rgtimerTfloat)rrcrpre timestampwindowrrrget_moving_windows zRedispyBridge.get_moving_window previous_key current_keytuple[int, float, int, float]cs~|||||g|gIdH}r=t|dpdtdt|dp$ddt|dp.dtdt|dp7ddfSdS)Nrr!r )rrr|)rVrgr2maxrs)rrwrxrerurrrget_sliding_windowsz RedispyBridge.get_sliding_windowcs8||}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)rgrrrUr)rrcrprerfrtacquiredrrr acquire_entrys  zRedispyBridge.acquire_entrycs:||}||}|||g|||gIdH}t|Sr)rgrWr)rrwrxrprerfrrrracquire_sliding_window_entrys   z*RedispyBridge.acquire_sliding_window_entryrscs.||}t||IdHdtS)z; :param key: the key to get the expiry for Nr)rgr}rZttlrrrjrrr get_expirys "zRedispyBridge.get_expirycs(z |IdHWdSYdS)z- check if storage is healthy NTF)rZpingrrrrchecks zRedispyBridge.checkcsT|d}|jj||jjjjjdIdH}d}|D] }||j|IdH7}q|S)Nrn) target_nodesr) rgr'keysrr5rOrQ ALL_NODESrk)rprefixrcountrcrrrresets  zRedispyBridge.resetN)r r) rrrrrrrrr r)rrr r)F)r)rr r)r r)r!)rcrdrer2rfr2r r2)rcrdr r2)rcrdr r)r rm)rcrdrpr2rer2r rq)rwrdrxrdrer2r ry) rcrdrpr2rer2rfr2r r) rwrdrxrdrpr2rer2rfr2r r)rcrdr rs)r r)__name__ __module__ __qualname__r __annotations__propertyrrBrMrSrZrbrhrirlrorvr~rrrrrrrrrr s@     $           r ) __future__rrrtypingrrlimits.aio.storage.redis.bridger limits.errorsr limits.typingrrredis.commandsredisr rrrrs