g |dZddlmZddlmZmZddlmZmZddl m Z m Z ddl m Z mZddlmZdZGd d eeZy ) z*Contains the MemoryDatabase implementation) LooseObjectDB) ObjectDBR ObjectDBW)OStreamIStream) BadObjectUnsupportedOperation)ZippedStoreShaWriterDecompressMemMapReader)BytesIO)MemoryDBcReZdZdZfdZdZdZdZdZdZ dZ d Z d Z xZ S) r a/A memory database stores everything to memory, providing fast IO and object retrieval. It should be used to buffer results and obtain SHAs before writing it to the actual physical storage, as it allows to query whether object already exists in the target storage before introducing actual IOc`t|td|_t |_y)Nzpath/doesnt/matter)super__init__r_dbdict_cache)self __class__s A/var/www/openai/venv/lib/python3.12/site-packages/gitdb/db/mem.pyrzMemoryDB.__init__'s&  !56f ctd)Nz$MemoryDB's always stream into memory)r )rstreams r set_ostreamzMemoryDB.set_ostream.s"#IJJrct}|jj||jj|}|j |j dt |jd}t|j|j|j||j|j<|S)NrF)close_on_deletion) r rrstorecloseseekr getvaluerbinshatypesizer)ristreamzstream decomp_streams rrzMemoryDB.store1s&( W%((..)  Q/w/?/?/AUZ[ &-gnngllGLLZg&h GNN#rc||jvSN)rrshas r has_objectzMemoryDB.has_object@sdkk!!rc$|j|Sr))rr*s rinfoz MemoryDB.infoCs{{3rc |j|}|jjd|S#t$r}t ||d}~wwxYw)Nr)rrr KeyErrorr)rr+ostreames rrzMemoryDB.streamGsH (kk#&G NN   "N (C.a ' (s+. A AAc,t|jSr))lenrrs rr$z MemoryDB.sizeQs4;;rc6|jjSr))rkeysr5s rsha_iterzMemoryDB.sha_iterTs{{!!rcd}|D]}|j|r|j|}t|jj}t |j |j ||}|j||dz }|S)arCopy the streams as identified by sha's yielded by sha_iter into the given odb The streams will be copied directly **Note:** the object will only be written if it did not exist in the target db :return: amount of streams actually copied into odb. If smaller than the amount of input shas, one or more objects did already exist in odbr)r,rr datarr#r$r)rr8odbcountr+r1sior%s r stream_copyzMemoryDB.stream_copyXs}C~~c"kk#&G'..--/0CgllGLL#sCG IIg  QJE r)__name__ __module__ __qualname____doc__rrrr,r.rr$r8r? __classcell__)rs@rr r s7A K " ( "rr N)rCgitdb.db.looser gitdb.db.baserr gitdb.baserr gitdb.excrr gitdb.streamr r ior __all__r rrrMs; 1(     My)Mr