o ;hv @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)] RedispyBridgeZmax_connectionszdict[str, float | str | bool]DEFAULT_CLUSTER_OPTIONSreturn-type[Exception] | tuple[type[Exception], ...]cCs |jjfSN) dependencyZ RedisErrorselfrS/var/www/html/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_urinetlocfindsplitappendintpathreplacerrasyncioZSentinel parsed_authsentinelZ master_forr%Z slave_forr$connection_getter) rrrrrZsentinel_configurationZconnection_optionsseplochostportrr(r use_sentinels, zRedispyBridge.use_sentinelc sZ|dd}rjjjdd|i|_njjjjjfi|_fdd_dS)Nconnection_poolcjSrr%_rrrr)Dz)RedispyBridge.use_basic..r)poprr3ZRedisr%Zfrom_urlurir6)rrr<rrr use_basic<s  zRedispyBridge.use_basicc sjjdd}g}jj|ddD]}|d\}}|jjjj|t |dqjjj d d|iij j |_ fdd_dS) Nrrr r!)r9r:Z startup_nodescr=rr>r?rrrr)TrAz+RedispyBridge.use_cluster..r)r+r,r-r.r/rr3clusterZ ClusterNoder0 RedisClusterr r4r%r6)rrr7Z cluster_hostsr8r9r:rrr 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]r6Fr'rcCs ||Sr)r6)rr'rrrget_connection^s zRedispyBridge.get_connectioncCsp||j|_||j|_||j|_||j|_ ||j |_ ||j |_ dSr)rNZregister_scriptZSCRIPT_MOVING_WINDOWrHZSCRIPT_ACQUIRE_MOVING_WINDOWrIZSCRIPT_CLEAR_KEYSrLZSCRIPT_INCR_EXPIRErMZSCRIPT_SLIDING_WINDOWrJZSCRIPT_ACQUIRE_SLIDING_WINDOWrKrrrrregister_scriptsas$ zRedispyBridge.register_scriptsrkeystrexpiryr0amountcs*||}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_keyrr0rM)rrPrRrSrrrincrvs zRedispyBridge.incrcs,||}t|jdd|IdHpdS)zC :param key: the key to get the counter value for Tr&Nr)rTr0rNgetrrPrrrrVs  zRedispyBridge.getcs$||}||IdHdS)z? :param key: the key to clear rate limits for N)rTrNdeleterWrrrclears zRedispyBridge.clear int | Nonecs tt||dgIdHS)N*)rr0rLrTrrrr 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)rTtimerHfloat)rrPr]rR 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)Nrrr )rrri)rJrTr0maxr`)rrdrerRrbrrrget_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)rTr_rIr)rrPr]rRrSraacquiredrrr acquire_entrys  zRedispyBridge.acquire_entrycs:||}||}|||g|||gIdH}t|Sr)rTrKr)rrdrer]rRrSrlrrracquire_sliding_window_entrys   z*RedispyBridge.acquire_sliding_window_entryr`cs.||}t||IdHdtS)z; :param key: the key to get the expiry for Nr)rTrjrNttlr_rWrrr get_expirys "zRedispyBridge.get_expirycs(z |IdHWdSYdS)z- check if storage is healthy NTF)rNZpingrrrrchecks zRedispyBridge.checkcsT|d}|jj||jjjjjdIdH}d}|D] }||j|IdH7}q|S)Nr[)Z target_nodesr) rTr%keysrr3rErFZ ALL_NODESrX)rprefixrrcountrPrrrresets  zRedispyBridge.resetN)r r ) rrrrrrrrr r)rrr r)F)r'rr r)r r)r)rPrQrRr0rSr0r r0)rPrQr r0)rPrQr r)r rZ)rPrQr]r0rRr0r r^)rdrQrerQrRr0r rf) rPrQr]r0rRr0rSr0r r) rdrQrerQr]r0rRr0rSr0r r)rPrQr r`)r r)__name__ __module__ __qualname__r __annotations__propertyrr;rDrGrNrOrUrVrYr\rcrkrmrnrprqrurrrrr s@     $           r ) __future__rr_typingrrZlimits.aio.storage.redis.bridgerZ limits.errorsrZ limits.typingrrZredis.commandsZredisr rrrrs