o 5ήc:@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-vg8gfxp4/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%s=Nonez%s=%sz**, name signaturez%You are decorating a non-function: %s)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_globalsgetrupdate)r/r0kwframe callermodulerrrrAs zFunctionMaker.updateFc Ks |t|}|p i}t|}|durtd||d}t|gdd|jdD}|D]} | dvr>td| |fq0| d sH|d 7}d t |j f} z t || d } t | |Wntd tjd t|tjd ||} |rz||d<|j| fi|| S)zBMake a new function from a given template and update the signatureNz not a valid function template %scSsg|]}|dqS)z *)strip.0r5rrr sz&FunctionMaker.make..,)_func__call_z%s is overridden in %s zsinglezError in generated code:)file __source__)varsDEFmatch SyntaxErrorgroupsetrsplit NameErrorendswithnext_compile_countcompileexecprintr<stderrrA) r/ src_templevaldict addsourceattrssrcmornamesnfilenamecoder0rrrmakes8         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. (rENrMcss|]}d|VqdS)z Nr)rHlinerrr z'FunctionMaker.create..zdef %(name)s(%(signature)s): ) isinstancestrrFrWr) splitlinesrj)robjbodyrar rr rbrcrrestrr0r/ibodyrrrcreates   zFunctionMaker.create)NNNNNNN)NF)NNNT) rr __qualname__r itertoolscountr[rrArj classmethodrwrrrrrAs 4 !rcCsB|j}||d<||d<tj|d||d}t|dr|j|_|S)zE decorate(func, caller) decorates a function using a caller. rLrKz)return _call_(_func_, %(shortsignature)s) __wrapped__rx) __globals__r+rrwr-rx)r0callerrafunrrrdecorates  rcCs|dur t||St|r |j}t|}d|j|jf}n$t|r6|jdkr-d}n|j}|}|j}n|jj}|j j }|j j}|j }||d<t|d<t jd|d|||j|d S) z=decorator(caller) converts a caller function into a decoratorNzHdecorator(%s) converts functions/generators into factories of %s objectsrrrL _decorate_z%s(func)zreturn _decorate_(func, _call_))rr r})rr!isclassrlowerrr"r __class____call____func__r~r+rrwr)r_funcr callerfuncrrarrr decorators2        r)_GeneratorContextManager)GeneratorContextManagerc@seZdZddZdS)ContextManagercCstj|dt||d|dS)zContext manager decoratorz.with _self_: return _func_(%(shortsignature)s))_self_rKr|)rrwr,)r/r0rrrrs zContextManager.__call__N)rrrxrrrrrrs rcOst|||i|Srrrr/gr3krrrr 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()r3 vancestorsaddjvarrrr(s  r(csLsJdddtjdffdd 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|trtdtt||fdS)z5Make sure one passes the expected number of argumentszExpected %d arguments, got %d%sN)lenr.) argumentswrongmsg) dispatch_argsrrcheck2s zdispatch_on..checkc sttj}t|kstdifddfddfdd}fdd }fd d }tjd t|d ||d S)z4Decorator turning a function into a generic functionzUnknown dispatch arguments %scsl|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)rH_rrrrIGszIdispatch_on..gen_func_dec..vancestors..cSsg|]}t|qSr)rV)rHrarrrrILs)rangerzipr__mro__r()typesrastypes_ttype_r)rrtypemaprrrBs z5dispatch_on..gen_func_dec..vancestorscs|g}t||D]5\}}t|}|dkr!td||f|dkr6|\}td||fijdd}n|j}||ddq |S)zG Get a list of virtual MROs, one for each type rEzAmbiguous dispatch for %s: %srNrl)rr RuntimeErrortyperr()rlistsrvasn_vasrmro)rrrr ancestorsNs 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)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)rHr3rrrrnrszVdispatch_on..gen_func_dec..dispatch_info...)tuple)rHancrrrrIrszLdispatch_on..gen_func_dec..dispatch_info..)ryproduct)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)rrGrrrrnwrozGdispatch_on..gen_func_dec.._dispatch..)rKeyErrorryrrZr@)rr rBrr combinationsr)rr0rrr _dispatchvs   z4dispatch_on..gen_func_dec.._dispatchz#return _f_(%s, %%(shortsignature)s))_f_)rdefaultrrrrr})rVrr rXrrwr,)r0argsetrrrrr dispatch_str)rr0rrr gen_func_dec8s      z!dispatch_on..gen_func_dec dispatch_on)r)rner)rrrrrr*s  Vrr) rrer<r!rry collectionsr __version__r namedtupler rr\rRrrr contextlibr ImportErrorrrrinitrr n_argsr contextmanagerr(rrrrrsL   !