o pf=@spdZddlmZddlZddlZddlZddlZddlZddlZdZ ej dkr2ddlm Z ddZ n Gd d d e Z d dZ ed d ZddZedZGddde ZddZd$ddZzddlmZWneyuddlmZYnwGdddeZe ejZeejZedkrej sddZee_nedkrej rn edkrddZee_eeZ!d d!Z"d"d#Z#dS)%zT Decorator module, see http://pypi.python.org/pypi/decorator for the documentation. )print_functionNz4.0.10)getfullargspeccCs|jSN)__init__clsr T/home/ertert/spirit/venv/lib/python3.10/site-packages/prometheus_client/decorator.pyget_init1sr c@s&eZdZdZddZddZejZdS)rz?A quick and dirty replacement for getfullargspec for Python 2.XcCs*t|\|_|_|_|_g|_d|_dSr)inspect getargspecargsvarargsvarkwdefaults kwonlyargskwonlydefaults)selffr r r r7s zgetfullargspec.__init__ccs&|jV|jV|jV|jVdSr)rrrr)rr r r __iter__=s  zgetfullargspec.__iter__N)__name__ __module__ __qualname____doc__rrr rr r r r r4s  rcCs|jjSr)r__func__rr r r r FsArgSpeczargs varargs varkw defaultscCst|}t|j|j|j|jS)z$A replacement for inspect.getargspec)rrrrrr)rspecr r r rNsrz\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__)rrrrrrzarg%dr formatvaluecSsdS)Nr )valr r r wsz(FunctionMaker.__init__..*z%s=Nonez%s=%sz**, name signaturez%You are decorating a non function: %s)shortsignaturerr+rdocrmoduler isfunctionrgetattr annotationssetattr enumeratersys version_info formatargspecr,listrappendrrjoin__dict__copydictrhasattr TypeError)rfuncr+r,rr.r/funcdictargspecaiargallargs allshortargsr r r rbsn            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 selfr.Nr=rr rr2r?rr/)r+rr1rr; __defaults____kwdefaults__r"r5 _getframeAttributeError f_globalsgetrupdate)rr@kwframe callermoduler r r rOs 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 %sr'cSsg|]}|dqS)z *)strip.0rEr r r sz&FunctionMaker.make..,)_func__call_z%s is overridden in %s zsinglezError in generated code:)file __source__)varsDEFmatch SyntaxErrorgroupsetr-split NameErrorendswithnext_compile_countcompileexecprintr5stderrrO) r src_templevaldict addsourceattrssrcmor+namesnfilenamecoder@r r r makes8         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. (r'Nr(rZcss|]}d|VqdS)z Nr )rUliner r r z'FunctionMaker.create..zdef %(name)s(%(signature)s): ) isinstancestrrSrdr: splitlinesrw)r objbodyrnrr.r/rorpr+restr,r@ribodyr r r creates   zFunctionMaker.create)NNNNNNN)NF)NNNT) rrrr itertoolscountrhrrOrw classmethodrr r r r rXs 9 "rcCs4t||d}tj|d||d}t|dr|j|_|S)zE decorate(func, caller) decorates a function using a caller. )rYrXz)return _call_(_func_, %(shortsignature)s) __wrapped__r)r=rrr>r)r@callerrnfunr r r decorates  rcCs|dur t||St|r|j}d|j|jf}nt|r0|jdkr)d}n|j}|j}n |jj}|jj}t |td}t j d|d|||j |dS) z=decorator(caller) converts a caller function into a decoratorNzHdecorator(%s) converts functions/generators into factories of %s objectsr r!)rY _decorate_z%s(func)zreturn _decorate_(func, _call_))r.r/r) rr isclassrlowerr0r __class____call__r=rrr)r_funcr+r.rnr r r decorators(        r)_GeneratorContextManager)GeneratorContextManagerc@seZdZddZdS)ContextManagercCstj|dt||d|dS)zContext manager decoratorz.with _self_: return _func_(%(shortsignature)s))_self_rXr)rrr=)rr@r r r rs zContextManager.__call__N)rrrrr r r r rs rcOst|||i|SrrrrgrCkr r r rsrcOst||||Srrrr r r r'scCsTd}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)r4 issubclassr9)rC vancestorsaddjvar r r r92s  r9csLsJdddtjdffdd 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*r$cs0|t|trtdtt||fdS)z5Make sure one passes the expected number of argumentszExpected %d arguments, got %d%sN)lenr?) argumentswrongmsg) dispatch_argsr r checkLs 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|]}gqSr r )rU_r r r rVaszIdispatch_on..gen_func_dec..vancestors..cSsg|]}t|qSr )rc)rUrar r r rVfs)rangerzipr__mro__r9)typesrastypes_ttype_r)rrtypemapr r r\s 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 r'zAmbiguous dispatch for %s: %srNr()rr RuntimeErrortyperr9)rlistsrvasn_vasrmro)rrr r ancestorshs 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 decsz@dispatch_on..gen_func_dec..register..decr )rr)rr)rr register{sz3dispatch_on..gen_func_dec..registercs<|g}tj|D]}|tdd|Dq |S)zI An utility to introspect the dispatch algorithm css|]}|jVqdSr)r)rUrCr r r rzszKdispatch_on..gen_func_dec..dispatch_info..)rproductr9tuple)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)rrTr r r rzr{zGdispatch_on..gen_func_dec.._dispatch..)rKeyErrorrrrgrN)rrrPrr combinationsr)rr@rr r _dispatchs   z4dispatch_on..gen_func_dec.._dispatchz#return _f_(%s, %%(shortsignature)s))_f_)rdefaultrrrrr)rcrrrerrr=)r@argsetrrrrr dispatch_str)rr@rrr gen_func_decRs      z!dispatch_on..gen_func_dec dispatch_on)r:rner)rrr rr rDs  Xrr)$r __future__r collectionsr rrrer5 __version__r6rr object namedtuplerrrir_rrr contextlibr ImportErrorrrrinitrrn_argsrcontextmanagerr9rr r r r sT