F2jG< "dZddlmZddlZddlZddlZddlZddlmZddl m Z ddlm Z m Z m Z mZmZejrddlmZmZgd Zd*d Zd+dZd,dZGddejeejejee ffeje ZGddee ZGddee ZGddZGd d!Z Gd"d#eeZ!Gd$d%eeZ"Gd&d'e eZ#Gd(d)e eZ$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)iscoroutinefunction)_HVAsyncHookCallableType AsyncHookTypeHookCallableTypeHookType)PreparedRequestResponse pre_requestpre_send on_uploadearly_responseresponsereturn HookType[_HV]c$dtDS)Nci|]}|gSr).0events V/home/thesage/.hermes/hermes-agent/venv/lib/python3.11/site-packages/niquests/hooks.py z!default_hooks..4s ) ) )%E2 ) ) ))HOOKSrrr default_hooksr3s ) )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 TypeErrorr r"r$r% callableshook _hook_datas r dispatch_hookr/7s }49IIcNNI ' I   $" I ' 'D -!T)66v66  - - -!T)__  -%& s =AAAsyncHookType[_HV] | NonecDK||S||}|rt|r|g}|D]m}t|r2 ||fi|d{V}nG#t$r||d{V}Yn*wxYw ||fi|}n#t$r||}YnwxYw||}n|S)zEDispatches a hook dictionary on a given piece of data asynchronously.N)r(r)rr*r+s rasync_dispatch_hookr2Ls/ }QVQZQZ[^Q_Q_I' I   $" I ' 'D"4(( 17'+tI'@'@'@'@!@!@!@!@!@!@JJ 777'+tI!6!6!6!6!6!6JJJ71!%i!:!:6!:!:JJ 111!%iJJJ1%& s$AA21A26 BBBcNeZdZddZddZdd Zdd Zdd Zd ZdZ dZ dZ dS)_BaseLifeCycleHookrNonec gggggd|_dSNr_storeselfs r__init__z_BaseLifeCycleHook.__init__ks%  f f  rr str | bytesvalue8list[HookCallableType[_HV] | AsyncHookCallableType[_HV]]c tdNzLifeCycleHook is Read OnlyNotImplementedError)r;r r>s r __setitem__z_BaseLifeCycleHook.__setitem__t!">???rr!c|j|SNr8r;r s r __getitem__z_BaseLifeCycleHook.__getitem__ws{3rc||SrGrrHs rr(z_BaseLifeCycleHook.getzs Cyrct|tsti}t}|jD]%\}}|||<||xx|j|z cc<&||_|SrG) isinstancer4r*r9items)r;other tmp_storecombined_hookshfnss r__add__z_BaseLifeCycleHook.__add__}s%!344 O 2D2F2Fk'')) , ,FAsIaL aLLLELO +LLLL )rc#$K|jEd{VdSrGr8r:s r__iter__z_BaseLifeCycleHook.__iter__s&;rc#,K|D]}|||fVdSrGrrHs rrMz_BaseLifeCycleHook.itemss6 ! !CtCy.  ! !rc tdrArBrHs r __delitem__z_BaseLifeCycleHook.__delitem__rErc*t|jSrG)lenr9r:s r__len__z_BaseLifeCycleHook.__len__s4;rNrr5)r r=r>r?rr5)r r!rr?)rr4) __name__ __module__ __qualname__r<rDrIr(rSrUrMrXr[rrrr4r4hs    @@@@        !!!@@@     rr4cLeZdZdZdfd Zdd Zdd Zdd ZddZddZ xZ S) LifeCycleHookzP A sync-only middleware to be used in your request/response lifecycles. rr5ct|j|jg|jg|jg|jg|jgddSr7 superr<r9updaterrrrrr; __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;rir%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.Nrrms 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.Nrrms 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_responserqrResponse | Nonec dSzJThe response generated from a Request. You may alter the response at will.Nrrvs rrzLifeCycleHook.responsernrr\rir r%r&rrjrir r%r&rr5rr r%r&rr5rr r%r&rrw r]r^r___doc__r<rrrrr __classcell__rgs@rrara          rracLeZdZdZdfd Zdd Zdd Zdd ZddZddZ xZ S)AsyncLifeCycleHookzR An async-only middleware to be used in your request/response lifecycles. rr5ct|j|jg|jg|jg|jg|jgddSr7rcrfs rr<zAsyncLifeCycleHook.__init__rhrrir r%r&rjc KdSrlrrms rrzAsyncLifeCycleHook.pre_request trc KdSrprrms rrzAsyncLifeCycleHook.pre_send trc KdSrsrrms rrzAsyncLifeCycleHook.on_uploads trrr c KdSrurrvs rrz!AsyncLifeCycleHook.early_responserrrwc KdSryrrvs rrzAsyncLifeCycleHook.responserrr\rzr{r|r}r~rs@rrrrrrcJeZdZUdZded<ded<ded<d d Zdd Zdd Zd S)_LeakyBucketMixinz$Shared leaky bucket algorithm logic.floatrateinterval float | None last_requestrr5c6||_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,,rNrrrr5)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_updaterrr5cr||_||n||_|j|_tj|_dSrG)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 rNrrrrrr5)rrr\)r]r^r_rrrrrrrrrr s..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: ... $@rrrr5ct||tj|_dSzjInitialize the leaky bucket limiter. Args: rate: Maximum requests per second N)rdr<r threadingLock_lockr;rrgs rr<zLeakyBucketLimiter.__init__>sB  %%%^%% rrir r%r&rjc |j5|}|dkrtj||dddn #1swxYwYdS)*Wait if needed to maintain the rate limit.rN)rrrsleeprr;rir%rs rrzLeakyBucketLimiter.pre_requestHs Z # #**,,I1}} 9%%%  " " "  # # # # # # # # # # # # # # # tsAAAArrrzr]r^r_rr<rrrs@rrr1s`  &&&&&&&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: ... rrrrr5ct||tj|_dSr)rdr<rasynciorrrs rr<z AsyncLeakyBucketLimiter.__init___s@  %%%\^^ rrir r%r&rjcK|j4d{V|}|dkrtj|d{V|dddd{Vn#1d{VswxYwYdS)rNr)rrrrrrs rrz#AsyncLeakyBucketLimiter.pre_requestis: # # # # # # # #**,,I1}}mI.........  " " "  # # # # # # # # # # # # # # # # # # # # # # # # # # # tsA A++ A58A5rrrzrrs@rrrRs`  $$$$$$$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: ... rNrrrrrr5ct|||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)rdr<rrrrr;rrrgs rr<zTokenBucketLimiter.__init__sD  h///^%% rrir r%r&rjc |j5|}|(tj||dddn #1swxYwYdSz1Wait until a token is available, then consume it.N)rrrrrrs rrzTokenBucketLimiter.pre_requests Z " "++--I$ 9%%%!!!  " " " " " " " " " " " " " " " ts?AAArNrrzrrs@rrrss`   & & & & & & &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: ... rNrrrrrr5ct|||tj|_dSr)rdr<rrrrrs rr<z AsyncTokenBucketLimiter.__init__sB  h///\^^ rrir r%r&rjcK|j4d{V|}|.tj|d{V|dddd{Vn#1d{VswxYwYdSr)rrrrrrs rrz#AsyncTokenBucketLimiter.pre_requests : " " " " " " " "++--I$mI.........!!!  " " " " " " " " " " " " " " " " " " " " " " " " " " " tsAA'' A14A1rrrzrrs@rrrs`   $ $ $ $ $ $ $rr)rr) r r!r"r#r$rr%r&rr) r r!r"r0r$rr%r&rr)%r __future__rrrrtypingcollections.abcr_compatrrrr r r TYPE_CHECKINGmodelsr r rrr/r2r!ListUnionGenericr4rarrrrrrrrrrrs*#""""" ******(((((( 211111111   *****8/ / / / /  #v{6<8HJ_8_+`aabdjdrsvdw/ / / d'''''&s+'''T'''''+C0'''T--------6%%%%%%%%P*MB/1CB*MD/1Cr