U [+dFC@sdZddlmZddlZddlZddlZddlZddlZddlZdZ ej dkrdddlm Z ddZ ne d d Zd d Z d dZ z ejZWnek rddZYnXzddlmZWnek rddZYnXedZGdddeZd*ddZd+ddZzddlmZWn"ek r6ddlmZYnXGdddeZe ejZeej Z!e!dkr~ej"s~d d!Zee_n,e!dkrej"rne!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 -/tmp/pip-unpacked-wheel-9pgd70u8/decorator.pyget_init0sr FullArgSpeczAargs varargs varkw defaults kwonlyargs kwonlydefaults annotationscCstt|gdifS)z?A quick and dirty replacement for getfullargspec for Python 2.XN)r _makeinspect getargspecfr r r r7srcCs|jjSr)r__func__rr r r r ;scCsdSNFr rr r r iscoroutinefunctionBsr)isgeneratorfunctioncCsdSrr )callerr r r rHsrz\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. r 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||_|dk rv||_|r||_|r||_|r||_|r||_t|d stt|d std |dS)N_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 enumeraterlistrappendrrjoinr$__dict__copydictrhasattrAssertionError TypeError)selffuncr#r$rr(r*ZfuncdictZargspecaiargallargsZ allshortargsr r r r]sZ           zFunctionMaker.__init__cKs|j|_t|dd|_t|di|_|j|_|jp4d|_t|dd|_ zt d}Wnt k rld}YnX|j dd}t|d||_|j|dS) z2Update the signature of func with the data in selfr(Nr5r-r?r&r*)r#r&r,r'r3r __defaults__r __kwdefaults__rsys _getframeAttributeError f_globalsgetr)update)r9r:kwframeZ callermoduler r r rGs  zFunctionMaker.updateFc Ks|t|}|pi}t|}|dkr2td||d}t|gdd|jdD}|D]} | dkr`td| |fq`| d s|d 7}d t |j } zt || d } t | |Wn2tk rtd tjd t|tjd YnX||} |r||d<|j| f|| 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 ExceptionprintrBstderrrG) r9Z src_templevaldict addsourceattrssrcmor#namesnfilenamecoder:r r r makes6        zFunctionMaker.makeTcKst|tr0|dd\} } | dd} d} n d} d} |} || | | |||} ddd|D}|d}|rt|rd |d d }nd |}| j |||f|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. (rJNrRcss|]}d|VqdS)z Nr )rMliner r r sz'FunctionMaker.create..rQz#async def %(name)s(%(signature)s): returnz return awaitzdef %(name)s(%(signature)s): ) isinstancestrrKrZr2 splitlinesrFrreplacerm)robjbodyrdrr(r*rerfr#restr$r:r9Zibodyrr r r creates"    zFunctionMaker.create)NNNNNNN)NF)NNNT)r&r) __qualname__r' itertoolscountr^rrrrrr rrGrm classmethodrzr r r r rPs" 4 "rr c Cst||d}d}t|D]$\}}d|}|||<||d7}qdtjkrTdkrlnnt|oht| }nt|}|rtj|d|||d} ntj|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. )rQrPz_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{) r5r/rBversionrrrrzr6r{) r:rextrasrdesr<extraexZgeneratorcallerZfunr r r decorates>  rc Cs|dk rt||Sd\}}t|r@|j}d|j|jf}n~t|r|jdkrZd}n|j}|j}|jj}t |j pxd}d |jj |||}|r|d7}|j }n|j j}|jj}t|td }tjd ||fd ||f|||j|d } |rd || _ | S)z=decorator(caller) converts a caller function into a decoratorN)rr zHdecorator(%s) converts functions/generators into factories of %s objectsrrr r"rO)Z_callZ _decorate_z %s(func, %s)zhif func is None: return lambda func: _decorate_(func, _call, (%s)) return _decorate_(func, _call, (%s)))r(r*rr)rrisclassr&lowerr+r'__code__ co_argcountlenr@r2 co_varnames __class____call__r5rrzr)) rZ_funcZ defaultargsrr#r(nargsZndefsrddecr r r decoratorsF          r)_GeneratorContextManager)GeneratorContextManagerc@seZdZddZdS)ContextManagercCstj|dt||d|dS)zContext manager decoratorz.with _self_: return _func_(%(shortsignature)s))Z_self_rPr)rrzr5)r9r:r r r r0s  zContextManager.__call__N)r&r)r{rr r r r r/srcOst||||Srrrr9gr;kr r r r:srcOst||||Srrrr r r r@scCst|Sr)_contextmanager)r:r r r contextmanagerGsrcCsPd}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/ issubclassr1)r; vancestorsaddjvar r r r1Ns  r1csLs tdddtjdffdd 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|tr,tdtt||fdS)z5Make sure one passes the expected number of argumentszExpected %d arguments, got %d%sN)rr8) argumentsZwrongmsg) dispatch_argsr r checkhszdispatch_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||kr2t||q2q"dd|DS)zU Get a list of sets of virtual ancestors for the given types cSsg|]}gqSr r )rM_r r r rN}szIdispatch_on..gen_func_dec..vancestors..cSsg|] }t|qSr )rY)rMrar r r rNs)rangerziprmror1)typesZrastypes_ttype_r)rrtypemapr r rxsz5dispatch_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 rJzAmbiguous dispatch for %s: %srNro)rr RuntimeErrortyperr1)rlistsrZvasZn_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&)rMr;r r r rqszKdispatch_on..gen_func_dec..dispatch_info..)r|productr1tuple)rlstZanc)rrr r dispatch_infos z8dispatch_on..gen_func_dec..dispatch_infocstdd|D}z |}Wntk r2Yn X|||Stj|}t||D]$}|}|dk rX|||SqX||S)Ncss|]}t|VqdSr)rrLr r r rqszGdispatch_on..gen_func_dec.._dispatch..)rKeyErrorr|rr]rF)rrrHrr combinationsr)rr:rr r _dispatchs   z4dispatch_on..gen_func_dec.._dispatchz#return _f_(%s, %%(shortsignature)s))Z_f_)rdefaultrrrrr)rYrrr[rrzr5)r:ZargsetrrrrrZ dispatch_str)rr:rrr gen_func_decns*     z!dispatch_on..gen_func_dec dispatch_on)r7r2rner&)rrr rr r`s  W r)r )N)'r' __future__rrerBrrr| collections __version__ version_inforr namedtupler rrDr ImportErrorr_rUobjectrrr contextlibrrrrinitrrZn_argsrrrr1rr r r r sb       " '