U
Wz8\ @ s d dl Z d dlmZ d dlmZmZmZmZ zd dl mZ W n e k
rZ e
e dZY nX dZ dZ dd Z
d
d Zd d
Zdd Zdd Zdd ZddddddZe jdd Zdd Zd d Zd d Zd!d" Zd#d$ Zd%d& Zd+ddddd(d)d*Z dS ), N) OrderedDict)
_CacheInfo _make_keypartialwraps)
ensure_futureasyncz1.0.2)
alru_cachec C s0 z
| W S tk
r* tj| d Y S X d S )Nloop)
create_futureAttributeErrorasyncioZFuturer
r :/opt/alt/python38/lib/python3.8/site-packages/async_lru.pyr s
r c C s t | dr| j} q | S )Nfunc)hasattrr )fnr r r unpartial s
r c C sD | r| d S | }|d k r2| | d S | | d S N)Z cancelledcancelZ exceptionZ
set_exceptionZ
set_resultresult)futtaskexcr r r _done_callback s
r c O s* t |||}|| jk}|r&| j| |S r )r _cachepop)wrappedtypedargskwargskeyexistsr r r _cache_invalidate* s
r$ c C s d | _ | _t | _t | _d S )Nr )hitsmissesr r settasks)r r r r _cache_clear5 s r) c C s` | j std| j| j koDt| j koDt| j koDdkn }|sVtdd| _ d S )Nzalru_cache is not closedr z#alru_cache was not closed correctlyF)closed RuntimeErrorr% r& lenr( r )r Z
was_closedr r r _open; s r- FT)r return_exceptionsr c C sB | j rtdd| _ |r4| jD ]}| s | q t| ||dS )Nzalru_cache is closedTr. r )r* r+ r( doner _wait_closed)r r r. r r r r r _closeM s
r2 c c sV |d krt }t j| j||d}|tt| |E d H }t jd|dE d H |S )Nr/ r r
)r get_event_loopZgatherr( add_done_callbackr
_close_waitedsleep)r r. r Z wait_closedZretr r r r1 _ s
r1 c C s | d S r ) cache_clear)r _r r r r5 t s r5 c C s t | j| j|t| jS r )r r% r& r, r )r maxsizer r r _cache_infox s r: c C s* z| j | W n tk
r$ Y nX d S r )r move_to_endKeyErrorr r" r r r
__cache_touch s r> c C s | j d7 _ t| | d S N )r% r> r= r r r
_cache_hit s rA c C s | j d7 _ t| | d S r? )r& r> r= r r r _cache_miss s rB c C sZ t |tr@| r6t|dd }|d kr*|d }t||}qV|| }n|d krRt }n|}|S )N__self__r )
isinstancestrgetattrr r3 )clsr! r fn_args fn_kwargsr Z_self_loopr r r _get_loop s
rK )rG r! cache_exceptionsr c sN fdd}| d kr"|S t | s4t| dr<|| S td| d S )Nc
s t }t|s td t dr2 t tj fddt |_
d_ t t
_t t _t t_t t_t t_S )Nz&Coroutine function is required, got {}_make_unbound_methodc ? s: j rtdtj| |d}t| |}j|}|d k r| spt | t
j ||dE d H S |j }|d ks rt | |
S j| t|d}| |}t||d}|tt| j| |jj |j|< d k r tjkr jjdd t| t
j ||dE d H S )Nzalru_cache is closed for {}r
F)Zlast)r* r+ formatrK _originr r getr0 rA r ZshieldZ
_exceptionr r r r r4 r r r( addremover, popitemrB )rH rI rJ r" r r coror )rM rG r r! r r9 r r r r r sD
z,alru_cache.<locals>.wrapper.<locals>.wrappedF)r r Ziscoroutinefunctionr+ rO r rN r coroutiner) rP r* r r:
cache_infor7 r$ Z
invalidater2 closer- open)r rP rM rG r! r r9 r )r r r wrapper s&
0 zalru_cache.<locals>.wrapperrN z {} decorating is not supported)callabler NotImplementedErrorrO )r r9 r rG r! rM r r[ r rZ r r s
Hr )NrL F) r collectionsr functoolsr r r r r ImportErrorrF __version____all__r r r r$ r) r- r2 rV r1 r5 r: r> rA rB rK r r r r r <module> s@