o ri @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 RedisErrorselfr_/var/www/html/pca-backend/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}|}|jjjjrjjddn|}|dur#tdjjj |fdij |p1iiij |_ j |_ j |_fdd_dS)N/z'service_name' not providedrcs|rrjSjSr)storage_replicastoragereadonlyrrrr4sz,RedispyBridge.use_sentinel..)copyextendoptions_from_uri locationspathreplacerrasyncioSentinel parsed_authsentinel master_forr# slave_forr"connection_getter)rrrrrsentinel_configurationconnection_optionsrr&r use_sentinels& zRedispyBridge.use_sentinelc sz|dd}rjjjdd|i|_njjr&jjjdi|_njjjjjfi|_fdd_ dS)Nconnection_poolcjSrr#_rrrr'Ez)RedispyBridge.use_basic..r) poprr.Redisr#r*emptyfrom_urlurir4)rrr8rrr use_basic8s   zRedispyBridge.use_basicc stg}jjD]\}}|jjjj|t|dqjjjdiij d|ij |_ fdd_ dS)N)hostport startup_nodescr9rr:r;rrrr'Wr=z+RedispyBridge.use_cluster..r) r*r+appendrr.cluster ClusterNodeint RedisClusterr r0r#r4)rr cluster_hostsrDrErrr use_clusterGs" zRedispyBridge.use_clusterzredis.commands.core.Scriptlua_moving_windowlua_acquire_moving_windowlua_sliding_windowlua_acquire_sliding_windowlua_clear_keyslua_incr_expirez"Callable[[bool], AsyncRedisClient]r4Fr%rcCs ||Sr)r4)rr%rrrget_connectionas zRedispyBridge.get_connectioncCsp||j|_||j|_||j|_||j|_ ||j |_ ||j |_ dSr)rTregister_scriptSCRIPT_MOVING_WINDOWrNSCRIPT_ACQUIRE_MOVING_WINDOWrOSCRIPT_CLEAR_KEYSrRSCRIPT_INCR_EXPIRErSSCRIPT_SLIDING_WINDOWrPSCRIPT_ACQUIRE_SLIDING_WINDOWrQrrrrregister_scriptsds$ zRedispyBridge.register_scriptskeystrexpiryrJamountcs*||}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_keyrrJrS)rr^r`rarrrincrys zRedispyBridge.incrcs,||}t|jdd|IdHpdS)zC :param key: the key to get the counter value for Tr$Nr)rbrJrTgetrr^rrrrds  zRedispyBridge.getcs$||}||IdHdS)z? :param key: the key to clear rate limits for N)rbrTdeletererrrclears zRedispyBridge.clear int | Nonecs tt||dgIdHS)N*)rrJrRrbrrrr 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])rbtimerNfloat)rr^rkr` 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 )rrrw)rPrbrJmaxrn)rrrrsr`rprrrget_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)rbrmrOr)rr^rkr`raroacquiredrrr acquire_entrys  zRedispyBridge.acquire_entrycs:||}||}|||g|||gIdH}t|Sr)rbrQr)rrrrsrkr`rarzrrracquire_sliding_window_entrys   z*RedispyBridge.acquire_sliding_window_entryrncs.||}t||IdHdtS)z; :param key: the key to get the expiry for Nr)rbrxrTttlrmrerrr get_expirys "zRedispyBridge.get_expirycs(z |IdHWdSYdS)z- check if storage is healthy NTF)rTpingrrrrchecks zRedispyBridge.checkcsT|d}|jj||jjjjjdIdH}d}|D] }||j|IdH7}q|S)Nri) target_nodesr) rbr#keysrr.rHrK ALL_NODESrf)rprefixrcountr^rrrresets  zRedispyBridge.resetN)r r) rrrrrrrrr r)rrr r)F)r%rr r)r r)r])r^r_r`rJrarJr rJ)r^r_r rJ)r^r_r r)r rh)r^r_rkrJr`rJr rl)rrr_rsr_r`rJr rt) r^r_rkrJr`rJrarJr r) rrr_rsr_rkrJr`rJrarJr r)r^r_r rn)r r)__name__ __module__ __qualname__r __annotations__propertyrr7rCrMrTr\rcrdrgrjrqryr{r|r~rrrrrrr s@               r ) __future__rrmtypingrrlimits.aio.storage.redis.bridger limits.errorsr limits.typingrrredis.commandsredisr rrrrs