o eFC@sdZddlmZddlZddlZddlZddlZddlZddlZdZ ej dkr2ddlm Z ddZ ne d d Zd d Z d dZ zejZWn eyRddZYnwzddlmZWn eyhddZYnwedZGdddeZd*ddZd+ddZzddlmZWneyddlmZYnwGdddeZe ejZeej Z!e!dkrej"sd d!Zee_ne!dkrej"rn e!d"krd#d!Zee_eeZ#d$d%Z$d&d'Z%d(d)Z&dS),zT Decorator module, see http://pypi.python.org/pypi/decorator for the documentation. )print_functionNz4.4.2))getfullargspeccCs|jSN)__init__clsr _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__moduler isfunctionrgetattr annotationssetattr enumeraterlistrappendr!rjoinr&__dict__copydictr hasattr TypeError)selffuncr%r&r r*r,funcdictargspecaiargallargs allshortargsr r r r]s\           zFunctionMaker.__init__cKs|j|_t|dd|_t|di|_|j|_|jpd|_t|dd|_ zt d}Wn t y5d}Ynw|j dd}t|d||_|j|dS) z2Update the signature of func with the data in selfr*Nr7r/r?r(r,)r%r(r.r)r5r __defaults__r"__kwdefaults__rsys _getframeAttributeError f_globalsgetr+update)r:r;kwframe callermoduler r r rKs  zFunctionMaker.updateFc Ks|t|}|p i}t|}|durtd||d}t|gdd|jdD}|D]} | dvr>td| |fq0| d sH|d 7}d t |j } z t || d } t | |Wntyrtd tjd t|tjd w||} |r}||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.0r@r r r sz&FunctionMaker.make..,)_func__call_z%s is overridden in %s zsinglezError in generated code:)file __source__)varsDEFsearch SyntaxErrorgroupsetr'split NameErrorendswithnext_compile_countcompileexec ExceptionprintrFstderrrK) r: src_templevaldict addsourceattrssrcmor%namesnfilenamecoder;r r r makes:          zFunctionMaker.makeTcKst|tr|dd\} } | dd} d} nd} d} |} || | | |||} ddd|D}|d}|rGt|rGd |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. (rONrWcss|]}d|VqdS)z Nr )rRliner r r z'FunctionMaker.create..rVz#async def %(name)s(%(signature)s): returnz return awaitzdef %(name)s(%(signature)s): ) isinstancestrrPrar4 splitlinesrJrreplaceru)robjbodyrlr r*r,rmrnr%restr&r;r:ibodyrr r r creates    zFunctionMaker.create)NNNNNNNr)NNNT)r(r+ __qualname__r) itertoolscountrerrrr r!r"rrKru classmethodrr r r r rPs 4 "rr c Cst||d}d}t|D]\}}d|}|||<||d7}q dtjkr*dkr6nn t|o4t| }nt|}|rHtj|d|||d} n tj|d |||d} t|d r\|j | _ | S) z decorate(func, caller) decorates a function using a caller. If the caller is a generator function, the resulting function will be a generator function. )rVrUz_e%d_r$z3.5z3.6z?for res in _call_(_func_, %s%%(shortsignature)s): yield res __wrapped__z,return _call_(_func_, %s%%(shortsignature)s)r) r7r1rFversionrrrrr8r) r;rextrasrlesr?extraexgeneratorcallerfunr r r decorates6  rc Cs|dur t||Sd\}}t|r |j}d|j|jf}n?t|rU|jdkr-d}n|j}|j}|jj}t |j pkr r r r:srcOst||||Srrrr r r r@scCst|Sr)_contextmanager)r;r r r contextmanagerGsrcCsTd}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)r1 issubclassr3)r> vancestorsaddjvar r r r3Ns  r3csLsJdddtjdffdd fdd}d |_|S) zr Factory of decorators turning a function into a generic function dispatching on the given arguments. zNo dispatch args passedz(%s,)r$rcs0|t|trtdtt||fdS)z5Make sure one passes the expected number of argumentszExpected %d arguments, got %d%sN)rr9) argumentswrongmsg) dispatch_argsr r checkhs zdispatch_on..checkc sttj}t|kstdifddfddfdd}fdd }fd d }tjd t|d ||d S)z4Decorator turning a function into a generic functionzUnknown dispatch arguments %scsn|ddttD}D]}t|||D]\}}}t||r.||vr.t||qqdd|DS)zU Get a list of sets of virtual ancestors for the given types cSsg|]}gqSr r )rR_r r r rS}szIdispatch_on..gen_func_dec..vancestors..cSsg|]}t|qSr )r`)rRrar r r rSs)rangerziprmror3)typesrastypes_ttype_r)rrtypemapr r rxs z5dispatch_on..gen_func_dec..vancestorscs|g}t||D]7\}}t|}|dkr!td||f|dkr7|\}td||fidd}n|}||ddq |S)zG Get a list of virtual MROs, one for each type rOzAmbiguous dispatch for %s: %srNrw)rr RuntimeErrortyperr3)rlistsrvasn_vasrr)rrr r 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 )rroperatorltr(r)rrrr r 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|]}|jVqdSr)r()rRr>r r r ryszKdispatch_on..gen_func_dec..dispatch_info..)rproductr3tuple)rlstanc)rrr r dispatch_infos z8dispatch_on..gen_func_dec..dispatch_infocstdd|D}z|}Wn tyYnw||i|Stj|}t||D]}|}|durA||i|Sq-|i|S)Ncss|]}t|VqdSr)rrQr r r ryrzzGdispatch_on..gen_func_dec.._dispatch..)rKeyErrorrrrdrJ)rrrLrr combinationsr)rr;rr r _dispatchs   z4dispatch_on..gen_func_dec.._dispatchz#return _f_(%s, %%(shortsignature)s))_f_)rdefaultrrrrr)r`rrrbrrr7)r;argsetrrrrr dispatch_str)rr;rrr gen_func_decns      z!dispatch_on..gen_func_dec dispatch_on)r4rner()rrr rr r`s  Wr)r r)'r) __future__rrerFrrr collections __version__ version_inforr namedtupler rrHr ImportErrorrfr\objectrrr contextlibrrrrinitrrn_argsrrrr3rr r r r sh            "'