c4ddlmZddlZddlmZmZmZmZddlm Z ddl m Z m Z ddl ZGddeZd Zd Zefd Zd Zd ZGddeZdZGddeZdZdZdS))warnN)ordering ambiguitiessuper_signatureAmbiguityWarning) expand_tuples)Variadic isvariadicceZdZdZdS)MDNotImplementedErrorz- A NotImplementedError for multiple dispatch N)__name__ __module__ __qualname____doc__;lib/python3.11/site-packages/multipledispatch/dispatcher.pyr r s7777rr cVtt|j|tdS)aC Raise 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' jo{ 3 35EFFFFFrc0tdtdS)z:Deprecated interface to temporarily disable ordering. z=halt_ordering is deprecated, you can safely remove this call.NrDeprecationWarningrrr halt_orderingrs& Grc0tdtdS)z9Deprecated 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]4}t||}|Vt|st|}5 t|}dVdS#t$rt|sJdVYdSwxYw)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,2s*>""G w--CS#&& # w--C w--CKKKKK    c?? " " "JJJJJJ sA..BBcD|sJtt||SN)allr,)r&r's rvariadic_signature_matchesr0\s'  .unEE F FFrceZdZdZdZddZdZedZedZ dZ e d Z e fd Zd Zd ZeZd ZdZdZdZdZe dZdZdZdZdZdS) DispatcheraW Dispatch 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_cachedocNcL|x|_|_i|_||_i|_dSr.)rrr3r6r5)selfrr6s r__init__zDispatcher.__init__zs)$(( DM  rcfd}|S)a register 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] c&j|fi|Sr.)add)funckwargsr8r&s r_zDispatcher.register.._s$ DHUD + +F + + +Krr)r8r&r>r?s``` rregisterzDispatcher.registers/4       rcttdr-tj|}|jSdS)N signature)hasattrinspectrB parametersvaluesclsr=r)s rget_func_paramszDispatcher.get_func_paramssB 7K ( ( +#D))C>((** * + +rc||}|rPtjfd|D}td|D}t fd|Dr|SdSdS)z; get annotations of function positional parameters c3FK|]}|jjjfv|VdSr.)kindPOSITIONAL_ONLYPOSITIONAL_OR_KEYWORD).0param Parameters r z2Dispatcher.get_func_annotations..sS99 0 6889e999999rc3$K|] }|jV dSr.) annotation)rOrPs rrRz2Dispatcher.get_func_annotations..s9 % %  % % % % % %rc3*K|] }|juVdSr.)empty)rOannrQs rrRz2Dispatcher.get_func_annotations..s*EE#3io-EEEEEErN)rIrDrQtupler/)rHr=params annotationsrQs @rget_func_annotationszDispatcher.get_func_annotationss$$T**  #)I9999999F   % %# % % %%%KEEEEEEEEE #"" # # # #rc V|s||}|r|}td|Dr*t|D]}|||dSg}t |dD]\}}t |t tfs>> 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|tVdSr.) isinstancerX)rOr*s rrRz!Dispatcher.add..s,;;#z#u%%;;;;;;rNr)start, c3lK|]/}t|tr|jnt|V0dSr.)r^typerstr)rOcs rrRz!Dispatcher.add..s\$D$D453=Q2E2E%0AJJ),Q$D$D$D$D$D$DrzTried 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^rblistjoin TypeErrorrlenappendr r3rXr5clearr4AttributeError) r8rBr=rZtyps new_signatureindexr*str_sigs rr<zDispatcher.adds& (33D99K ('  ;;;;; ; ; %i00 % %t$$$$ F #IQ777 * *JE3cD$<00 ;))$D$D9B$D$D$DDDi"%gggtyy!:;;; #t$$ *C NN*#Es88q=#@ $$Xc!f%56666$$S))))+/ 5''(      DD sF F('F(cZ |jS#t$r|cYSwxYwr.)r4rmreorderr8s rrzDispatcher.orderings> "> ! " " "<<>> ! ! ! "s  **ct|jx|_}t|j}|r ||||Sr.)rr3r4r)r8rodambs rrszDispatcher.reordersF&tz222$*%%  $ Ls # # # rc td|D} |j|}nN#t$rA|j|}|s(t d|jdt |d||j|<YnwxYw ||i|S#t$rc|j|}t||D]} ||i|ccYS#t$rYwxYwt d|jdt |dwxYw)Nc,g|]}t|SrrbrOargs r z'Dispatcher.__call__.. 111StCyy111rCould not find signature for : <>zMatching functions for z(> found, but none completed successfully) rXr5KeyErrordispatchNotImplementedErrorr str_signaturer dispatch_iterr#)r8argsr>r&r=r3s r__call__zDispatcher.__call__ s11D11122 &;u%DD & & & 4=%(D 7))YYY e 4 4 4 46777"&DK     & 4((( ($   &D&.E KKK  400000000,D&%III}U3333  s@ )AA43A48B'C-(B4/C-4 C>C-C,C-cd|jzS)Nz)rrts r__str__zDispatcher.__str__(s 49,,rc||jvr |j|S t|j|S#t$rYdSwxYw)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 rrzDispatcher.dispatch,s`, DJ  %:e$ $ **E233 3   44 s / ==c'VKt|}|jD]}t||kr5ttt||r|j|}|VJt|r6t |dr!t||r|j|}|VdS)N)rjrr/mapr$r3r r0)r8r&nrBresults rrzDispatcher.dispatch_iterJs JJ ! !I9~~" !s3z5)+L+L'M'M !I. Y !Jy}$=$= !-eY??!!Z 2F LLL ! !rc@tdt|j|S)z Deterimine appropriate implementation for this type signature .. deprecated:: 0.4.4 Use ``dispatch(*types)`` instead z-resolve() is deprecated, use dispatch(*types))rrrrs rresolvezDispatcher.resolveVs- <  ! ! !t}e$$rc |j|jdS)Nrr3rrts r __getstate__zDispatcher.__getstate__as %% %rc|d|_|d|_t|j|_t |_dS)Nrr3)rr3rr4dictr5)r8ds r __setstate__zDispatcher.__setstate__es8fI wZ !$*--ff rc,d|jzg}|jr||jg}|jdddD]}|j|}|jr\dt |z}|dt|zdzz }||jz }||r|t ||r+|dd |zd |S) NzMultiply dispatched method: %srz Inputs: <%s> - zOther signatures: z  ) rr6rkrr3rrrjstriprh)r8docsotherr)r=ss rrzDispatcher.__doc__ks049<= 8 " KK ! ! !=2& 1 1C:c?D| 1$}S'9'99S3q66\D((T\''))) A ]3//0000  J KK1HMM%4H4HH I I I{{4   rcF|jtt|jSr.)rrrbr)r8rs r_helpzDispatcher._helpst}c$oo.66rc4t|j|dS)z: Print docstring for the function corresponding to inputs N)printrr8rr>s rhelpzDispatcher.helps jdj$     rc||jtt|}|stdt |S)NzNo function found)rrrbrisource)r8rr=s r_sourcezDispatcher._sources;t}c$oo. 1/00 0d||rc4t|j|dS)z< Print source code for the function corresponding to inputs N)rrrs rrzDispatcher.sources ldlD!"""""rr.)rrrr __slots__r9r@ classmethodrIr[r<propertyrrrsrr__repr__rrrrrrrrrrrrr2r2bs*JI>++[+ ##[#&@@@D""X" $2<---H< ! ! ! % % %%%% !!X!,777!!! #####rr2cbdtj|z}|tj|z}|S)Nz File: %s )rD getsourcefile getsource)r=rs rrrs1.t444A G d # ##A Hrc8eZdZdZdZedZdZdZdS)MethodDispatcherzP Dispatch methods based on type signature See Also: Dispatcher objrHcttdrAtj|}tj|jddSdS)NrBr)rCrDrBitlislicerErFrGs rrIz MethodDispatcher.get_func_paramssS 7K ( ( @#D))C:cn3355q$?? ? @ @rc"||_||_|Sr.r)r8instanceowners r__get__zMethodDispatcher.__get__s rctd|D}|j|}|s(td|jdt |d||jg|Ri|S)Nc,g|]}t|Srrzr{s rr}z-MethodDispatcher.__call__..r~rrrr)rXrrrrr)r8rr>r&r=s rrzMethodDispatcher.__call__s11D11122t}e$ I%%'+yyy-2F2F2F2F'HII ItDH.t...v...rN) rrrrrrrIrrrrrrrs_ I@@[@  /////rrc@dd|DS)zc String representation of type signature >>> str_signature((int, float)) 'int, float' r`c3$K|] }|jV dSr.)r)rOrHs rrRz str_signature..s$11cS\111111rrh)r)s rrrs% 9911S111 1 11rcdz}|dz }|D]*}|ddd|Dzdzz }+|dz }|dfd |Dz }|S) z! The text for ambiguity warnings z. Ambiguities exist in dispatched function %s z;The following signatures may result in ambiguous behavior:  r`c3@K|]}dt|zdzVdS)[]N)r)rOrs rrRzwarning_text..s4AAqcM!,,,s2AAAAAArrz, Consider making the following additions: rcZg|]'}dtt|zdzz(S)z @dispatch(z ) def %s(...))rr)rOrrs rr}z warning_text..sQAAA78& oa6H6H(I(II+d23AAArr)rrwtextpairs` rrrs >$ GD JJDII  IIAADAAA A ABDHI I >>DFKKAAAArsNNNNNNNNNNNN ********88888/888GGG"#1'''TGGG o#o#o#o#o#o#o#o#d    /////z///8222     r