a |_s@sdZgdZddlmZddlmZddlmZddlm Z ddl m Z dZ d Z e e fd d Ze e fd d ZefddZefddZefddZefddZefddZefddZefddZefddZefddZefd d!Zefd"d#Zefd$d%Zd&efd'efd(efgd(efd)efd&efgd)efd(efd'efgd'efd&efd)efgd*Zd+d,Zd-d.Zzdd/l mZWne!yYn0e"Z#e#fd0d1Z$zdd2l m$Z$Wne!yYn0Gd3d4d4Z%zdd5l m%Z%Wne!yYn0Gd6d7d7e"Z&d8d9Z'ed:gd;Z(Gdd?Z0dXdBdCZ1dDdEZ2zddFl m2Z2Wne!y|Yn0dGdHZ3dIdJZ4dYdLdMZ5dNdOZ6dPdQZ7dRdSZ8GdTdUdUZ9e"Z:GdVdWdWZ;dKS)ZzEfunctools.py - Tools for working with functions and callable objects )update_wrapperwrapsWRAPPER_ASSIGNMENTSWRAPPER_UPDATEStotal_orderingcache cmp_to_key lru_cachereducepartial partialmethodsingledispatchsingledispatchmethodcached_property)get_cache_token) namedtuple)recursive_repr)RLock) GenericAlias) __module____name__ __qualname____doc____annotations__)__dict__c Csd|D]2}zt||}Wnty(Yq0t|||q|D]}t||t||iq<||_|S)aUpdate a wrapper function to look like the wrapped function wrapper is the function to be updated wrapped is the original function assigned is a tuple naming the attributes assigned directly from the wrapped function to the wrapper function (defaults to functools.WRAPPER_ASSIGNMENTS) updated is a tuple naming the attributes of the wrapper that are updated with the corresponding attribute from the wrapped function (defaults to functools.WRAPPER_UPDATES) )getattrAttributeErrorsetattrupdate __wrapped__)wrapperwrappedassignedupdatedattrvaluer&%/usr/local/lib/python3.9/functools.pyr#s rcCstt|||dS)aDecorator factory to apply update_wrapper() to a wrapper function Returns a decorator that invokes update_wrapper() with the decorated function as the wrapper argument and the arguments to wraps() as the remaining arguments. Default arguments are as for update_wrapper(). This is a convenience function to simplify applying partial() to update_wrapper(). r!r"r#)r rr(r&r&r'rAs rcCs$||}||ur|S| o"||kS)zIReturn a > b. Computed by @total_ordering from (not a < b) and (a != b).__lt__selfotherNotImplementedZ op_resultr&r&r' _gt_from_ltYs r/cCs"||}||ur|S|p ||kS)zEReturn a <= b. Computed by @total_ordering from (a < b) or (a == b).r)r+r&r&r' _le_from_lt`s r0cCs||}||ur|S| S)z=Return a >= b. Computed by @total_ordering from (not a < b).r)r+r&r&r' _ge_from_ltgs r1cCs$||}||ur|S| p"||kS)zJReturn a >= b. Computed by @total_ordering from (not a <= b) or (a == b).__le__r+r&r&r' _ge_from_lens r4cCs"||}||ur|S|o ||kS)zFReturn a < b. Computed by @total_ordering from (a <= b) and (a != b).r2r+r&r&r' _lt_from_leus r5cCs||}||ur|S| S)z=Return a > b. Computed by @total_ordering from (not a <= b).r2r+r&r&r' _gt_from_le|s r6cCs$||}||ur|S| o"||kS)zIReturn a < b. Computed by @total_ordering from (not a > b) and (a != b).__gt__r+r&r&r' _lt_from_gts r9cCs"||}||ur|S|p ||kS)zEReturn a >= b. Computed by @total_ordering from (a > b) or (a == b).r7r+r&r&r' _ge_from_gts r:cCs||}||ur|S| S)z=Return a <= b. Computed by @total_ordering from (not a > b).r7r+r&r&r' _le_from_gts r;cCs$||}||ur|S| p"||kS)zJReturn a <= b. Computed by @total_ordering from (not a >= b) or (a == b).__ge__r+r&r&r' _le_from_ges r>cCs"||}||ur|S|o ||kS)zFReturn a > b. Computed by @total_ordering from (a >= b) and (a != b).r<r+r&r&r' _gt_from_ges r?cCs||}||ur|S| S)z=Return a < b. Computed by @total_ordering from (not a >= b).r<r+r&r&r' _lt_from_ges r@r8r3r=r*)r*r3r8r=csVfddtD}|stdt|}t|D]"\}}||vr.||_t||q.S)z6Class decorator that fills in missing ordering methodscs(h|] }t|dtt|dur|qSN)robject).0opclsr&r' z!total_ordering..z6must define at least one ordering operation: < > <= >=)_convert ValueErrormaxrr)rFZrootsrootZopnameZopfuncr&rEr'rsrcsGfdddt}|S)z,Convert a cmp= function into a key= functioncsZeZdZdgZddZfddZfddZfdd Zfd d Zfd d Z dZ dS)zcmp_to_key..KobjcSs ||_dSrArM)r,rMr&r&r'__init__szcmp_to_key..K.__init__cs|j|jdkSNrrNr,r-mycmpr&r'r*szcmp_to_key..K.__lt__cs|j|jdkSrPrNrQrRr&r'r8szcmp_to_key..K.__gt__cs|j|jdkSrPrNrQrRr&r'__eq__szcmp_to_key..K.__eq__cs|j|jdkSrPrNrQrRr&r'r3szcmp_to_key..K.__le__cs|j|jdkSrPrNrQrRr&r'r=szcmp_to_key..K.__ge__N) rrr __slots__rOr*r8rTr3r=__hash__r&rRr&r'Ks     rW)rB)rSrWr&rRr'rsr)rcCsXt|}|tur value Apply a function of two arguments cumulatively to the items of a sequence, from left to right, so as to reduce the sequence to a single value. For example, reduce(lambda x, y: x+y, [1, 2, 3, 4, 5]) calculates ((((1+2)+3)+4)+5). If initial is present, it is placed before the items of the sequence in the calculation, and serves as a default when the sequence is empty. z0reduce() of empty sequence with no initial valueN)iter_initial_missingnext StopIteration TypeError)ZfunctionZsequenceZinitialitr%Zelementr&r&r'r s    r )r csJeZdZdZdZfddZddZeddZd d Z d d Z Z S) r zSNew function with partial application of the given arguments and keywords. )funcargskeywordsr __weakref__cs^t|stdt|dr8|j|}i|j|}|j}tt||}||_||_||_|S)Nz#the first argument must be callabler^) callabler\hasattrr_r`r^superr __new__)rFr^r_r`r, __class__r&r'res  zpartial.__new__cOs*i|j|}|jg|j|Ri|SrAr`r^r_)r,r_r`r&r&r'__call__*szpartial.__call__cCst|j}t|jg}|dd|jD|dd|jDt|jdkrld|dd |dS|dd |dS) Ncss|]}t|VqdSrA)repr)rCxr&r&r' 2rHz#partial.__repr__..css |]\}}|d|VqdS)=Nr&rCkvr&r&r'rl3rH functoolsz functools.(, )) typerrjr^extendr_r`itemsrjoin)r,Zqualnamer_r&r&r'__repr__.s  zpartial.__repr__cCs*t||jf|j|j|jpd|jp$dffSrA)rur^r_r`rr,r&r&r' __reduce__8szpartial.__reduce__cCst|tstdt|dkr0tdt||\}}}}t|rrt|trr|dur`t|trr|durzt|tsztdt|}|duri}nt|turt|}|duri}||_||_||_ ||_ dS)Nz(argument to __setstate__ must be a tuplezexpected 4 items in state, got zinvalid partial state) isinstancetupler\lenrbdictrurr^r_r`)r,stater^r_kwds namespacer&r&r' __setstate__<s4    zpartial.__setstate__) rrrrrUrerirryr{r __classcell__r&r&rfr'r s  r )r c@sFeZdZdZddZddZddZd d d Zed d Z e e Z dS)r zMethod descriptor with partial application of the given arguments and keywords. Supports wrapping existing descriptors and handles non-descriptor callables as instance methods. cOsft|s t|ds td|t|trP|j|_|j||_i|j||_n||_||_||_dS)N__get__z${!r} is not callable or a descriptor) rbrcr\formatr}r r^r_r`)r,r^r_r`r&r&r'rObs  zpartialmethod.__init__cCsNdtt|j}ddd|jD}d}|j|jj|jj |j ||dS)Nrscss|]\}}d||VqdS)z{}={!r}N)rrnr&r&r'rlwsz)partialmethod.__repr__..z*{module}.{cls}({func}, {args}, {keywords}))modulerFr^r_r`) rxmaprjr_r`rwrrgrrr^)r,r_r`Z format_stringr&r&r'ryus  zpartialmethod.__repr__csfdd}j|_|_|S)Ncs,ij|}j|gj|Ri|SrArh)Z cls_or_selfr_r`rzr&r'_methodsz3partialmethod._make_unbound_method.._method)__isabstractmethod__Z_partialmethod)r,rr&rzr'_make_unbound_methods z"partialmethod._make_unbound_methodNcCst|jdd}d}|durh|||}||jurht|g|jRi|j}z |j|_WntyfYn0|dur|||}|S)Nr) rr^r r_r`__self__rrr)r,rMrFgetresultZnew_funcr&r&r'rs    zpartialmethod.__get__cCst|jddSNrFrr^rzr&r&r'rsz"partialmethod.__isabstractmethod__)N) rrrrrOryrrpropertyr classmethodr__class_getitem__r&r&r&r'r Zs   r cCst|tr|j}q|SrA)r}r r^r^r&r&r'_unwrap_partials rZ CacheInfo)hitsmissesmaxsizeZcurrsizec@s(eZdZdZdZefddZddZdS) _HashedSeqz This class guarantees that hash() will be called no more than once per element. This is important because the lru_cache() will hash the key multiple times on a cache miss. hashvaluecCs||dd<|||_dSrAr)r,Ztuphashr&r&r'rOs z_HashedSeq.__init__cCs|jSrArrzr&r&r'rVsz_HashedSeq.__hash__N)rrrrrUrrOrVr&r&r&r'rs rc s|}|r&||7}|D] } || 7}q|rh||fdd|D7}|r||fdd|D7}n$||dkr|d|vr|dSt|S)aMake a cache key from optionally typed positional and keyword arguments The key is constructed in a way that is flat as possible rather than as a nested structure that would take more memory. If there is only a single argument and its data type is known to cache its hash value, then that argument is returned without a wrapper. This saves space and improves lookup speed. c3s|]}|VqdSrAr&rCrprur&r'rlrHz_make_key..c3s|]}|VqdSrAr&rrr&r'rlrHr)rwvaluesr) r_rtypedZkwd_markZ fasttypesr~rurkeyitemr&rr' _make_keys   rFcs~ttrdkrldnTtr\ttr\d}t|t}fdd|_t||Sdurltdfdd}|S) aLeast-recently-used cache decorator. If *maxsize* is set to None, the LRU features are disabled and the cache can grow without bound. If *typed* is True, arguments of different types will be cached separately. For example, f(3.0) and f(3) will be treated as distinct calls with distinct results. Arguments to the cached function must be hashable. View the cache statistics named tuple (hits, misses, maxsize, currsize) with f.cache_info(). Clear the cache and statistics with f.cache_clear(). Access the underlying function with f.__wrapped__. See: http://en.wikipedia.org/wiki/Cache_replacement_policies#Least_recently_used_(LRU) rrcs dSNrrr&r&rr&r'rHzlru_cache..Nz=Expected first argument to be an integer, a callable, or Nonecs(t|t}fdd|_t||S)Ncs dSrr&r&rr&r'rrHz8lru_cache..decorating_function..)_lru_cache_wrapper _CacheInfocache_parametersr) user_functionr rr&r'decorating_functionsz&lru_cache..decorating_function) r}intrbboolrrrrr\)rrrr rr&rr'rs   rcstt d\id djjt gddgdd< dkrh fdd}nN dur fdd}n* fdd} fdd } fd d }||_||_|S) N)rrrFcsd7|i|}|SNrr&)r_rr)rrr&r'r sz#_lru_cache_wrapper..wrappercsL||}|}|ur*d7|Sd7|i|}||<|Srr&)r_rrr)r cache_getrmake_keyrsentinelrrr&r'r &s  c sp ||} |}|dur|\}}}}||<||< }||< <||< |<d7|WdS d7 Wdn1s0Y|i|} |vrnr } || <|| <|  } } d < <| =| |<n6 }| ||g}||< <|< kWdn1sb0Y|Srr&) r_rrlinkZ link_prevZ link_nextZ_keyrZlastZoldrootZoldkeyZ oldresult)KEYNEXTPREVRESULTrr cache_lenfullrlockrrrrLrrr&r'r 5sB  &  *cs8 WdS1s*0YdS)zReport cache statisticsNr&r&)rrrrrrr&r' cache_infonsz&_lru_cache_wrapper..cache_infocsP8ddgdd<ddWdn1sB0YdS)z$Clear the cache and cache statisticsNrF)clearr&)rrrrrrLr&r' cache_clearss z'_lru_cache_wrapper..cache_clear)rBrr__len__rrr)rrrrr rrr&)rrrrrrrrrrrrrrrLrrrr'r s* *9 r)rcCstdd|S)z@Simple lightweight unbounded cache. Sometimes called "memoize".N)r)r)rr&r&r'rsrcCsg}dd|D}|s|S|D]2}|d}|D]}||ddvr.d}qq.qRq|durbtd|||D]}|d|krp|d=qpqdS)zMerges MROs in *sequences* to a single MRO using the C3 algorithm. Adapted from http://www.python.org/download/releases/2.3/mro/. cSsg|] }|r|qSr&r&rCsr&r&r' rHz_c3_merge..rrNzInconsistent hierarchy) RuntimeErrorappend)Z sequencesrs1Z candidates2seqr&r&r' _c3_merges"  rNc stt|jD]$\}tdrt|j|}q8qd}rDtngt|jd|}g}t|j|d}D]0t|rttfdd|jDst|qt|D] qfdd|D}fdd|D}fd d|D} t |gg||| |g|g|gS) aComputes the method resolution order using extended C3 linearization. If no *abcs* are given, the algorithm works exactly like the built-in C3 linearization used for method resolution. If given, *abcs* is a list of abstract base classes that should be inserted into the resulting MRO. Unrelated ABCs are ignored and don't end up in the result. The algorithm inserts ABCs where their functionality is introduced, i.e. issubclass(cls, abc) returns True for the class itself but returns False for all its direct base classes. Implicit ABCs for a given class (either registered or inferred from the presence of a special method like __len__) are inserted directly after the last ABC explicitly listed in the MRO of said class. If two implicit ABCs end up next to each other in the resulting MRO, their ordering depends on the order of types in *abcs*. __abstractmethods__rNc3s|]}t|VqdSrA) issubclass)rCb)baser&r'rlsz_c3_mro..csg|]}t|dqSabcs_c3_mrorCrrr&r'rrHz_c3_mro..csg|]}t|dqSrrrrr&r'rrHcsg|]}t|dqSrrrrr&r'rrH) enumeratereversed __bases__rcrlistranyrremover) rFriZboundaryZexplicit_basesZabstract_basesZ other_basesZexplicit_c3_mrosZabstract_c3_mrosZ other_c3_mrosr&)rrr'rsD   rcstjfddfddDfddfddDtg}D]}g}|D]0}|vrht|rh|fdd|jDqh|s||qX|jtd d |D] }|D]}||vr||qqqXt|d S) zCalculates the method resolution order for a given class *cls*. Includes relevant abstract base classes (with their respective bases) from the *types* iterable. Uses a modified C3 linearization algorithm. cs|vot|dot|S)N__mro__)rcr)typ)basesrFr&r' is_relatedsz _compose_mro..is_relatedcsg|]}|r|qSr&r&rCn)rr&r'rrHz _compose_mro..cs&D]}||kr||jvrdSqdS)NTF)r)rr-)typesr&r'is_strict_basesz$_compose_mro..is_strict_basecsg|]}|s|qSr&r&r)rr&r'rrHcsg|]}|vr|qSr&r&r)type_setr&r'rrHT)rreverser)setr__subclasses__rrsortrr)rFrmrorfoundsubZsubclsr&)rrFrrrrr' _compose_mros*    rcCstt||}d}|D]R}|dur\||vrX||jvrX||jvrXt||sXtd||qj||vr|}q||S)a^Returns the best matching implementation from *registry* for type *cls*. Where there is no registered implementation for a specific type, its method resolution order is used to find a more generic implementation. Note: if *registry* does not contain an implementation for the base *object* type, this function may return None. NzAmbiguous dispatch: {} or {})rkeysrrrrr)rFregistryrZmatchtr&r&r' _find_impls rcsddl}ddl}i|dfddd fdd fdd}t|d d |t<|_|_||_j |_ t |||S) akSingle-dispatch generic function decorator. Transforms a function into a generic function, which can have different behaviours depending upon the type of its first argument. The decorated function acts as the default implementation, and additional implementations can be registered using the register() attribute of the generic function. rNc sxdur"t}|kr"|z |}WnDtyrz |}Wntydt|}Yn0||<Yn0|S)zgeneric_func.dispatch(cls) -> Runs the dispatch algorithm to return the best available implementation for the given *cls* registered on *generic_func*. N)rrKeyErrorr)rFZ current_tokenZimpl) cache_tokendispatch_cacherr&r'dispatch0s    z singledispatch..dispatchcs|durttr fddStdi}|s@tdd}ddlm}tt||\}ttstd |d d |<durt d rt  |S) zgeneric_func.register(cls, func) -> func Registers a new implementation for the given *cls* on a *generic_func*. Ncs |SrAr&)f)rFregisterr&r'rPrHz2singledispatch..register..rz(Invalid first argument to `register()`: zS. Use either `@register(some_class)` or plain `@register` on an annotated function.r)get_type_hintszInvalid annotation for z. z is not a class.r) r}rurr\typingrrZrXrwrcrr)rFr^annrZargname)rrrrrEr'rGs,     z singledispatch..registercs*|std|dj|i|S)Nz( requires at least 1 positional argumentr)r\rg)r_Zkw)rfuncnamer&r'r hszsingledispatch..wrapperrzsingledispatch function)N) rweakrefZWeakKeyDictionaryrrBrrMappingProxyTyperrZ _clear_cacher)r^rrr r&)rrrrrrr'r s !   r c@s8eZdZdZddZd ddZd ddZed d ZdS) r zSingle-dispatch generic method descriptor. Supports wrapping existing descriptors and handles non-descriptor callables as instance methods. cCs4t|s t|ds t|dt||_||_dS)Nrz is not callable or a descriptor)rbrcr\r dispatcherr^r,r^r&r&r'rOs zsingledispatchmethod.__init__NcCs|jj||dS)zgeneric_method.register(cls, func) -> func Registers a new implementation for the given *cls* on a *generic_method*. r)rr)r,rFmethodr&r&r'rszsingledispatchmethod.registercs0fdd}j|_j|_t|j|S)Ncs(j|dj}||i|SrP)rrrgr)r_kwargsrrFrMr,r&r'rsz-singledispatchmethod.__get__.._method)rrrr^)r,rMrFrr&rr'rs  zsingledispatchmethod.__get__cCst|jddSrrrzr&r&r'rsz)singledispatchmethod.__isabstractmethod__)N)N) rrrrrOrrrrr&r&r&r'r zs   r c@s.eZdZddZddZdddZeeZdS) rcCs ||_d|_|j|_t|_dSrA)r^attrnamerrrrr&r&r'rOszcached_property.__init__cCs8|jdur||_n"||jkr4td|jd|ddS)Nz?Cannot assign the same cached_property to two different names (z and z).)rr\)r,ownernamer&r&r' __set_name__s  zcached_property.__set_name__Nc Cs|dur |S|jdurtdz |j}Wn6ty^dt|jd|jd}t|dYn0||jt}|tur|jv||jt}|tur| |}z|||j<Wn6tydt|jd|jd}t|dYn0Wdn1s0Y|S)NzGCannot use cached_property instance without calling __set_name__ on it.zNo '__dict__' attribute on z instance to cache z property.zThe '__dict__' attribute on z7 instance does not support item assignment for caching ) rr\rrrurr _NOT_FOUNDrr^)r,instancerrmsgvalr&r&r'rs:      .zcached_property.__get__)N) rrrrOrrrrrr&r&r&r'rs r)rF)N)r?r@rIrrZ _functools ImportErrorrBrYr r r rrrrrstrr~rurrrrrrrrrr r rrr&r&r&r's                     AJ   .t  -)\(