3
\rz + @ s d Z ddddddddd d
d g Zyd d
lmZ W n ek
rB Y nX d dlmZ d dlmZ d dl m
Z
d dl m Z d dl
mZ yd dlmZ W n" ek
r G dd dZY nX dZZd[Zeefd dZeefd dZefd d Zefd d!Zefd"d#Zefd$d%Zefd&d'Zefd(d)Zefd*d+Z efd,d-Z efd.d/Z efd0d1Z efd2d3Z efd4d5Z!d6efd7efd8efgd8efd9efd6efgd9e fd8e fd7e fgd7e fd6e fd9e!fgd:Z"d;d Z#dd d Z%yd d?lm%Z% W n ek
r* Y nX G d@d
d
e&Z'edAdBdCdDdEgZ(G dFdG dGe)Z*e& fe+e,e-e.dHhe/e.e0fdIdJZ1d\dMdZ2dNdO Z3yd dPlm3Z3 W n ek
r Y nX dQdR Z4d]dSdTZ5dUdV Z6dWdX Z7dYd Z8dHS )^zEfunctools.py - Tools for working with functions and callable objects
update_wrapperwrapsWRAPPER_ASSIGNMENTSWRAPPER_UPDATEStotal_ordering
cmp_to_key lru_cachereducepartial
partialmethodsingledispatch )r )get_cache_token)
namedtuple)MappingProxyType)WeakKeyDictionary)recursive_repr)RLockc @ s e Zd ZdZdd Zdd ZdS )r z/Dummy reentrant lock for builds without threadsc C s d S )N )selfr r !/usr/lib64/python3.6/functools.py __enter__ s zRLock.__enter__c C s d S )Nr )r exctypeexcinstexctbr r r __exit__ s zRLock.__exit__N)__name__
__module__ __qualname____doc__r r r r r r r s r r r r r __annotations____dict__c
C sn x<|D ]4}yt ||}W n tk
r, Y qX t| || qW x$|D ] }t | |jt ||i qDW || _| S )a Update 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 r r r , s
c C s t t| ||dS )a Decorator 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 r )r' r( r) r r r r J s c C s$ | j |}||kr|S | o"| |kS )zIReturn a > b. Computed by @total_ordering from (not a < b) and (a != b).)__lt__)r otherNotImplemented op_resultr r r _gt_from_ltb s
r0 c C s | j |}|p| |kS )zEReturn a <= b. Computed by @total_ordering from (a < b) or (a == b).)r, )r r- r. r/ r r r _le_from_lti s
r1 c C s | j |}||kr|S | S )z=Return a >= b. Computed by @total_ordering from (not a < b).)r, )r r- r. r/ r r r _ge_from_ltn s
r2 c C s$ | j |}||kr|S | p"| |kS )zJReturn a >= b. Computed by @total_ordering from (not a <= b) or (a == b).)__le__)r r- r. r/ r r r _ge_from_leu s
r4 c C s" | j |}||kr|S |o | |kS )zFReturn a < b. Computed by @total_ordering from (a <= b) and (a != b).)r3 )r r- r. r/ r r r _lt_from_le| s
r5 c C s | j |}||kr|S | S )z=Return a > b. Computed by @total_ordering from (not a <= b).)r3 )r r- r. r/ r r r _gt_from_le s
r6 c C s$ | j |}||kr|S | o"| |kS )zIReturn a < b. Computed by @total_ordering from (not a > b) and (a != b).)__gt__)r r- r. r/ r r r _lt_from_gt s
r8 c C s | j |}|p| |kS )zEReturn a >= b. Computed by @total_ordering from (a > b) or (a == b).)r7 )r r- r. r/ r r r _ge_from_gt s
r9 c C s | j |}||kr|S | S )z=Return a <= b. Computed by @total_ordering from (not a > b).)r7 )r r- r. r/ r r r _le_from_gt s
r: c C s$ | j |}||kr|S | p"| |kS )zJReturn a <= b. Computed by @total_ordering from (not a >= b) or (a == b).)__ge__)r r- r. r/ r r r _le_from_ge s
r< c C s" | j |}||kr|S |o | |kS )zFReturn a > b. Computed by @total_ordering from (a >= b) and (a != b).)r; )r r- r. r/ r r r _gt_from_ge s
r= c C s | j |}||kr|S | S )z=Return a < b. Computed by @total_ordering from (not a >= b).)r; )r r- r. r/ r r r _lt_from_ge s
r> r7 r3 r; r, )r, r3 r7 r; c sZ fddt D }|s tdt|}x.t | D ]"\}}||kr0||_t || q0W S )z6Class decorator that fills in missing ordering methodsc s( g | ] }t |d t t|d k r|qS )N)r! object).0op)clsr r
s z"total_ordering..z6must define at least one ordering operation: < > <= >=)_convert
ValueErrormaxr r# )rB rootsrootopnameopfuncr )rB r r s c s G fdddt }|S )z,Convert a cmp= function into a key= functionc sZ e Zd ZdgZdd Z fddZ fddZ fdd Z fd
d Z fd d
Z dZ
dS )zcmp_to_key..Kobjc S s
|| _ d S )N)rK )r rK r r r __init__ s z cmp_to_key..K.__init__c s | j |j dk S )Nr )rK )r r- )mycmpr r r, s z cmp_to_key..K.__lt__c s | j |j dkS )Nr )rK )r r- )rM r r r7 s z cmp_to_key..K.__gt__c s | j |j dkS )Nr )rK )r r- )rM r r __eq__ s z cmp_to_key..K.__eq__c s | j |j dkS )Nr )rK )r r- )rM r r r3 s z cmp_to_key..K.__le__c s | j |j dkS )Nr )rK )r r- )rM r r r; s z cmp_to_key..K.__ge__N) r r r __slots__rL r, r7 rN r3 r; __hash__r )rM r r K s rQ )r? )rM rQ r )rM r r s )r c sJ e Zd ZdZdZ fddZd d
Ze d d Zd
d Z dd Z
Z S )r zSNew function with partial application of the given arguments
and keywords.
funcargskeywordsr __weakref__c s | s t dt| dk r t d| ^}}} t|s:t dt| } t|drv|j| } |jj }|j| |}~|j }t
t |j |}||_ | |_||_|S )Nz1descriptor '__new__' of partial needs an argument z*type 'partial' takes at least one argumentz#the first argument must be callablerR )
TypeErrorlencallabletuplehasattrrS rT copyr$ rR superr __new__)rS rT rB rR Ztmpkwr ) __class__r r r^ s(
zpartial.__new__c O s: | s t d| ^}} |jj }|j| |j|j| |S )Nz2descriptor '__call__' of partial needs an argument)rW rT r\ r$ rR rS )rS rT r Z newkeywordsr r r __call__ s
zpartial.__call__c C s t | j}t| jg}|jdd | jD |jdd | jj D t | jdkrld| ddj | dS | ddj | dS ) Nc s s | ]}t |V qd S )N)repr)r@ xr r r s z#partial.__repr__..c s s | ]\}}| d |V qdS )=Nr )r@ kvr r r rc s functoolsz
functools.(z, ))
typer ra rR extendrS rT itemsr join)r qualnamerS r r r __repr__ s
zpartial.__repr__c C s* t | | jf| j| j| jp d | jp$d ffS )N)rj rR rS rT r )r r r r
__reduce__# s zpartial.__reduce__c C s t |tstdt|dkr0tdt| |\}}}}t| szt |t sz|d k rft |t sz|d k rt |t rtdt|}|d kri }nt|tk rt|}|d kri }|| _|| _|| _ || _
d S )Nz(argument to __setstate__ must be a tuple z expected 4 items in state, got zinvalid partial state)
isinstancerZ rW rX rY dictrj r rR rS rT )r staterR rS kwds namespacer r r __setstate__' s(
zpartial.__setstate__)rR rS rT r rU ) r r r r rO r^ r` r ro rp rw
__classcell__r r )r_ r r s
)r c @ s< e Zd ZdZdd Zdd Zdd Zdd Zed
d Z d S )
r
zMethod descriptor with partial application of the given arguments
and keywords.
Supports wrapping existing descriptors and handles non-descriptor
callables as instance methods.
c O sr t | r$t|d r$tdj|t|tr\|j| _|j| | _|jj | _| jj
| n|| _|| _|| _d S )N__get__z${!r} is not callable or a descriptor) rY r[ rW formatrr r
rR rS rT r\ r$ )r rR rS rT r r r rL M s
zpartialmethod.__init__c C sN dj tt| j}dj dd | jj D }d}|j| jj| jj | j
||dS )Nz, c s s | ]\}}d j ||V qdS )z{}={!r}N)rz )r@ re rf r r r rc c s z)partialmethod.__repr__..z*{module}.{cls}({func}, {args}, {keywords}))modulerB rR rS rT ) rm mapra rS rT rl rz r_ r r rR )r rS rT
format_stringr r r ro a s
zpartialmethod.__repr__c s fdd} j |_ |_|S )Nc s< j j }|j| | ^}}|f j t| } j||S )N)rT r\ r$ rS rZ rR )rS rT
call_keywords cls_or_selfrest call_args)r r r _methodm s
z3partialmethod._make_unbound_method.._method)__isabstractmethod___partialmethod)r r r )r r _make_unbound_methodl s z"partialmethod._make_unbound_methodc C s t | jdd }d }|d k rd|||}|| jk rdt|f| j| j}y |j|_W n tk
rb Y nX |d kr|| j j||}|S )Nry ) r! rR r rS rT __self__r" r ry )r rK rB getresultnew_funcr r r ry w s
zpartialmethod.__get__c C s t | jddS )Nr F)r! rR )r r r r r s z"partialmethod.__isabstractmethod__N)
r r r r rL ro r ry propertyr r r r r r
E s CacheInfohitsmissesmaxsizecurrsizec @ s( e Zd ZdZdZefddZdd ZdS )
_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.
hashvaluec C s || d d < ||| _ d S )N)r )r tuphashr r r rL s z_HashedSeq.__init__c C s | j S )N)r )r r r r rP s z_HashedSeq.__hash__N)r r r r rO r rL rP r r r r r s r Nc
s | }|r*||7 }x|j D ] } || 7 }qW |rl|| fdd| D 7 }|r|| fdd|j D 7 }n$||dkr |d |kr|d S t|S )a Make 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.
c 3 s | ]} |V qd S )Nr )r@ rf )rj r r rc s z _make_key..c 3 s | ]} |V qd S )Nr )r@ rf )rj r r rc s r )rl valuesr )
rS ru typedkwd_mark fasttypesrZ rj rX keyitemr )rj r _make_key s r Fc s. dk r t t r td fdd}|S )a Least-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_algorithms#Least_Recently_Used
Nz)Expected maxsize to be an integer or Nonec s t | t}t|| S )N)_lru_cache_wrapper
_CacheInfor )
user_functionr& )r r r r decorating_function s z&lru_cache..decorating_function)rr intrW )r r r r )r r r r s c s t t d\ i d
djjt
g d d gd d < dkrh
fdd}nN d kr
fdd}n*
fd d}
fd
d }
fd d
}||_||_|S )Nr r rV Fc s | |} d7 |S )Nr r )rS ru r )r r r r r& s
z#_lru_cache_wrapper..wrapperc sH | |}|}|k r*d7 |S | |}| |< d7 |S )Nr r )rS ru r r )cache cache_getr make_keyr sentinelr r r r r& s
c s2
| |} f |}|d k rr|\}}}}||< ||<
}| |<
< ||<
|< d7 |S W d Q R X | |} |krnr
} || < || < |
}
} d
<
< |
= | |< n6
}|
||g}| |<
< |< k d7 W d Q R X |S )Nr r ) rS ru r linkZ link_prevZ link_nextZ_keyr ZlastZoldrootZoldkeyZ oldresult)KEYNEXTPREVRESULTr r cache_lenfullr lockr r r rH r r r r r& sB
c s S Q R X dS )zReport cache statisticsNr r )r r r r r r r r
cache_infoG s z&_lru_cache_wrapper..cache_infoc s< . j ddgdd< d dW dQ R X dS )z$Clear the cache and cache statisticsNr F)clearr )r r r r r rH r r cache_clearL s
z'_lru_cache_wrapper..cache_clear)r r rV r )r? r r __len__r r r )r r r r r& r r r )r r r r r r r r r r r r r r rH r r r r r s* *9 r )r c C s g }xdd | D } | s |S x:| D ]2}|d }x$| D ]}||dd kr4d}P q4W P q"W |dkrht d|j| x | D ]}|d |krx|d= qxW qW dS )zMerges MROs in *sequences* to a single MRO using the C3 algorithm.
Adapted from http://www.python.org/download/releases/2.3/mro/.
c S s g | ] }|r|qS r r )r@ sr r r rC k s z _c3_merge..r r NzInconsistent hierarchy) RuntimeErrorappend) sequencesr s1 candidates2seqr r r _c3_mergec s$
r c
s& x8t t| jD ]"\}tdrt| j| }P qW d} rFt ng t| jd| }g }t| j|d }x: D ]2t| rxtfdd| jD rx|j qxW x|D ] j qW fdd|D } fdd|D } fd d|D } t
| gg| | | |g |g |g S )
a Computes 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__r Nc 3 s | ]}t | V qd S )N)
issubclass)r@ b)baser r rc s z_c3_mro..c s g | ]}t | d qS ))abcs)_c3_mro)r@ r )r r r rC s z_c3_mro..c s g | ]}t | d qS ))r )r )r@ r )r r r rC s c s g | ]}t | d qS ))r )r )r@ r )r r r rC s ) enumeratereversed __bases__r[ rX listr anyr remover )
rB r iboundaryexplicit_basesabstract_bases other_basesexplicit_c3_mrosabstract_c3_mros
other_c3_mrosr )r r r r ~ s(
r c s t j fddfddD fddfddD t g }xD ]}g }x<|j D ]0}| krlt|rl|jfdd|jD qlW |s|j| qZ|jtd d
x,|D ]$}x |D ]}||kr|j| qW qW qZW t|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.
c s | kot | dot| S )N__mro__)r[ r )typ)basesrB r r
is_related s z _compose_mro..is_relatedc s g | ]} |r|qS r r )r@ n)r r r rC s z _compose_mro..c s( x" D ]}| |kr| |j krdS qW dS )NTF)r )r r- )typesr r is_strict_base s
z$_compose_mro..is_strict_basec s g | ]} |s|qS r r )r@ r )r r r rC s c s g | ]}| kr|qS r r )r@ r )type_setr r rC s T)r reverse)r )setr __subclasses__r r sortrX r )rB r mror foundsubsubclsr )r rB r r r r r _compose_mro s*
r c C sx t | |j }d}xZ|D ]R}|dk r^||kr\|| jkr\|| jkr\t|| r\tdj||P ||kr|}qW |j|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.
Nz Ambiguous dispatch: {} or {})r keysr r r rz r )rB registryr matchtr r r
_find_impl s
r c s i t d fddd
fdd fdd}t| dd | t< |_|_t|_j|_t || |S ) al Single-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.
Nc s| dk r"t } |kr"j | y | }W nH tk
rv y | }W n tk
rh t| }Y nX || < Y nX |S )zgeneric_func.dispatch(cls) ->
Runs the dispatch algorithm to return the best available implementation
for the given *cls* registered on *generic_func*.
N)r
r KeyErrorr )rB
current_tokenimpl) cache_tokendispatch_cacher r r dispatch s z singledispatch..dispatchc sB |dkr fddS | < dkr6t dr6t j |S )zgeneric_func.register(cls, func) -> func
Registers a new implementation for the given *cls* on a *generic_func*.
Nc s
| S )Nr )f)rB registerr r s z2singledispatch..register..r )r[ r
r )rB rR )r r r r )rB r r s z singledispatch..registerc s& | st d | d j| |S )Nz( requires at least 1 positional argumentr )rW r_ )rS kw)r funcnamer r r& " s z singledispatch..wrapperr zsingledispatch function)N)
r r! r? r r r r r _clear_cacher )rR r& r )r r r r r r r r s
)r r r r r )r )r F)N)9r __all__
_functoolsr ImportErrorabcr
collectionsr r r weakrefr reprlibr _threadr r r r r r. r0 r1 r2 r4 r5 r6 r8 r9 r: r< r= r> rD r r r r? r
r r r r str frozensetrj rZ rX r r r r r r r r r r r r s
NN
%t
-)