cC& dZdZddlZddlZddlmZmZddlmZddlm Z m Z m Z m Z m Z ddlmZddlmZGd d eZGd d eZGd deZGddeZe de Ze d*ddde ede de egeffdZe d*dede ede defdZd*dZGddeZdZdZdZGdd eZGd!d"eZGd#d$eZd%Z d&Z!d+d'e"d(e e#de fd)Z$y),z* A few useful function/method decorators. zrestructuredtext enN) process_timetime)isgeneratorfunction)AnyOptionalCallableoverloadTypeVar)getfullargspec) method_typecFeZdZddeedeeddfdZd deedefdZy) cached_decoratorN cacheattrkeyargreturnc ||_||_yN)rr)selfrrs ;/usr/lib/python3/dist-packages/logilab/common/decorators.py__init__zcached_decorator.__init__&s" callableobjct|r Jd|z|Jtt|jdk(s|jdk(r&t ||j }|jS|jr1t||j|j }|jSt||j }|jS)Nz#cannot cache generator function: %sr) rlenr argsr_SingleValueCacher_MultiValuesKeyArgCache_MultiValuesCacheclosure)rrcaches r__call__zcached_decorator.__call__*s&{3 1K ? 3&&& ~k*// 0A 59I%k4>>BE }} [[+KdnnUE}}&k4>>BE}}rNNr) __name__ __module__ __qualname__rstrintrrr"rrrr%s=(3- Y] HX$6 ( rrc@eZdZd dedeeddfdZdZdefdZdZ y) rNrrrcp||_|d|jz|_y||jk7sJ||_y)Nz _%s_cache_)callabler$r)rrrs rrz_SingleValueCache.__init__9s<#  )K,@,@@DN 4 44 44&DNrc |j|jS#t$r/|j|g|}t ||j||cYSwxYwr)__dict__rKeyErrorr,setattr)_SingleValueCache__merrvalues rr"z_SingleValueCache.__call__AsS ==0 0 !DMM$..E D$..% 0L s5AAcfd}|_ jj|_jj|_|S#t$rY|SwxYw)Nc(j|i|Sr)r")rkwargsrs rwrappedz*_SingleValueCache.closure..wrappedJs 4==$1&1 1r) cache_objr,__doc__r$ Exceptionrr6s` rr z_SingleValueCache.closureIsZ 2 ! "mm33GO#}}55G    s6A AAcP|jj|jdyr)r.popr)rholders rclearz_SingleValueCache.clearWsDNND1rr) r$r%r&rrr'rr"r r>r)rrrr8s7'H'#'RV'  2rrceZdZdZdZy)rc |j|j}|S#t$ri}t||j|Y|SwxYwr)r.rr/r0)rr=_caches r _get_cachez_MultiValuesCache._get_cache\sK 4__T^^4F  4F FDNNF 3  4s"AAc|j|} ||S#t$r|j|g|||<||cYSwxYwr)rBr/r,)_MultiValuesCache__merrr5rAs rr"z_MultiValuesCache.__call__dsQ& $<  (4==55F4L$<  s $??N)r$r%r&rBr"r)rrrr[s  rrc >eZdZddededeeddffd ZdZxZ S) rNrrrrc<tt| ||||_yr)superrrr)rrrr __class__s rrz _MultiValuesKeyArgCache.__init__ns %t5k9M rc|j|}||jdz } ||S#t$r!|j|g|i|||<||cYSwxYw)Nr)rBrr/r,)_MultiValuesKeyArgCache__merrr5rAkeys rr"z _MultiValuesKeyArgCache.__call__rsh&4;;?# #;  '$-->t>v>F3K#;  s*'AAr) r$r%r&rr(rr'rr" __classcell__)rHs@rrrms.Hchsm_crr_T)boundrrr5rc yrr)rrr5s rcachedrQsrc yrr)rPs rrQrQsrc :||d<tdi|}||S||S)z0Simple decorator to cache result of method call.rr))r)rrr5 decorators rrQrQs0F8 *6*I%%rceZdZdZdZdZejje jddk7r e de fdZd d Z y) cachedpropertyaProvides a cached property equivalent to the stacking of @cached and @property, but more efficient. After first usage, the becomes part of the object's __dict__. Doing: del obj. empties the cache. Idea taken from the pyramid_ framework and the mercurial_ project. .. _pyramid: http://pypi.python.org/pypi/pyramid .. _mercurial: http://pypi.python.org/pypi/Mercurial )r6cb |j||_y#t$rtd|zwxYw)Nz!%s must have a __name__ attribute)r$AttributeError TypeErrorr6r:s rrzcachedproperty.__init__s> K     K?'IJ J Ks .rz sphinx-buildrcLt|jdd}d|rd|zzSdzS)Nr8z+%sz %s)getattrr6)rdocs rr8zcachedproperty.__doc__s/$,, 48C@TWFSL` `]_` `rNcr||S|j|}t||jj||Sr)r6r0r$)rinstobjtypevals r__get__zcachedproperty.__get__s5 <Kll4 dll++S1 rr)r$r%r&r8 __slots__rospathbasenamesysargvpropertyr'rbr)rrrVrVsW I ww $6  aS a  arrVc|j}t||}t|tr |j}|j Sr)rHr\ isinstancerifgetr7)objfuncnameclsmembers rget_cache_implrqs6 --C S( #F&(#   rc:t||j|y)zClear a cache handled by the :func:`cached` decorator. If 'x' class has @cached on its method `foo`, type >>> clear_cache(x, 'foo') to purge this method's cache on the instance. N)rqr>)rmrns r clear_cacherss3!'',rct||j} t|||j|y#t$rYywxYw)z/Copy cache for from cacheobj to obj.N)rqrr0r.r/)rmrncacheobjrs r copy_cachervsBsH-77I Y 1 1) <=    s 3 ??c"eZdZdZdZdZdZy) wpropertyzSimple descriptor expecting to take a modifier function as first argument and looking for a _ to retrieve the attribute. c:||_d|jz|_y)Nz_%s)setfuncr$attrname)rrzs rrzwproperty.__init__s  0 00 rc(|j||yr)rz)rrmr2s r__set__zwproperty.__set__s S% rc6|Jt||jSr)r\r{)rrmros rrbzwproperty.__get__ssDMM**rN)r$r%r&r8rr}rbr)rrrxrxs1!+rrxceZdZdZdZdZy) classpropertyz>this is a simple property-like class but for class attributes.c||_yrget)rrs rrzclassproperty.__init__s rc$|j|Srr)rr_ros rrbzclassproperty.__get__sxx}rN)r$r%r&r8rrbr)rrrrsHrrc"eZdZdZdZdZdZy) iclassmethodzDescriptor for method which should be available as class method if called on the class or instance method if called on an instance. c||_yr)func)rrs rrziclassmethod.__init__s  rcv|!t|j||jSt|j||Sr)r rrH)rinstancer`s rrbziclassmethod.__get__s4  tyy'73D3DE E499h88rctd)Nzcan't set attribute)rX)rrr2s rr}ziclassmethod.__set__s233rN)r$r%r&r8rrbr}r)rrrrs9 4rrcfd}|S)Nct}t}|i|}tdjt|z t|z fz|S)Nz%s clock: %.9f / time: %.9f)rrprintr$)rr5tcresfs rwrapztimed..wrapsL F N   +qzz<>A;MtvXYz.ZZ[ rr))rrs` rtimedrs Krcfd}|S)zDecorator taking two methods to acquire/release a lock as argument, returning a decorator function which will call the inner method after having called acquire(self) et will call release(self) afterwards. cJdtdtdtdtffd }|S)Nrrr5rcX| |g|i||S#|wxYwrr))rrr5acquirerreleases rwrapperz*locked..decorator..wrappers0 DM ///  s  ))r)rrrrs` rrTzlocked..decorators+ # c S S rr))rrrTs`` rlockedrs  rklass methodnamecfd}|S)a}Decorator extending class with the decorated callable. This is basically a syntactic sugar vs class assignment. >>> class A: ... pass >>> @monkeypatch(A) ... def meth(self): ... return 12 ... >>> a = A() >>> a.meth() 12 >>> @monkeypatch(A, 'foo') ... def meth(self): ... return 12 ... >>> a.foo() 12 cz xs |j}t|||S#t$rtd|zwxYw)NzI%s has no __name__ attribute: you should provide an explicit `methodname`)r$rXr0)rnamerrs rrTzmonkeypatch..decorator9sU .D tT"   >@DE  s":r))rrrTs`` r monkeypatchr$s*  rr#r)%r8 __docformat__rdrgrrinspectrtypingrrrr r r logilab.common.compatr objectrrrrrMr(rQrVrqrsrvrxrrrrtyper'rr)rrrsR$1& #'=="- v& 2 2F ) $ /  T" 6:&.smFI rdBh   8C=3SU &&V&R- ++"F464"& t # ( r