Bj-UdZddlmZddlmZddlmZddlmZddlmZddl m Z ddl m Z dd l m Z dd l m Z dd l mZdd lZd dlmZd dlmZd dlmZd dlmZd dlmZd dlmZeeeeeeefegeeezfZded<e GddZeZ ee!eeefee gd fd zfZ"ded<GddZ#e#Z$Gdde#Z%y )z Hook callers and relay. ) annotations)Callable)Mapping)Sequence)Set)Any)Final)final) TYPE_CHECKING) TypeAliasN) HookimplOpts) HookspecOpts) _Namespace)HookSpec)_Plugin)HookImplr _HookExecc*eZdZdZdZddZerddZyy) HookRelayzaHook holder object for performing 1:N hook calls where N is the number of registered plugins.)__dict__cyz:meta private:Nselfs Q/root/repos/pocs/video-evals/.venv/lib/python3.12/site-packages/pluggy/_caller.py__init__zHookRelay.__init__'cyNr)rnames r __getattr__zHookRelay.__getattr__,rr N)returnNone)r#strr% HookCaller)__name__ __module__ __qualname____doc__ __slots__rr r$rr rrr s I;r r _CallHistoryceZdZdZdZ d ddZddZ ddZddZddZ dd Z dd Z dd Z dd Z dd ZddZ d ddZ ddZddZy) r(zCA caller of all registered implementations of a hook specification.) _call_history _hookexec _hookimplsr#specNc|||_||_g|_d|_d|_||J|j ||yyr)r#r1r2r0r3set_specification)rr# hook_executespecmodule_or_class spec_optss rrzHookCaller.__init__CsR  ,2426%)  *( ((  " "#6 B +r c|jduSr")r3rs rhas_speczHookCaller.has_spec^syy$$r c|j9td|jjd|jjt ||j||_|j drg|_yy)NzHook z( is already registered within namespace historic)r3 ValueErrorr# namespacergetr0)rr7r8s rr5zHookCaller.set_specificationbsu 99  )*$$(II$7$7#8: 0$))YG == $!#D  %r c|jduS)z2Whether this caller is :ref:`historic `.N)r0rs r is_historiczHookCaller.is_historicps!!--r c|jDcgc]}|j|k7s|}}t|t|jk(rtd|d||jddycc}w)z?Remove all hook implementations registered by the given plugin.zplugin z not foundN)r2pluginlenr=)rrCimpl remainings r_remove_pluginzHookCaller._remove_plugintsa&*ooOd9NTO O y>S1 1wvj ;< <&Ps A*A*c6|jjS)z6Get all registered hook implementations for this hook.)r2copyrs r get_hookimplszHookCaller.get_hookimpls{s##%%r cxt|jD]!\}}|js |js|}nt |j}|js |jr|t |j}}nd|}}|j r|jj ||y|jr|jj ||y|dz }||k\r=|j|jr$|dz}||k\r|j|jr$|jj |dz|y)z,Add an implementation to the callback chain.rr N) enumerater2 hookwrapperwrapperrDtrylastinserttryfirst)rhookimplimethod splitpointstartends r _add_hookimplzHookCaller._add_hookimpls "4??3 .IAv!!V^^  . T__-J   8#3#3#S%93EJ3E    OO " "5( 3    OO " "3 1aAu*!3!r#rs r__repr__zHookCaller.__repr__sdii]!,,r c~|j|jjs|Si|jj|Sr")r3 kwargdefaults)rkwargss r_apply_defaultszHookCaller._apply_defaultss7 99 DII$;$;M4$))))4V44r c|jrk|jjD]Q}|vsdjfd|jjD}tjd|ddyyy)Nz, c3<K|]}|vr t|ywr")repr).0argnamer_s r z;HookCaller._verify_all_args_are_provided..s'*##&0 W *sz Argument(s) zE which are declared in the hookspec cannot be found in this hook call) stacklevel)r3argnamesjoinwarningswarn)rr_re notincalls ` r_verify_all_args_are_providedz(HookCaller._verify_all_args_are_provideds~ 9999-- &( $ *'+yy'9'9*!IMM&yk2<< $% #  r c H|jrJd|j|}|j||jr&|jjj ddnd}|j |j|jj||S)zCall the hook. Only accepts keyword arguments, which should match the hook specification. Returns the result(s) of calling all registered plugins, see :ref:`calling`. ACannot directly call a historic hook - use call_historic instead. firstresultF) rAr`rnr3optsr?r1r#r2rI)rr_ call_kwargsrqs r__call__zHookCaller.__call__s##% O %**62  **;7BF))diinn((>QV ~~ IIt++-{K  r c\|jJ|xsi}|j|}|j||jj||f|j |j |j j|d}|yt|tr|D] }|| yy)a<Call the hook with given ``kwargs`` for all registered plugins and for all plugins which will be registered afterwards, see :ref:`historic`. :param result_callback: If provided, will be called for each non-``None`` result obtained from a hook implementation. NF) r0r`rnappendr1r#r2rI isinstancelist)rresult_callbackr_resxs r call_historiczHookCaller.call_historics!!---2%%f- **62 !!6?";<nnTYY(<(<(>N  "  c4  #" # !r c|jrJd|j|}|j|ddddddd}|jj }|D]}t dd||}t |dz }|dk\rg||js||js||jr:|dz}|dk\r0||jr||jr*||jr:|j|dz||jr&|jjjddnd}|j|j|||S) zCall the hook with some additional temporarily participating methods using the specified ``kwargs`` as call parameters, see :ref:`call_extra`.rpFN)rNrM optionalhookrOrQspecnamezr rrq)rAr`rnr2rIrrDrMrNrQrPr3rrr?r1r#) rmethodsr_rr hookimplsrTrRrSrqs r call_extrazHookCaller.call_extrasU ##% O %%%f- **62 !  OO((*  .Fh=HI"Aq&1))Yq\-A-AQ<((Q q&1))Yq\-A-AQ<((   QUH - .CG))diinn((>QV ~~diiFKHHr c|jre|jJ|jD]G\}}|j|j|g|d}|s(|+t |t sJ||dIyy)zAApply call history to a new hookimpl if it is marked as historic.NFr)rAr0r1r#rwrx)rrTr_ryrzs r_maybe_apply_historyzHookCaller._maybe_apply_historys}    %%1 11+/+=+= ,'nnTYY&%H?6%c4000#CF+  , r )NN) r#r'r6rr7z_Namespace | Noner8zHookspecOpts | Noner%r&)r%bool)r7rr8rr%r&)rCrr%r&r%zlist[HookImpl])rRrr%r&r%r')r_Mapping[str, object]r%r)r_rr%r&)r_objectr%r)ryzCallable[[Any], None] | Noner_zMapping[str, object] | Noner%r&)rzSequence[Callable[..., object]]r_rr%r)rTrr%r&)r)r*r+r,r-rr:r5rArGrJrXr\r`rnrtr|rrrr rr(r(8sMI26)- CC C/ C ' C  C6% $' $  $  $.'&40-5 , ,9=.2#5#,#  #8!I6!I@T!I !IF ,r r(cZeZdZdZdZd dZed dZed dZed dZ d dZ y)_SubsetHookCallerztA proxy to another HookCaller which manages calls to all registered plugins except the ones from remove_plugins.)_orig_remove_pluginscd||_||_|j|_|j|_yr")rrr#r1)rorigremove_pluginss rrz_SubsetHookCaller.__init__+s' -II r c|jjDcgc]}|j|jvr|c}Scc}wr")rr2rCr)rrEs rr2z_SubsetHookCaller._hookimpls1s@ -- {{$"6"66    s!=c.|jjSr")rr3rs rr3z_SubsetHookCaller.spec9szzr c.|jjSr")rr0rs rr0z_SubsetHookCaller._call_history=szz'''r c"d|jdS)Nz<_SubsetHookCaller rZr[rs rr\z_SubsetHookCaller.__repr__As$TYYM33r N)rr(rzAbstractSet[_Plugin]r%r&r)r%zHookSpec | None)r%z_CallHistory | Noner) r)r*r+r,r-rpropertyr2r3r0r\rr rrrsT4I (   ((4r r)&r, __future__rcollections.abcrrrr AbstractSettypingrr r r r rk_configrr _decoratorsrr_implrrr'rrrxr__annotations__r _HookRelaytupler.r( _HookCallerrrr rrs#$#$. !!#! (8 gc6k2D9 T&\ 9  < < <  '#v+ # 5 < <= i Y,Y,z *4 *4r