U Cv f2@shddlmZddlZddlmZddlmZmZmZm Z m Z e dZ e dZ Gddde e e fZ dS) ) annotationsN) OrderedDict)AnyCallableIteratorMutableMappingTypeVarKVc@seZdZUdZded<ded<ded<ded <d Zd'ddd d dZdddddZdddddZddddddZ dddddZ dddd Z ddd!d"Z e ddd#d$Zejddd%d&d$Zd S)(LRUCachealThread-safe LRUCache based on an OrderedDict. All dict operations (__getitem__, __setitem__, __contains__) update the priority of the relevant key and take O(1) time. The dict is iterated over in order from the oldest to newest key, which means that a complete pass over the dict should not affect the order of any entries. When a new item is set and the maximum size of the cache is exceeded, the oldest item is dropped and called with ``on_evict(key, value)``. The ``maxsize`` property can be used to view or adjust the capacity of the cache, e.g., ``cache.maxsize = new_size``. zOrderedDict[K, V]_cacheint_maxsizezthreading.RLock_lockzCallable[[K, V], Any] | None _on_evict)r rrrN)maxsizeon_evictcCsDt|tstd|dkr"td||_t|_t|_ ||_ dS)a  Parameters ---------- maxsize : int Integer maximum number of items to hold in the cache. on_evict : callable, optional Function to call like ``on_evict(key, value)`` when items are evicted. zmaxsize must be an integerrmaxsize must be non-negativeN) isinstancer TypeError ValueErrorrrr threadingRLockrr)selfrrr=/tmp/pip-unpacked-wheel-h316xyqg/xarray/backends/lru_cache.py__init__!s  zLRUCache.__init__r r )keyreturnc Cs8|j(|j|}|j||W5QRSQRXdSN)rr move_to_endrrvaluerrr __getitem__4s  zLRUCache.__getitem__None)capacityrcCs<t|j|kr8|jjdd\}}|jdk r|||qdS)z9Shrink the cache if necessary, evicting the oldest items.F)lastN)lenr popitemr)rr%rr"rrr_enforce_size_limit;s zLRUCache._enforce_size_limit)rr"rc Csl|j\||jkr&|j|=||j|<n8|jrH||jd||j|<n|jdk r^|||W5QRXdS)N)rr rr)rr!rrr __setitem__Bs    zLRUCache.__setitem__cCs |j|=dSr)r )rrrrr __delitem__PszLRUCache.__delitem__z Iterator[K])rcCstt|jSr)iterlistr rrrr__iter__SszLRUCache.__iter__cCs t|jSr)r'r r/rrr__len__XszLRUCache.__len__cCs|jS)z1Maximum number of items can be held in the cache.)rr/rrrr[szLRUCache.maxsize)sizerc Cs6|dkrtd|j||||_W5QRXdS)z9Resize the cache, evicting the oldest items if necessary.rrN)rrr)r)rr2rrrr`s  )N)__name__ __module__ __qualname____doc____annotations__ __slots__rr#r)r+r,r0r1propertyrsetterrrrrr s" r ) __future__rr collectionsrtypingrrrrrr r r rrrrs