o i8@sHdZddlmZmZddlmZddlmZddlm Z ddl m Z ddl m Z mZddlmZdd lmZdd lmZdd lmZmZmZdd lmZmZdd lmZmZmZm Z m!Z!m"Z"Gddde"eZ#Gddde#e!eZ$Gddde#eZ%Gddde#e eZ&Gddde#eeZ'Gddde#eeZ(Gddde#eeZ)dS)zB Base abstract class for managed key-value store implementations. )ABCabstractmethod)Lock) defaultdict)Sequence) TracebackType)Any SupportsFloat)DEFAULT_COLLECTION_NAME)StoreSetupError) ManagedEntry)now prepare_ttl prepare_ttls)Selfoverride)AsyncCullProtocolAsyncDestroyCollectionProtocolAsyncDestroyStoreProtocol!AsyncEnumerateCollectionsProtocolAsyncEnumerateKeysProtocolAsyncKeyValueProtocolc sneZdZUdZeed<eed<eeefed<eeefed<eed<dddedBd dffd d Z dfs z2BaseStore._get_managed_entries..Nr%)r"r+r:r%r=r&_get_managed_entriescszBaseStore._get_managed_entriesr6csJ|p|j}|j|dIdH|j||dIdH}|sdS|jr"dS|jS)ajRetrieve a value by key from the specified collection. Args: collection: The collection to retrieve the value from. If no collection is provided, it will use the default collection. key: The key to retrieve the value from. Returns: The value associated with the key, or None if not found or expired. r6Nr;)rr7r9 is_expiredvaluer"r8r+ managed_entryr%r%r&geths z BaseStore.getcs@|p|j}|j|dIdH|j||dIdH}dd|DS)Nr6r:r+cSs g|] }|r |js |jndqSN)r@rAr<entryr%r%r&r>s z&BaseStore.get_many..rr7r?r"r:r+entriesr%r%r&get_manys  zBaseStore.get_manycsL|p|j}|j|dIdH|j||dIdH}|r|jr dS|j|jfS)Nr6r;NN)rr7r9r@rAttlrBr%r%r&rNs   z BaseStore.ttlcs@|p|j}|j|dIdH|j||dIdH}dd|DS)zRetrieve multiple values and TTLs by key from the specified collection. Returns a list of tuples of the form (value, ttl_seconds). Missing or expired entries are represented as (None, None). r6NrEcSs&g|]}|r|js|j|jfndqS)rM)r@rArNrGr%r%r&r>s&z&BaseStore.ttl_many..rIrJr%r%r&ttl_manys  zBaseStore.ttl_manyrCcr')z9Store a managed entry by key in the specified collection.Nr%)r"r+r8rCr%r%r&_put_managed_entryzBaseStore._put_managed_entrymanaged_entriescs4t||ddD]\}}|j|||dIdHqdS)zBStore multiple managed entries by key in the specified collection.T)strictr+r8rCN)ziprP)r"r+r:rRr8rCr%r%r&_put_managed_entriesszBaseStore._put_managed_entries)r+rNrArNcsN|p|j}|j|dIdHt|t|dtd}|j|||dIdHdS)zEStore a key-value pair in the specified collection with optional TTL.r6N)trArN created_atrT)rr7r rr rP)r"r8rAr+rNrCr%r%r&puts z BaseStore.putvaluescCsdt|t|krd}t|d|r%t|tr%t|t|kr%d}t|dt|t|d}|||fS)zPrepare multiple managed entries for a put_many operation. Inheriting classes can use this method if they need to modify a put_many operation.zGput_many called but a different number of keys and values were providedNzKput_many called but a different number of keys and ttl values were provided)rWcount)len ValueError isinstancerr)r"r:r[rNmsgttl_for_entriesr%r%r&_prepare_put_manys   zBaseStore._prepare_put_manycsd|p|j}|j|dIdH|j|||d\}}fddt|D}|j|||dIdHdS)z;Store multiple key-value pairs in the specified collection.r6N)r:r[rNcs$g|]\}}t||tdqS)rX)r r )r<irArar%r&r>sz&BaseStore.put_many..)r+r:rR)rr7rb enumeraterV)r"r:r[r+rNrRr%rdr&put_manys  zBaseStore.put_manycr')zs*         v