S)@sfdZdZdddgZddlZddlZddlZejdkrmddlmZd d Zn"Gd d d e Zd d Zej dZ Gddde Z dddZ ddZy2ddlmZedefedeZWnRek rUddlmZddZedefededeZYnXe eZdS)zT Decorator module, see http://pypi.python.org/pypi/decorator for the documentation. z3.4.0 decorator FunctionMakercontextmanagerN3)getfullargspeccCs|jS)N)__init__)clsr +/usr/lib/python3/dist-packages/decorator.pyget_init*sr c@s.eZdZdZddZddZdS)rz?A quick and dirty replacement for getfullargspec for Python 2.XcCs=tj|\|_|_|_|_g|_d|_dS)N)inspectZ getargspecargsvarargsvarkwdefaults kwonlyargskwonlydefaults)selffr r r r/s' zgetfullargspec.__init__ccs$|jV|jV|jV|jVdS)N)r rrr)rr r r __iter__4szgetfullargspec.__iter__N)__name__ __module__ __qualname____doc__rrr r r r r-s  rcCs |jjS)N)r__func__)rr r r r 9sz\s*def\s*([_\w][_\w\d]*)\s*\(c @sseZdZdZdddddddddZddZdddd Zedddd d d ZdS) rz 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||_|r4|j|_|jdkr6d|_n|j|_|j|_tj|r4t |}t |di|_ x'dD]} t || t || qWx1t |jD] \} } t |d | | qWtjd krtjd d d|dd|_|_nt|j} t|j} |jro| jd|j| jd|jn|jr| jdnx9|jD].} | jd| | jd| | fqW|jr| jd|j| jd|jndj| |_dj| |_|jj|_q4n|rF||_n|dk r^||_n|rp||_n|r||_n|r||_n|r||_nt|dstt|dstd|ndS)Nz_lambda___annotations__r rrrrrzarg%drZ formatvaluecSsdS)Nr )valr r r Zsz(FunctionMaker.__init__..*z%s=Nonez%s=%sz**z, name signaturez%You are decorating a non function: %s)zargszvarargszvarkwzdefaultsz kwonlyargszkwonlydefaults) shortsignaturerr"rdocrmoduler isfunctionrgetattr annotationssetattr enumerater sysversionZ formatargspecr#listrappendrrjoin__dict__copydictrhasattrAssertionError TypeError)rfuncr"r#rr&r'ZfuncdictZargspecaiargZallargsZ allshortargsr r r rEs`       2          zFunctionMaker.__init__cKs|j|_t|dd|_t|di|_t|df|_t|dd|_t|dd|_tj dj j dd }t|d ||_ |jj |dS) z2Update the signature of func with the data in selfr&Nr4rrr*r?r')r"rr)rr2 __defaults____kwdefaults__rr- _getframe f_globalsgetrupdate)rr8kwZ callermoduler r r rC~s zFunctionMaker.updateFc Ks_|t|}|pi}tj|}|dkrJtd|n|jd}t|gdd|jjdD}x3|D]+} | dkrtd | |fqqW|j d s|d 7}ny#t |d d } t | |Wn1t d dt jt |dt jYnX||} |rK||ds z&FunctionMaker.make..,_func__call_z%s is overridden in %s zZsinglezError in generated code:fileZ __source__)z_func_rJ)varsDEFmatch SyntaxErrorgroupsetr%split NameErrorendswithcompileexecprintr-stderrrC) rZ src_templevaldict addsourceattrssrcZmor"namesncoder8r r r makes2       zFunctionMaker.makeTcKst|trF|jjdd\} } | dd} d} nd} d} |} || | | |||} djdd|jD}| jd||||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. (r NrKcss|]}d|VqdS)z Nr )rFliner r r sz'FunctionMaker.create..zdef %(name)s(%(signature)s): r$) isinstancestrrErSr1 splitlinesra)robjZbodyrZrr&r'r[r\r"restr#r8rZibodyr r r creates  " zFunctionMaker.create) rrrrrrCra classmethodrjr r r r r?s  8 c Cs|dk rN|jj}||d<||drrrz%s(%s)zreturn decorator(_call_, %s)r&r') __globals__r3rrjr Zisclassrlowerr rr r(r __class____call__rrr)Zcallerr8rZr"Z callerfuncr&Zfunr r r rs>           cCs(tj|dtd|d|d|S)zContext manager decoratorz.with _self_: return _func_(%(shortsignature)s)Z_self_rIrl)rrjr4)rr8r r r rpsrp)_GeneratorContextManagerContextManager)GeneratorContextManagercOstj||||S)N)rsr)rrr9kr r r rsr)r __version____all__r-rer r.rr objectrVrNrrrp contextlibrqtyper4rr ImportErrorrsrrr r r r !s.$ ~)