i&< dZddlmZddlZddlZddlZddlZddlmZddlm Z m Z m Z m Z m Z ejrddlmZmZgdZd)d Zd*dZd+dZGddejeejeje e ffeje ZGddee ZGddee ZGddZGdd ZGd!d"eeZGd#d$eeZ Gd%d&eeZ!Gd'd(eeZ"dS),a requests.hooks ~~~~~~~~~~~~~~ This module provides the capabilities for the Requests hooks system. Available hooks: ``pre_request``: The prepared request just got built. You may alter it prior to be sent through HTTP. ``pre_send``: The prepared request got his ConnectionInfo injected. This event is triggered just after picking a live connection from the pool. ``on_upload``: Permit to monitor the upload progress of passed body. This event is triggered each time a block of data is transmitted to the remote peer. Use this hook carefully as it may impact the overall performance. ``response``: The response generated from a Request. ) annotationsN)MutableMapping)_HVAsyncHookCallableType AsyncHookTypeHookCallableTypeHookType)PreparedRequestResponse pre_requestpre_send on_uploadearly_responseresponsereturn HookType[_HV]c$dtDS)Nci|]}|gSr).0events I/root/projects/butler/venv/lib/python3.11/site-packages/niquests/hooks.py z!default_hooks..3s ) ) )%E2 ) ) ))HOOKSrrr default_hooksr2s ) )5 ) ) ))rkeystrhooksHookType[_HV] | None hook_datarkwargs typing.Anyc ||S||}|rAt|r|g}|D],} ||fi|}n#t$r||}YnwxYw||}-|S)z6Dispatches a hook dictionary on a given piece of data.)getcallable TypeErrorrr!r#r$ callableshook _hook_datas r dispatch_hookr.6s }49IIcNNI ' I   $" I ' 'D -!T)66v66  - - -!T)__  -%& s =AAAsyncHookType[_HV] | NonecNK||S||}|rt|r|g}|D]r}tj|r2 ||fi|d{V}nG#t$r||d{V}Yn*wxYw ||fi|}n#t$r||}YnwxYw||}s|S)zEDispatches a hook dictionary on a given piece of data asynchronously.N)r'r(asyncioiscoroutinefunctionr)r*s rasync_dispatch_hookr3Ks2 }QVQZQZ[^Q_Q_I' I   $" I ' 'D*400 17'+tI'@'@'@'@!@!@!@!@!@!@JJ 777'+tI!6!6!6!6!6!6JJJ71!%i!:!:6!:!:JJ 111!%iJJJ1%& s$ AA76A7; BBBcNeZdZddZddZdd Zdd Zdd Zd ZdZ dZ dZ dS)_BaseLifeCycleHookrNonec gggggd|_dSNr _storeselfs r__init__z_BaseLifeCycleHook.__init__js%  f f  rr str | bytesvalue8list[HookCallableType[_HV] | AsyncHookCallableType[_HV]]c tdNzLifeCycleHook is Read OnlyNotImplementedError)r<rr?s r __setitem__z_BaseLifeCycleHook.__setitem__s!">???rr c|j|SNr9r<rs r __getitem__z_BaseLifeCycleHook.__getitem__vs{3rc||SrHrrIs rr'z_BaseLifeCycleHook.getys Cyrct|tsti}t}|jD]%\}}|||<||xx|j|z cc<&||_|SrH) isinstancer5r)r:items)r<other tmp_storecombined_hookshfnss r__add__z_BaseLifeCycleHook.__add__|s%!344 O 2D2F2Fk'')) , ,FAsIaL aLLLELO +LLLL )rc#$K|jEd{VdSrHr9r;s r__iter__z_BaseLifeCycleHook.__iter__s&;rc#,K|D]}|||fVdSrHrrIs rrNz_BaseLifeCycleHook.itemss6 ! !CtCy.  ! !rc tdrBrCrIs r __delitem__z_BaseLifeCycleHook.__delitem__rFrc*t|jSrH)lenr:r;s r__len__z_BaseLifeCycleHook.__len__s4;rNrr6)rr>r?r@rr6)rr rr@)rr5) __name__ __module__ __qualname__r=rErJr'rTrVrNrYr\rrrr5r5gs    @@@@        !!!@@@     rr5cLeZdZdZdfd Zdd Zdd Zdd ZddZddZ xZ S) LifeCycleHookzP A sync-only middleware to be used in your request/response lifecycles. rr6ct|j|jg|jg|jg|jg|jgddSr8 superr=r:updaterrrrrr< __class__s rr=zLifeCycleHook.__init__o   $ 01!]O"n-#'#6"7!]O        rprepared_requestr r$r%PreparedRequest | Nonec dSzTThe prepared request just got built. You may alter it prior to be sent through HTTP.Nrr<rjr$s rrzLifeCycleHook.pre_requesttrc dSzThe prepared request got his ConnectionInfo injected. This event is triggered just after picking a live connection from the pool. You may not alter the prepared request.Nrrns rrzLifeCycleHook.pre_send trc dSaPermit to monitor the upload progress of passed body. This event is triggered each time a block of data is transmitted to the remote peer. Use this hook carefully as it may impact the overall performance. You may not alter the prepared request.Nrrns rrzLifeCycleHook.on_uploads trrr c dSzAn early response caught before receiving the final Response for a given Request. Like but not limited to 103 Early Hints.Nrr<rr$s rrzLifeCycleHook.early_responserrrResponse | Nonec dSzJThe response generated from a Request. You may alter the response at will.Nrrws rrzLifeCycleHook.responserorr]rjr r$r%rrkrjr r$r%rr6rr r$r%rr6rr r$r%rrx r^r_r`__doc__r=rrrrr __classcell__rhs@rrbrb          rrbcLeZdZdZdfd Zdd Zdd Zdd ZddZddZ xZ S)AsyncLifeCycleHookzR An async-only middleware to be used in your request/response lifecycles. rr6ct|j|jg|jg|jg|jg|jgddSr8rdrgs rr=zAsyncLifeCycleHook.__init__rirrjr r$r%rkc KdSrmrrns rrzAsyncLifeCycleHook.pre_request trc KdSrqrrns rrzAsyncLifeCycleHook.pre_send trc KdSrtrrns rrzAsyncLifeCycleHook.on_uploads trrr c KdSrvrrws rrz!AsyncLifeCycleHook.early_responserrrxc KdSrzrrws rrzAsyncLifeCycleHook.responserrr]r{r|r}r~rrs@rrrrrrcJeZdZUdZded<ded<ded<d d Zdd Zdd Zd S)_LeakyBucketMixinz$Shared leaky bucket algorithm logic.floatrateinterval float | None last_requestrr6c6||_d|z |_d|_dS)N?)rrr)r<rs r_init_leaky_bucketz$_LeakyBucketMixin._init_leaky_buckets" d   rchtj}|j||jz }|j|z }nd}|S)zDCompute wait time and update state. Returns wait time (may be <= 0).Ng)time monotonicrrr<nowelapsed wait_times r _compute_waitz_LeakyBucketMixin._compute_waits>n   (D--G /IIIrc6tj|_dS)zRecord that a request was made.N)rrrr;s r_record_requestz!_LeakyBucketMixin._record_requests N,,rNrrrr6)rrr])r^r_r`r__annotations__rrrrrrrrsv..KKKOOO!!!! ------rrcTeZdZUdZded<ded<ded<ded<dd Zdd Zdd Zd S)_TokenBucketMixinz$Shared token bucket algorithm logic.rrcapacitytokens last_updaterrr6cr||_||n||_|j|_tj|_dSrH)rrrrrr)r<rrs r_init_token_bucketz$_TokenBucketMixin._init_token_buckets8 $,$8d m >++rctj}||jz }t|j|j||jzz|_||_|jdkr|xjdzc_dSd|jz |jz }|S)zUReplenish tokens and try to acquire one. Returns wait time if needed, None otherwise.rNrrrminrrrrs r_acquire_tokenz _TokenBucketMixin._acquire_tokensn(($-w7J)JKK  ;#   KK3 KK4t{*di7I rctj}||jz }t|j|j||jzz|_||_|xjdzc_dS)z*Called after waiting to consume the token.rNr)r<rrs r _post_waitz_TokenBucketMixin._post_wait%sYn(($-w7J)JKK  s rNrrrrrr6)rrr])r^r_r`rrrrrrrrrrs..KKKOOOMMM,,,,    rrc.eZdZdZddfd Zdd ZxZS)LeakyBucketLimiterapRate limiter using the leaky bucket algorithm. Requests "leak" out at a constant rate. When a request arrives, it waits until enough time has passed since the last request to maintain the rate. Usage:: limiter = LeakyBucketLimiter(rate=10.0) # 10 requests per second with niquests.Session(hooks=limiter) as session: ... $@rrrr6ct||tj|_dSzjInitialize the leaky bucket limiter. Args: rate: Maximum requests per second N)rer=r threadingLock_lockr<rrhs rr=zLeakyBucketLimiter.__init__=sB  %%%^%% rrjr r$r%rkc |j5|}|dkrtj||dddn #1swxYwYdS)*Wait if needed to maintain the rate limit.rN)rrrsleeprr<rjr$rs rrzLeakyBucketLimiter.pre_requestGs Z # #**,,I1}} 9%%%  " " "  # # # # # # # # # # # # # # # tsAAAArrr{r^r_r`rr=rrrs@rrr0s`  &&&&&&&rrc.eZdZdZddfd Zdd ZxZS)AsyncLeakyBucketLimiteraRate limiter using the leaky bucket algorithm. Requests "leak" out at a constant rate. When a request arrives, it waits until enough time has passed since the last request to maintain the rate. Usage:: limiter = AsyncLeakyBucketLimiter(rate=10.0) # 10 requests per second async with niquests.AsyncSession(hooks=limiter) as session: ... rrrrr6ct||tj|_dSr)rer=rr1rrrs rr=z AsyncLeakyBucketLimiter.__init__^s@  %%%\^^ rrjr r$r%rkcK|j4d{V|}|dkrtj|d{V|dddd{Vn#1d{VswxYwYdS)rNr)rrr1rrrs rrz#AsyncLeakyBucketLimiter.pre_requesths: # # # # # # # #**,,I1}}mI.........  " " "  # # # # # # # # # # # # # # # # # # # # # # # # # # # tsA A++ A58A5rrr{rrs@rrrQs`  $$$$$$$rrc.eZdZdZddfd ZddZxZS)TokenBucketLimitera{Rate limiter using the token bucket algorithm. Tokens are added to a bucket at a constant rate up to a maximum capacity. Each request consumes one token. Allows bursts up to the bucket capacity. Usage:: limiter = TokenBucketLimiter(rate=10.0, capacity=50.0) # 10/s, burst of 50 with niquests.Session(hooks=limiter) as session: ... rNrrrrrr6ct|||tj|_dSzInitialize the token bucket limiter. Args: rate: Token replenishment rate (tokens per second) capacity: Maximum bucket capacity (defaults to rate, allowing 1 second burst) N)rer=rrrrr<rrrhs rr=zTokenBucketLimiter.__init__sD  h///^%% rrjr r$r%rkc |j5|}|(tj||dddn #1swxYwYdSz1Wait until a token is available, then consume it.N)rrrrrrs rrzTokenBucketLimiter.pre_requests Z " "++--I$ 9%%%!!!  " " " " " " " " " " " " " " " ts?AAArNrr{rrs@rrrrs`   & & & & & & &rrc.eZdZdZddfd ZddZxZS)AsyncTokenBucketLimiteraRate limiter using the token bucket algorithm. Tokens are added to a bucket at a constant rate up to a maximum capacity. Each request consumes one token. Allows bursts up to the bucket capacity. Usage:: limiter = AsyncTokenBucketLimiter(rate=10.0, capacity=50.0) # 10/s, burst of 50 async with niquests.AsyncSession(hooks=limiter) as session: ... rNrrrrrr6ct|||tj|_dSr)rer=rr1rrrs rr=z AsyncTokenBucketLimiter.__init__sB  h///\^^ rrjr r$r%rkcK|j4d{V|}|.tj|d{V|dddd{Vn#1d{VswxYwYdSr)rrr1rrrs rrz#AsyncTokenBucketLimiter.pre_requests : " " " " " " " "++--I$mI.........!!!  " " " " " " " " " " " " " " " " " " " " " " " " " " " tsAA'' A14A1rrr{rrs@rrrs`   $ $ $ $ $ $ $rr)rr) rr r!r"r#rr$r%rr) rr r!r/r#rr$r%rr)#r __future__rr1rrtypingcollections.abcrrrrr r TYPE_CHECKINGmodelsr r rrr.r3r ListUnionGenericr5rbrrrrrrrrrrrs*#""""" ****** 211111111   *****8/ / / / /  #v{6<8HJ_8_+`aabdjdrsvdw/ / / d'''''&s+'''T'''''+C0'''T--------6%%%%%%%%P*MB/1CB*MD/1Cr