Ki X&UdZddlmZmZddlmZddlmZddlm Z m Z ddl m Z ddl m Z ddlmZmZdd lmZmZdd lmZdd lmZdd lmZdd lmZddlmZmZddlm Z m!Z!ddl"m#Z#ddl$m%Z%m&Z&ddl'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-e e.e e.e e.efffZ/ee.ee.ee.efffZ0eiZ1e0e2d<de/de0fdZ3Gdde-eZ4Gdde4e,eZ5Gdde4eZ6Gdde4e+eZ7Gdd e4e*eZ8Gd!d"e4e)eZ9Gd#d$e4e(eZ:y%)&zB Base abstract class for managed key-value store implementations. )ABCabstractmethod)Lock) defaultdict)MappingSequence)AsyncExitStack)datetime)MappingProxyType TracebackType)Any SupportsFloat)DEFAULT_COLLECTION_NAME)StoreSetupError) bear_enforce) ManagedEntry)PassthroughStrategySanitizationStrategy)BasicSerializationAdapterSerializationAdapter)prepare_entry_timestamps)Selfoverride)AsyncCullProtocolAsyncDestroyCollectionProtocolAsyncDestroyStoreProtocol!AsyncEnumerateCollectionsProtocolAsyncEnumerateKeysProtocolAsyncKeyValueProtocolDEFAULT_SEED_DATAseedreturnct|jDcic]<\}}|t|jDcic]\}}|t|c}}>c}}}}Scc}}wcc}}}}w)aConvert mutable seed data to an immutable frozen structure. This function converts the nested mapping structure of seed data into immutable MappingProxyType objects at all levels. Using immutable structures prevents accidental modification of seed data after store initialization and ensures thread-safety. Args: seed: The mutable seed data mapping: {collection: {key: {field: value}}}. Returns: An immutable frozen version of the seed data using MappingProxyType. )r items)r! collectionr$keyvalues e/home/jay/workspace/scripts/.codegraph-venv/lib/python3.12/site-packages/key_value/aio/stores/base.py_seed_to_frozen_seed_datar)%s @D@J@J@L M Mj{jtv{%V[VaVaVc&d Us,zBaseStore.__init__Qs2 %6'24'8$*5d*;'.DJB? "4"O8O&;&Z?X?Z#*C*\GZG\'1Q1jUhUj.%  & & ( c Kyw)z4Initialize the store (called once before first use).Nr?s r(_setupzBaseStore._setupr%c Kyw)zKInitialize the collection (called once before first use of the collection).NrCr?r%s r(_setup_collectionzBaseStore._setup_collectionrFrGr&cJ|j||j|fS)Nr%)r&)_sanitize_collection _sanitize_keyr?r%r&s r(_sanitize_collection_and_keyz&BaseStore._sanitize_collection_and_keys*((J(?ASASX[ASA\\\rAcr|jj||jj|SN)r')r2validatesanitizerIs r(rMzBaseStore._sanitize_collections3 ..77j7I55>>Z>PPrAcr|jj||jj|SrR)r1rSrT)r?r&s r(rNzBaseStore._sanitize_keys3 ''00s0;..77c7BBrAcK|jjD]\\}}|j|d{|jD]*\}}|j|t ||d{,^y7D7 w)z+Seed the store with the data from the seed.rLN)r&r'r%)r3r$setup_collectionputdict)r?r%r$r&r's r( _seed_storezBaseStore._seed_stores~!%!1!1!3 R J'':'> > >#kkm R Uhh3d5kjhQQQ R R >Qs!6BA=:B3A?4 B?BcK|jsj|j4d{|js8 |jd{d|_|jd{dddd{yy7Z77#t$r,}t d|d|j j i|d}~wwxYw7R7D#1d{7swYyxYww)a Initialize the store if not already initialized. This method is called automatically before any store operations and uses a lock to ensure thread-safe lazy initialization. It can also be called manually to ensure the store is ready before performing operations. The setup process includes calling the `_setup()` hook and seeding the store with initial data if provided. This method is idempotent - calling it multiple times has no additional effect after the first call. Nz!Failed to setup key value store: storemessage extra_infoT)r,r-rE Exceptionrr@__name__rZ)r?es r(setupzBaseStore.setups##'' - -++!"kkm++ ,0D(**,,, - - -$ -,$!-&Gs$KY`bfbpbpbybyXz !!- - - - -sCA:C B:A>A<A> B:#B6$B:( C3B84C<A>> B3'B..B33B:8C:C C C CcK|jd{|j|sb|j|4d{|j|s* |j|d{d|j|<dddd{yy7v7Q7)#t$r}t d|d|i|d}~wwxYw70#1d{7swYyxYww)aInitialize a specific collection if not already initialized. This method is called automatically before any collection-specific operations and uses a per-collection lock to ensure thread-safe lazy initialization. It can also be called manually to ensure a collection is ready before performing operations on it. The setup process includes calling the `_setup_collection()` hook for store-specific collection initialization. This method is idempotent - calling it multiple times for the same collection has no additional effect after the first call. Args: collection: The name of the collection to initialize. NrLzFailed to setup collection: r%r]T)rcr/r.rJr`r)r?r%rbs r(rWzBaseStore.setup_collectionsjjl..z:33J? G G66zBH"44 4KKKCGD33J?  G G G;  GL$H-8TUVTW6Xfrt~eAGHHH  G G G GsCB &CBCB9B'B(B,B9; CB7CCB B4B//B44B97C9C ?C C Cc Kyw)zs :4#:#: ##z#:::$<$E! Az&2:bl&m %%!'&   ;  s!&A,A(9A,"A*#A,*A,valuescPK|xs |j}|j|d{t|t|k7rd}t|dt |\}}}|D cgc]} t | ||} } |j ||| |||d{y7tcc} w7 w)z;Store multiple key-value pairs in the specified collection.rLNzGput_many called but a different number of keys and values were providedrr)r%rgr}rxr~r)r4rWlen ValueErrorrrr) r?rgrr%rxmsgr~ ttl_secondsrr'r}s r(put_manyzBaseStore.put_manyOs :4#:#: ##z#::: t9F #[CS/t +.F3.O+ KEK/L|A|%T^ku/v/L/L''!+!! (    ;/L s-&B&B=B&&B;B&B$B&B&c Kyw)zrErJtuplerPrMrNrZrcrWrrrfrlistrjrrrYr rorufloatrxrzr|r rrrrXrrintrrrr< __classcell__r@s@r(r+r+7sU (d33 +CI 6600 44';;   >BAEHL)-&* , 4d:,$8$#> , +?*E ,  $J ,t#,, ,\CZSZTZ]s]]sTWx]QsQsQCCCR-0GCGDG2KcKK W[H[KK__8C=_UYZfimZmUn_  "& ) )$J ) c3h$  ))8 NRd8C=dtdW[\`adfiai\jmq\qWrdd >B >S >t >uTRUWZRZ^^bMbdilpdpMpGq > > "& ysmy$J y eDcNT)54<78 9 yy"    $     sm  ",/  T\ tO 6 X\z~  S  c):  3QU:  cpswcw  DH     "&$(  sm c*+ $J  T !     < # 3 4   Xc] PS X[  AEPPC$JP$PP  QUThsmTC$JTZ]TT rAr+c xeZdZdZed dddedzdedzdeefdZe dddededzdeefdZ y) BaseEnumerateKeysStorez{An abstract base class for enumerate key-value stores. Subclasses must implement the _get_collection_keys method. Nlimitr%rr"cK|xs |j}|j|d{|j||d{S77w)*List all keys in the specified collection.rLN)r%r)r4rW_get_collection_keysr?r%rs r(rgzBaseEnumerateKeysStore.keyssQ :4#:#: ##z#:::..*E.RRR ;Rrc Kyw)rNrCrs r(rz+BaseEnumerateKeysStore._get_collection_keysrFrGN) rarrrrrrrrgrrrCrAr(rrs StSS4ZS#*SX\]`XaSSQU99C$J9Z^_bZc99rArc eZdZUdZeed<eed<eed<dddeded d ffd Zdd Z d e fd Z de e d zde d zded zd ed zfdZddZdfd ZxZS)BaseContextManagerStoreaAn abstract base class for context manager stores. Stores that accept a client parameter should pass `client_provided_by_user=True` to the constructor. This ensures the store does not manage the lifecycle of user-provided clients (i.e., does not close them). The base class provides an AsyncExitStack that stores can use to register cleanup callbacks. Stores should add their cleanup operations to the exit stack as needed. The base class handles entering and exiting the exit stack. _client_provided_by_user _exit_stack_exit_stack_enteredF)client_provided_by_userrkwargsr"Nc ^||_t|_d|_t |di|y)aInitialize the context manager store with client ownership configuration. Args: client_provided_by_user: Whether the client was provided by the user. If True, the store will not manage the client's lifecycle (will not close it). Defaults to False. **kwargs: Additional arguments to pass to the base store constructor. FNrC)rr rrr=r>)r?rrr@s r(r>z BaseContextManagerStore.__init__s1)@%)+#(  "6"rAc|K|js*|jjd{d|_yy7 w)z'Ensure the exit stack has been entered.NT)rr __aenter__rDs r(_ensure_exit_stack_enteredz2BaseContextManagerStore._ensure_exit_stack_entereds7''""--/ / /'+D $( / *<:<cvK|jd{|jd{|S77wr)rrcrDs r(rz"BaseContextManagerStore.__aenter__s5--///jjl  0s959799exc_type exc_value tracebackcK|jr.|jj|||d{}d|_|Sy7wNF)rr __aexit__)r?rrrresults r(rz!BaseContextManagerStore.__aexit__sD  # #++55h 9UUF',D $M Vs-A>Ac|K|jr*|jjd{d|_yy7 wr)rracloserDs r(closezBaseContextManagerStore.closes7  # #""))+ + +',D $ $ +rcrK|jd{t| d{y77w)zInitialize the store if not already initialized. This override ensures the exit stack is entered before the store's _setup() method is called, allowing stores to register cleanup callbacks during setup. N)rr=rc)r?r@s r(rczBaseContextManagerStore.setups3--///gmo 0s737577r)rarrrr:rr r r>rrrtype BaseExceptionr rrrcrrs@r(rrs #":? #4 #3 #SW #, $ ]+d2 ?Lt?S `mpt`t   -  rArcheZdZdZedddedzdeefdZe dddedzdeefdZ y)BaseEnumerateCollectionsStorez~An abstract base class for enumerate collections stores. Subclasses must implement the _get_collection_names method. Nrrr"cvK|jd{|j|d{S77w)DList all available collection names (may include empty collections).Nr)rc_get_collection_namesr?rs r( collectionsz)BaseEnumerateCollectionsStore.collectionss7jjl//e/<<< <959799c Kyw)rNrCrs r(rz3BaseEnumerateCollectionsStore._get_collection_namesrFrG) rarrrrrrrrrrrCrAr(rrse 7;=#*=S == AESC$JS$s)SSrArc<eZdZdZedefdZedefdZy)BaseDestroyStorezlAn abstract base class for destroyable stores. Subclasses must implement the _delete_store method. r"crK|jd{|jd{S77w)zDestroy the store.N)rc _delete_storerDs r(destroyzBaseDestroyStore.destroy s2jjl''))) )737577c Kyw)zDelete the store.NrCrDs r(rzBaseDestroyStore._delete_storerrGN) rarrrrr:rrrrCrAr(rrs= *t**  T  rArcDeZdZdZededefdZededefdZ y)BaseDestroyCollectionStorezvAn abstract base class for destroyable collections. Subclasses must implement the _delete_collection method. r%r"cvK|jd{|j|d{S77w)zDestroy the collection.NrL)rc_delete_collectionrIs r(destroy_collectionz-BaseDestroyCollectionStore.destroy_collection s7jjl,, ,CCC Crc Kyw)zDelete the collection.NrCrIs r(rz-BaseDestroyCollectionStore._delete_collection'rrGN) rarrrrrr:rrrrCrAr(rrsO D3D4DD  c d  rArc4eZdZdZeddZeddZy) BaseCullStorezaAn abstract base class for cullable stores. Subclasses must implement the _cull method. NcrK|jd{|jd{S77wzCull the store.N)rc_cullrDs r(cullzBaseCullStore.cull3s/jjlZZ\!! !rc KywrrCrDs r(rzBaseCullStore._cull:rrGr)rarrrrrrrrCrAr(rr-s/ ""   rArN);rabcrr asyncio.locksrrrcollections.abcrr contextlibr r typesr r typingr rkey_value.shared.constantsrkey_value.shared.errorsr)key_value.shared.type_checking.bear_sprayr$key_value.shared.utils.managed_entryr#key_value.shared.utils.sanitizationrr$key_value.shared.utils.serializationrr#key_value.shared.utils.time_to_livertyping_extensionsrr!key_value.aio.protocols.key_valuerrrrrrrrrr rr)r+rrrrrrrCrAr(rs2$#-%1%>3B=Y`H,gc738+<&<==>(.>sDTUXZ]U]D^?^._)_`+;B+?(?N7L$` %s` F 9Y(BC9(DiDNSI/PRUS$ y";S & ,JC & I0# rA