a HapA@sdZddlZddlZddlZddlZddlZddlmZddlmZm Z m Z dZ e dZ ejjZejjZGdddeZd d Zdd dZddZdddZGdddeZeeZddZddZddZdS)zv Decorator module, see https://github.com/micheles/decorator/blob/master/docs/documentation.md for the documentation. N)_GeneratorContextManager)getfullargspeciscoroutinefunctionisgeneratorfunctionz5.1.1z\s*def\s*([_\w][_\w\d]*)\s*\(c@sZeZdZdZeZdZZZ Z Z Z dddZ ddZdd d Zedd d ZdS) FunctionMakerz An object with the ability to create functions with a given signature. It has attributes name, doc, module, signature, defaults, dict and methods update and make. NcCs||_|rZ|j|_|jdkr$d|_|j|_|j|_t|rZt |}t |di|_ dD]} t || t || qZt |jD]\} } t |d| | q|t|j} t|j} |jr| d|j| d|jn|jr| d|jD]$} | d| | d| | fq|jr6| d |j| d |jd | |_d | |_|j|_|rf||_|durv||_|r||_|r||_|r||_|r||_t|d sJt|d std |dS)NzZ_lambda___annotations__)argsvarargsvarkwdefaults kwonlyargskwonlydefaultszarg%d*z%s=Nonez%s=%sz**, name signaturez%You are decorating a non function: %s)shortsignature__name__r__doc__doc __module__moduleinspectZ isroutinergetattr annotationssetattr enumerater listr appendr r joinr__dict__copydictr hasattr TypeError)selffuncrrr rrZfuncdictZargspecaiargZallargsZ allshortargsrr(lib/python3.9/site-packages/decorator.py__init__@sZ           zFunctionMaker.__init__cKs|j|_t|dd|_t|di|_|j|_|jp4d|_t|dd|_ zt d}Wnt yjd}Yn0|j dd}t|d||_|j|dS) zD Update the signature of func with the data in self rNr#r?rr)rrrrr!r __defaults__r__kwdefaults__rsys _getframeAttributeError f_globalsgetrupdate)r&r'kwframeZ callermodulerrr+r6ts   zFunctionMaker.updateFc Ks|t|}|pi}t|}|dur2td||d}t|gdd|jdD}|D]} | dvr`td| |fq`| d s|d 7}d t |j } zt || d } t | |Wn0tytd tjd t|tjd Yn0||} |r||d<|j| fi|| S)zT Make a new function from a given template and update the signature Nz not a valid function template %scSsg|]}|dqS)z *)strip.0r*rrr+ z&FunctionMaker.make..,)_func__call_z%s is overridden in %s zZsinglezError in generated code:)fileZ __source__)varsDEFsearch SyntaxErrorgroupsetrsplit NameErrorendswithnext_compile_countcompileexec Exceptionprintr1stderrr6) r&Z src_templevaldict addsourceattrssrcmornamesnfilenamecoder'rrr+makes6         zFunctionMaker.makeTcKst|tr0|dd\} } | dd} d} n d} d} |} || | | |||} ddd|D}|d}|rt|rd |d d }nd |}| j |||fi|S) z Create a function from the strings name, signature and body. evaldict is the evaluation dictionary. If addsource is true an attribute __source__ is added to the result. The attributes attrs are added, if any. (r9NrBcss|]}d|VqdS)z Nr)r<linerrr+ r>z'FunctionMaker.create..rAz#async def %(name)s(%(signature)s): returnz return awaitzdef %(name)s(%(signature)s): ) isinstancestrr:rJr splitlinesr5rreplacer])clsobjbodyrTr rrrUrVrrestrr'r&Zibodycallerrrr+creates    zFunctionMaker.create)NNNNNNN)NF)NNNT)rr __qualname__r itertoolscountrNr r r r r rr,r6r] classmethodrlrrrr+r3s 4 $rcCs$|j|i|}||j|jfS)zA Fix args and kwargs to be consistent with the signature )ZbindZapply_defaultsr kwargs)r rqsigZbarrr+fixsrsrFcs*ttr(fdd}n2trFfdd}nfdd}j|_j|_|_|_j|_z j |_ Wnt yYn0z j |_ Wnt yYn0z j |_ Wnt yYn0z j |_ Wnt yYn0z|jjWnt y$Yn0|S)af Decorates a function/generator/coroutine using a caller. If kwsyntax is True calling the decorated functions with keyword syntax will pass the named arguments inside the ``kw`` dictionary, even if such argument are positional, similarly to what functools.wraps does. By default kwsyntax is False and the the arguments are untouched. cs4st||\}}g|Ri|IdHSNrsr r7rkextrasr'kwsyntaxrrrr+funszdecorate..func?s>st||\}}g|Ri|D] }|Vq.dSrtru)r r7resrwrr+rzscs.st||\}}g|Ri|Srtrurvrwrr+rzs)rrrrrr __wrapped__ __signature__rmr/r3r0rrr!r6)r'rkrxryrzrrwr+decorates@         r~csfdd}|S)z A version of "decorator" implemented via "exec" and not via the Signature object. Use this if you are want to preserve the `.__code__` object properties (https://github.com/micheles/decorator/issues/129). cstj|dt|d||jdS)Nz)return _call_(_func_, %(shortsignature)s))rAr@)r|rm)rrlr#rmr'rkrr+dec s  zdecoratorx..decr)rkrrrr+ decoratorxs rcs|durt|dSt}dd|jDd fdd }|jd|_j|_j|_|_ j |_ t dd|_ |j j |S) zG decorator(caller) converts a caller function into a decorator NrcSsg|]}|jtur|qSr)kindPOSr<prrr+r=r>zdecorator..csXt|d}|tfdd|dD|durFfddSt|SdS)Nr9c3s(|] }|jtur|j|jVqdSrt)defaultEMPTYr5rr)r7rr+ra"s z)decorator..dec..cst|Srtr~r)rkrxryrr+&r>z(decorator..dec..)lentupler~)r'r r7ZnarkZ dec_paramsry)rxr7r+r s  zdecorator..dec) parametersr0)N)r~rrrvaluesrfr}rrr|rmrr0r!r6)rkZ_funcryrrrrrr+ decorators  rc@seZdZddZddZdS)ContextManagercOst||||dSrt)rr,)r&gr(krrr+r,7szContextManager.__init__csfdd}t||S)NcsPjjgjRij||i|WdS1sB0YdSrt) __class__r'r kwds)fr(rr&rr+rk;s z'ContextManager.__call__..callerr)r&r'rkrrr+__call__:s zContextManager.__call__N)rrrmr,rrrrr+r6srcCst|Srt)_contextmanagerrrrr+contextmanagerDsrcCsPd}t|D]0\}}t||r&d}q>t||r |||<d}q |rL||dS)z_ Append ``a`` to the list of the virtual ancestors, unless it is already included. TFN)r issubclassr)r( vancestorsaddjvarrr+rKs  rcsLs Jdddtjdffdd fdd}d |_|S) zr Factory of decorators turning a function into a generic function dispatching on the given arguments. zNo dispatch args passedz(%s,)rcs0|t|tr,tdtt||fdS)z5Make sure one passes the expected number of argumentszExpected %d arguments, got %d%sN)rr%)Z argumentsZwrongmsg) dispatch_argsrr+checkeszdispatch_on..checkc sttj}t|ks&tdifddfddfdd}fdd }fd d }tjd t|d ||d S)z4Decorator turning a function into a generic functionzUnknown dispatch arguments %scsn|ddttD}D]<}t|||D]*\}}}t||r2||vr2t||q2q"dd|DS)zU Get a list of sets of virtual ancestors for the given types cSsg|]}gqSrr)r<_rrr+r=zr>zIdispatch_on..gen_func_dec..vancestors..cSsg|] }t|qSr)rI)r<rarrr+r=r>)rangerziprmror)typesZrastypes_ttype_r)rrtypemaprr+rusz5dispatch_on..gen_func_dec..vancestorscs|g}t||D]p\}}t|}|dkrDtd||fn4|dkrp|\}td||fidd}n|}||ddq|S)zG Get a list of virtual MROs, one for each type r9zAmbiguous dispatch for %s: %srNr_)rr RuntimeErrortyperr)rZlistsrZvasZn_vasrr)rrrr+ ancestorss z4dispatch_on..gen_func_dec..ancestorscsfdd}|S)zU Decorator to register an implementation for the given types cs&t|jtjd|j|<|S)Nz in )rr operatorltr)r)rrrrr+rsz@dispatch_on..gen_func_dec..register..decr)rr)rr)rr+registersz3dispatch_on..gen_func_dec..registercs<|g}tj|D]}|tdd|Dq|S)zI An utility to introspect the dispatch algorithm css|] }|jVqdSrt)r)r<r(rrr+rar>zKdispatch_on..gen_func_dec..dispatch_info..)rnproductrr)rZlstZanc)rrrr+ dispatch_infos z8dispatch_on..gen_func_dec..dispatch_infocstdd|D}z |}Wnty0Yn0||i|Stj|}t||D](}|}|durZ||i|SqZ|i|S)Ncss|]}t|VqdSrt)rr;rrr+rar>zGdispatch_on..gen_func_dec.._dispatch..)rKeyErrorrnrrMr5)rr r7rr combinationsr)rr'rrr+ _dispatchs   z4dispatch_on..gen_func_dec.._dispatchz#return _f_(%s, %%(shortsignature)s))Z_f_)rrrrrrr|)rIrr rKrrlr#)r'ZargsetrrrrrZ dispatch_str)rr'rrr+ gen_func_decks      z!dispatch_on..gen_func_dec dispatch_on)r rner)rrrrr+r]s  W r)rF)NF)rrer1rrrn contextlibrrrr __version__rOrEZ ParameterZPOSITIONAL_OR_KEYWORDremptyrobjectrrsr~rrrrrrrrrrr+s,   6 !