o Yi:@s>dZddlZddlZddlZddlZddlZddlZddlmZdZddZ e ddZ d d Z e d ZGd d d ZddZdddZzddlmZWney\ddlmZYnwGdddeZeejZeejZedkr~ejs~ddZee_nedkrejrn edkrddZee_eeZddZddZdS) zU Decorator module, see https://pypi.python.org/pypi/decorator for the documentation. N)getfullargspecz4.0.5cCs|jSN)__init__)clsr;/tmp/pip-target-1s0edx8b/lib/python/scipy/_lib/decorator.pyget_init.srArgSpeczargs varargs varkw defaultscCst|}t|j|j|j|jS)z$A replacement for inspect.getargspec)rr argsvarargsvarkwdefaults)fspecrrr getargspec7srz\s*def\s*([_\w][_\w\d]*)\s*\(c@sJeZdZdZeZ  d ddZddZddd Z e  dd d Z dS) 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||_|r|j|_|jdkrd|_|j|_|j|_t|rt |}t |di|_ dD] } t || t || q+t |jD] \} } t |d| | q_lambda___annotations__)r r r r kwonlyargskwonlydefaultszarg%d*z=None=z**, name signaturez#You are decorating a non-function: )shortsignature__name__r__doc__doc __module__moduleinspect isfunctionrgetattr annotationssetattr enumerater listr appendrr joinr__dict__copydictr hasattr TypeError)selffuncrrr rr!funcdictargspecaiargallargs allshortargsrrrrKs\          zFunctionMaker.__init__cKs|j|_t|dd|_t|di|_t|dd|_t|dd|_t|dd|_zt d}Wn t y9d }Ynw|j d d }t|d ||_ |j|dS) z2Update the signature of func with the data in selfrNr-r rrr%?rr!)rrr$rr+ __defaults____kwdefaults__rsys _getframeAttributeError f_globalsgetr update)r0r1kwframe callermodulerrrrBs zFunctionMaker.updateFc Ks|t|}|p i}t|}|durtd||d}t|gdd|jdD}|D]} | dvr@t| d|q1| d sJ|d 7}d t |j f} z t || d } t | |Wntd tjd t|tjd ||} |r|||d<|j| fi|| S)zBMake a new function from a given template and update the signatureNznot a valid function template cSsg|]}|dqS)z *)strip.0r6rrr sz&FunctionMaker.make..,)_func__call_z is overridden in  zsinglezError in generated code:)file __source__)varsDEFmatch SyntaxErrorgroupsetrsplit NameErrorendswithnext_compile_countcompileexecprintr=stderrrB) r0 src_templevaldict addsourceattrssrcmornamesnfilenamecoder1rrrmakes8        zFunctionMaker.makeTcKst|tr|dd\} } | dd} d} nd} d} |} || | | |||} ddd|D}| jd|||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. (rFNrNcss|]}d|VqdS)z Nr)rIlinerrr z'FunctionMaker.create..zdef %(name)s(%(signature)s): ) isinstancestrrGrXr* splitlinesrk)robjbodyrbr rr!rcrdrrestrr1r0ibodyrrrcreates   zFunctionMaker.create)NNNNNNN)NF)NNNT) rr __qualname__r itertoolscountr\rrBrk classmethodrxrrrrrAs 4 !rcCsB|j}||d<||d<tj|d||d}t|dr|j|_|S)zE decorate(func, caller) decorates a function using a caller. rMrLz)return _call_(_func_, %(shortsignature)s) __wrapped__ry) __globals__r,rrxr.ry)r1callerrbfunrrrdecorates  rcCs|dur t||St|r#|j}t|}d|jd|jd}n$t|r9|jdkr0d}n|j}|}|j}n|jj}|j j }|j j}|j }||d<t|d<t j|d d |||j|d S) z=decorator(caller) converts a caller function into a decoratorNz decorator(z2) converts functions/generators into factories of z objectsrrrM _decorate_z(func)zreturn _decorate_(func, _call_))rr!r~)rr"isclassrlowerrr#r __class____call____func__rr,rrxr )r_funcr callerfuncrrbrrr decorators2         r)_GeneratorContextManager)GeneratorContextManagerc@seZdZddZdS)ContextManagercCstj|dt||d|dS)zContext manager decoratorz.with _self_: return _func_(%(shortsignature)s))_self_rLr})rrxr-)r0r1rrrrs zContextManager.__call__N)rr ryrrrrrrs rcOst|||i|Srrrr0gr4krrrr srcOst||||SrrrrrrrscCsTd}t|D]\}}t||rd}n t||r|||<d}q|r(||dSdS)z_ Append ``a`` to the list of the virtual ancestors, unless it is already included. TFN)r' issubclassr))r4 vancestorsaddjvarrrr)s  r)csPsJddddtjdffdd fdd }d |_|S) zr Factory of decorators turning a function into a generic function dispatching on the given arguments. zNo dispatch args passedrlrz,)cs*|t|trtdtddS)z5Make sure one passes the expected number of argumentsz Expected z% arguments, got {len(arguments)}{msg}N)lenr/) argumentswrongmsg) dispatch_argsrrcheck2szdispatch_on..checkc sttj}t|kstdifddfddfdd}fdd }fd d }tjd d t|d||d S)z4Decorator turning a function into a generic functionzUnknown dispatch arguments csl|ddttD}D]}t|||D]\}}}t||r-||jvr-t||qqdd|DS)zU Get a list of sets of virtual ancestors for the given types cSsg|]}gqSrr)rI_rrrrJGszIdispatch_on..gen_func_dec..vancestors..cSsg|]}t|qSr)rW)rIrarrrrJLs)rangerzipr__mro__r))typesrastypes_ttype_r)rrtypemaprrrBs z5dispatch_on..gen_func_dec..vancestorscs|g}t||D]7\}}t|}|dkr#td|d||dkr8|\}td||fijdd}n|j}||ddq |S)zG Get a list of virtual MROs, one for each type rFzAmbiguous dispatch for z: rNrm)rr RuntimeErrortyperr))rlistsrvasn_vasrmro)rrrr ancestorsNsz4dispatch_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)rrrrrdecgsz@dispatch_on..gen_func_dec..register..decr)rr)rr)rrregisterasz3dispatch_on..gen_func_dec..registercs$|ddtj|D}|S)zI An utility to introspect the dispatch algorithm cSsg|] }tdd|DqS)css|]}|jVqdSr)r)rIr4rrrrorszVdispatch_on..gen_func_dec..dispatch_info...)tuple)rIancrrrrJrszLdispatch_on..gen_func_dec..dispatch_info..)rzproduct)rlst)rrrr dispatch_infoms  z8dispatch_on..gen_func_dec..dispatch_infocstdd|D}z|}Wn tyYnw||i|Stj|}t||D]}|}|durA||i|Sq-|i|S)Ncss|]}t|VqdSr)rrHrrrrowrpzGdispatch_on..gen_func_dec.._dispatch..)rKeyErrorrzrr[rA)rr rCrr combinationsr)rr1rrr _dispatchvs   z4dispatch_on..gen_func_dec.._dispatchz return _f_(z, %%(shortsignature)s))_f_)rdefaultrrrrr~)rWrr rYrrxr-)r1argsetrrrrr dispatch_str)rr1rrr gen_func_dec8s      z!dispatch_on..gen_func_dec dispatch_on)r*rner)rrrrrr*s  Vrr) rrer=r"rrz collectionsr __version__r namedtupler rr]rSrrr contextlibr ImportErrorrrrinitrr n_argsr contextmanagerr)rrrrrsL   !