o Y^FC@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 +/usr/lib/python3/dist-packages/decorator.pyget_init0sr FullArgSpeczAargs varargs varkw defaults kwonlyargs kwonlydefaults annotationscCstt|gdifS)z?A quick and dirty replacement for getfullargspec for Python 2.XN)r _makeinspectZ getargspecfr r r r7srcCs|jjSr)r__func__rr r r r ;scCdSNFr rr r r iscoroutinefunctionBr)isgeneratorfunctioncCrrr )callerr r r rHrrz\s*def\s*([_\w][_\w\d]*)\s*\(c@sbeZdZdZeZdZZZ Z Z Z  dddZ ddZdd d Ze  dd 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. r NcCs||_|r|j|_|jdkrd|_|j|_|j|_t|rt |}t |di|_ dD] } t || t || q+t |jD] \} } t |d| | q_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__copydictrhasattr TypeError)selffuncr$r%rr)r+ZfuncdictZargspecaiargZallargsZ 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)Nr6r.r?r'r+)r$r'r-r(r4r __defaults__r!__kwdefaults__rsys _getframeAttributeError f_globalsgetr*update)r9r:kwframeZ callermoduler r r rFs  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 zZsinglezError in generated code:)fileZ __source__)varsDEFsearch SyntaxErrorgroupsetr&split NameErrorendswithnext_compile_countcompileexec ExceptionprintrAstderrrF) r9Z 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. (rINrQcss|]}d|VqdS)z Nr )rLliner r r z'FunctionMaker.create..rPz#async def %(name)s(%(signature)s): returnz return awaitzdef %(name)s(%(signature)s): ) isinstancestrrJrYr3 splitlinesrErreplacerl)robjbodyrcrr)r+rdrer$restr%r:r9Zibodyrr r r creates    zFunctionMaker.create)NNNNNNNr)NNNT)r'r* __qualname__r( itertoolscountr]rrrrr r!rrFrl classmethodrzr 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. )rPrOz_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{) r6r0rAversionrrrrzr7r{) r:rextrasrcesr<ZextraexZgeneratorcallerZfunr 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 p.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 )rL_r r r rM}szIdispatch_on..gen_func_dec..vancestors..cSsg|]}t|qSr )rX)rLrar r r rMs)rangerziprmror2)typesZrastypes_tZtype_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 rIzAmbiguous dispatch for %s: %srNrn)rr RuntimeErrortyperr2)rZlistsrZvasZn_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')rLr;r r r rpszKdispatch_on..gen_func_dec..dispatch_info..)r|productr2tuple)rZlstZanc)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)rrKr r r rprqzGdispatch_on..gen_func_dec.._dispatch..)rKeyErrorr|rr\rE)rrrGrr combinationsr)rr:rr r _dispatchs   z4dispatch_on..gen_func_dec.._dispatchz#return _f_(%s, %%(shortsignature)s))Z_f_)rdefaultrrrrr)rXrrrZrrzr6)r:ZargsetrrrrrZ dispatch_str)rr:rrr gen_func_decns      z!dispatch_on..gen_func_dec dispatch_on)r3rner')rrr rr r`s  Wr)r r)'r(Z __future__rrerArrr| collections __version__ version_inforr namedtupler rrCr ImportErrorr^rTobjectrrr contextlibrrrrZinitrrZn_argsrrrr2rr r r r sh            "'