H&h4ddlmZddlZddlmZmZmZmZddlm Z ddl m Z m Z ddl ZGddeZd Zd Zefd Zd Zd ZGddeZdZGddeZdZdZy))warnN)ordering ambiguitiessuper_signatureAmbiguityWarning) expand_tuples)Variadic isvariadicceZdZdZy)MDNotImplementedErrorz+A NotImplementedError for multiple dispatchN)__name__ __module__ __qualname____doc__j/mounts/lovelace/software/anaconda3/envs/py312/lib/python3.12/site-packages/multipledispatch/dispatcher.pyr r s5rr cLtt|j|ty)aBRaise warning when ambiguity is detected Parameters ---------- dispatcher : Dispatcher The dispatcher on which the ambiguity was detected ambiguities : set Set of type signature pairs that are ambiguous within this dispatcher See Also: Dispatcher.add warning_text N)r warning_textnamer) dispatcherrs rambiguity_warnr s joo{ 35EFrc$tdty)z5Deprecated interface to temporarily disable ordering.z=halt_ordering is deprecated, you can safely remove this call.NrDeprecationWarningrrr halt_orderingrsGrc$tdty)z4Deprecated interface to temporarily resume ordering.zrestart_ordering is deprecated, if you would like to eagerly orderthe dispatchers, you should call the ``reorder()`` method on each dispatcher.Nr) on_ambiguitys rrestart_orderingr &s   rc#Kt|}t|}|D])}t||}|t|rt|}+ t|}dy#t$rt|sJdYywxYww)ajCheck if a set of input types matches a variadic signature. Notes ----- The algorithm is as follows: Initialize the current signature to the first in the sequence For each type in `types`: If the current signature is variadic If the type matches the signature yield True Else Try to get the next signature If no signatures are left we can't possibly have a match so yield False Else yield True if the type matches the current signature Get the next signature FTN)iternext issubclassr StopIteration)typesfull_signaturesigitersigtypmatchess rvariadic_signature_matches_iterr,0s*>"G w-CS#& #w-C  w-CK  c? "J s.7A8 A8 AA8A52A84A55A8c4|sJtt||SN)allr,)r&r's rvariadic_signature_matchesr0Zs  .unE FFrceZdZdZdZddZdZedZedZ dZ e d Z e fd Zd Zd ZeZd ZdZdZdZdZe dZdZdZdZdZy) DispatcheraVDispatch methods based on type signature Use ``dispatch`` to add implementations Examples -------- >>> from multipledispatch import dispatch >>> @dispatch(int) ... def f(x): ... return x + 1 >>> @dispatch(float) ... def f(x): ... return x - 1 >>> f(3) 4 >>> f(3.0) 2.0 )rrfuncs _ordering_cachedocNcJ|x|_|_i|_||_i|_yr.)rrr3r6r5)selfrr6s r__init__zDispatcher.__init__ys%$(( DM  rcfd}|S)aregister dispatcher with new implementation >>> f = Dispatcher('f') >>> @f.register(int) ... def inc(x): ... return x + 1 >>> @f.register(float) ... def dec(x): ... return x - 1 >>> @f.register(list) ... @f.register(tuple) ... def reverse(x): ... return x[::-1] >>> f(1) 2 >>> f(1.0) 0.0 >>> f([1, 2, 3]) [3, 2, 1] c0j|fi|Sr.)add)funckwargsr8r&s r_dfz Dispatcher.register.._dfs DHHUD +F +Krr)r8r&r>r?s``` rregisterzDispatcher.registers6  rcttdr/tj|}|jj Sy)N signature)hasattrinspectrB parametersvaluesclsr=r)s rget_func_paramszDispatcher.get_func_paramss4 7K (##D)C>>((* * )rc|j|}|rDtjfd|D}td|D}t fd|Dr|Syy)z1get annotations of function positional parametersc3jK|]*}|jjjfvr|,ywr.)kindPOSITIONAL_ONLYPOSITIONAL_OR_KEYWORD).0param Parameters r z2Dispatcher.get_func_annotations..s:::--y/N/NOPs03c34K|]}|jywr.) annotation)rOrPs rrRz2Dispatcher.get_func_annotations..sEU 0 0Ec3:K|]}|juywr.)empty)rOannrQs rrRz2Dispatcher.get_func_annotations..sE#3ioo-EsN)rIrDrQtupler/)rHr=params annotationsrQs @rget_func_annotationszDispatcher.get_func_annotationssc$$T* ))I#F EfEEKEEE""F rc |s|j|}|r|}td|Dr#t|D]}|j||yg}t |dD]\}}t |t tfs6djd|D}td|d|d |jt |trN|t|k7r td t|dk7r td |jt|d |j|||jt|<|j j# |`y#t&$rYywxYw) aKAdd new types/method pair to dispatcher >>> D = Dispatcher('add') >>> D.add((int, int), lambda x, y: x + y) >>> D.add((float, float), lambda x, y: x + y) >>> D(1, 2) 3 >>> D(1, 2.0) Traceback (most recent call last): ... NotImplementedError: Could not find signature for add: When ``add`` detects a warning it calls the ``on_ambiguity`` callback with a dispatcher/itself, and a set of ambiguous type signature pairs as inputs. See ``ambiguity_warn`` for an example. c3<K|]}t|tywr.) isinstancerY)rOr*s rrRz!Dispatcher.add..s;#z#u%;sNr)start, c3jK|]+}t|tr |jn t|-ywr.)r_typerstr)rOcs rrRz!Dispatcher.add..s+$FG*Q"5AJJ3q6A$s13zTried to dispatch on non-type: z In signature: In function: z+Variadic signature must be the last elementzVariadic signature must contain exactly one element. To use a variadic union type place the desired types inside of a tuple, e.g., [(int, str)]r)r\anyr r< enumerater_rclistjoin TypeErrorrlenappendr r3rYr5clearr4AttributeError) r8rBr=r[typs new_signatureindexr*str_sigs rr<zDispatcher.addsh&33D9K'  ;; ;%i0 %t$ %  #IQ7 *JE3cD$<0))$KT$ *-gtyyB#t$C N*#$QRRs8q=#@ $$Xc!f%56$$S)1 *4,0 5'(     sE E! E!cZ |jS#t$r|jcYSwxYwr.)r4rnreorderr8s rrzDispatcher.orderings, ">> ! "<<> ! "s **c~t|jx|_}t|j}|r ||||Sr.)rr3r4r)r8rodambs rrtzDispatcher.reorders7&tzz22$**%  s # rc t|Dcgc] }t|c}} |j|} ||i|Scc}w#t$rH|j|}|s%t d|j dt|d||j|<Y^wxYw#t$rb|j|}t||D]} ||i|ccYS#t$rYwxYwt d|j dt|dwxYw)NCould not find signature for : <>zMatching functions for z(> found, but none completed successfully) rYrcr5KeyErrordispatchNotImplementedErrorr str_signaturer dispatch_iterr#)r8argsr>argr&r=r3s r__call__zDispatcher.__call__ s D1StCy12 &;;u%D (( (2 & 4==%(D)yy-"68"&DKK  &% &D&&.E K 000,  &II!%(   sE<ABABB(D>C D CDC*Dc d|jzS)Nz)rrus r__str__zDispatcher.__str__+s 499,,rc||jvr|j|S t|j|S#t$rYywxYw)aQDeterimine appropriate implementation for this type signature This method is internal. Users should call this object as a function. Implementation resolution occurs within the ``__call__`` method. >>> from multipledispatch import dispatch >>> @dispatch(int) ... def inc(x): ... return x + 1 >>> implementation = inc.dispatch(int) >>> implementation(3) 4 >>> print(inc.dispatch(float)) None See Also: ``multipledispatch.conflict`` - module to determine resolution order N)r3r#rr%r8r&s rr~zDispatcher.dispatch0sO, DJJ ::e$ $ ***E23 3  s7 AAc'2Kt|}|jD]y}t||k(r.ttt||r|j |}|?t|sKt |dsZt||sg|j |}|{yw)N)rkrr/mapr$r3r r0)r8r&nrBresults rrzDispatcher.dispatch_iterNs J !I9~"s3z5)+L'MI. YJy}$=-eY?!ZZ 2F L !sA"B%B4 BBc@tdt|j|S)zDeterimine appropriate implementation for this type signature .. deprecated:: 0.4.4 Use ``dispatch(*types)`` instead z-resolve() is deprecated, use dispatch(*types))rrr~rs rresolvezDispatcher.resolveYs! <>PQt}}e$$rc4|j|jdS)Nrr3rrus r __getstate__zDispatcher.__getstate__cs DJJ77rc~|d|_|d|_t|j|_t |_y)Nrr3)rr3rr4dictr5)r8ds r __setstate__zDispatcher.__setstate__fs1fI wZ !$**-f rcd|jzg}|jr|j|jg}|jdddD]}|j|}|j rQdt |z}|dt|zdzz }||j jz }|j|o|jt ||r#|jddj|zdj|S) NzMultiply dispatched method: %srz Inputs: <%s> - zOther signatures: z  ) rr6rlrr3rrrkstripri)r8docsotherr)r=ss rrzDispatcher.__doc__ls0499<= 88 KK !==2& 1C::c?D||$}S'99S3q6\D((T\\'')) A ]3/0 1  KK1HMM%4HH I{{4  rcP|jtt|jSr.)r~rrcr)r8rs r_helpzDispatcher._helpst}}c$o.666rc4t|j|y)z8Print docstring for the function corresponding to inputsN)printrr8rr>s rhelpzDispatcher.helps jdjj$ rcl|jtt|}|s tdt |S)NzNo function found)r~rrcrjsource)r8rr=s r_sourcezDispatcher._sources1t}}c$o./0 0d|rc4t|j|y)z:Print source code for the function corresponding to inputsN)rrrs rrzDispatcher.sources ldllD!"rr.)rrrr __slots__r9r@ classmethodrIr\r<propertyrrrtrr__repr__r~rrrrrrrrrrrr2r2`s,JIB++ ##$@D"" $2B-H< !%8 !!,7! #rr2cfdtj|z}|tj|z}|S)Nz File: %s )rD getsourcefile getsource)r=rs rrrs1..t44A G  d ##A Hrc0eZdZdZdZedZdZdZy)MethodDispatcherzODispatch methods based on type signature See Also: Dispatcher objrHcttdrDtj|}tj|j j ddSy)NrBr)rCrDrBitlislicerErFrGs rrIz MethodDispatcher.get_func_paramssA 7K (##D)C::cnn335q$? ? )rc"||_||_|Sr.r)r8instanceowners r__get__zMethodDispatcher.__get__s rct|Dcgc] }t|c}}|j|}|s%td|jdt |d||j g|i|Scc}w)Nrzr{r|)rYrcr~rrrr)r8rr>rr&r=s rrzMethodDispatcher.__call__snD1StCy12t}}e$%99mE24 DHH.t.v..2sA/N) rrrrrrrIrrrrrrrs, I@@  /rrc2djd|DS)zbString representation of type signature >>> str_signature((int, float)) 'int, float' rac34K|]}|jywr.)r)rOrHs rrRz str_signature..s1cS\\1rU)ri)r)s rrrs 991S1 11rc d|z}|dz }|D]#}|ddjd|Dzdzz }%|dz }|dj|Dcgc]}d tt|zd |zz!c}z }|Scc}w) zThe text for ambiguity warningsz. Ambiguities exist in dispatched function %s z;The following signatures may result in ambiguous behavior:  rac3>K|]}dt|zdzyw)[]N)r)rOrs rrRzwarning_text..s L!}Q'7!7#!= Lsrz, Consider making the following additions: rz @dispatch(z ) def %s(...))rirr)rrxtextpairrs rrrs >$ GD JJDT tyy Lt LLLtSST >>DFKK  =);< rsrNN *6/6G"#1'TG r#r#j  /z/>2 r