o Nie*@sddlmZzddlmZWneyddlmZYnwddlmZddlmZddl m Z m Z m Z  dd d Zed d ZGd ddeZeaddZddZddZddZdS)) namedtuple)Iterable)partial)local)Promiseasync_instanceget_default_schedulerF) AnyListSizedCallableOptionalTupleUnionIteratorHashablecs*tdfddtdtDS)Nrc3s |] }||VqdSN).0i chunk_size iterable_objrF/home/ubuntu/.local/lib/python3.10/site-packages/promise/dataloader.py s  zget_chunks..r)maxrangelen)rrrrr get_chunkss  rLoaderzkey,resolve,rejectc@sbeZdZdZdZdZ       dddZdddZddZd d Z d d Z d dZ ddZ dS) DataLoaderTNcCsx|dur||_t|jstd||dur||_|dur!||_|dur(||_|p-dd|_|p2i|_g|_ ||_ dS)NznDataLoader must be have a batch_load_fn which accepts List and returns Promise>, but got: {}.cSs|Srr)xrrrKsz%DataLoader.__init__..) batch_load_fncallable TypeErrorformatbatchmax_batch_sizecache get_cache_key_promise_cache_queue _scheduler)selfr$r(r)r*r+ cache_map schedulerrrr__init__+s"    zDataLoader.__init__cCs^|dur td|||}|jr|j|}|r|Stt|j|}|jr-||j|<|S)z[ Loads a key, returning a `Promise` for the value represented by that key. NzCThe loader.load() function must be called with a value,but got: {}.) r&r'r+r*r,getrrdo_resolve_reject)r/key cache_keycached_promisepromiserrrloadPs   zDataLoader.loadcCsP|jt|||dt|jdkr&|jr ttt||jdSt|dSdS)N)r5resolverejectr) r-appendr rr(enqueue_post_promise_jobrdispatch_queuer.)r/r5r:r;rrrr4os  zDataLoader.do_resolve_rejectcs0t|ts td|tfdd|DS)a. Loads multiple keys, promising an array of values >>> a, b = await my_loader.load_many([ 'a', 'b' ]) This is equivalent to the more verbose: >>> a, b = await Promise.all([ >>> my_loader.load('a'), >>> my_loader.load('b') >>> ]) zJThe loader.loadMany() function must be called with Array but got: {}.csg|]}|qSr)r9)rr5r/rr sz(DataLoader.load_many..) isinstancerr&r'rall)r/keysrr?r load_many~s zDataLoader.load_manycCs||}|j|d|S)zu Clears the value at `key` from the cache, if it exists. Returns itself for method chaining. N)r+r,pop)r/r5r6rrrclears zDataLoader.clearcCs|j|S)z Clears the entire cache. To be used when some event results in unknown invalidations across this particular `DataLoader`. Returns itself for method chaining. )r,rFr?rrr clear_alls zDataLoader.clear_allcCsB||}||jvrt|trt|}nt|}||j|<|S)z Adds the provied key and value to the cache. If the key already exists, no change is made. Returns itself for method chaining. )r+r,rA Exceptionrr;r:)r/r5valuer6r8rrrprimes      zDataLoader.prime)NNNNNNNr) __name__ __module__ __qualname__r(r)r*r2r9r4rDrFrGrJrrrrr!%s$ % r!cs>ttds tdt_stfdd}tj|dS)Nresolved_promisecstdSr)rinvoke)vfnr1rron_promise_resolvesz4enqueue_post_promise_job..on_promise_resolve)hasattrr*rr:rNr then)rRr1rSrrQrr=s  r=cCsR|j}g|_|j}|r"|t|kr"t||}|D]}t||qdSt||dS)zh Given the current state of a Loader instance, perform a batch load from its current queue. N)r-r)rrdispatch_queue_batch)loaderqueuer)chunkschunkrrrr>s  r>c sddDz|}Wnty'}z t||WYd}~dSd}~ww|r/t|ts.zDataLoader must be constructed with a function which accepts Array and returns Promise>, but the function did not return a Promise: {}.cspt|ts td|t|tkrtd|t|D]\}}t|tr0||q!||q!dS)NzDataLoader must be constructed with a function which accepts Array and returns Promise>, but the function did not return a Promise of an Array: {}.zDataLoader must be constructed with a function which accepts Array and returns Promise>, but the function did not return a Promise of an Array of the same length as the Array of keys. Keys: {} Values: {}) rArr&r'rziprHr;r:)valuesr[rIrCrXrrbatch_promise_resolveds    z4dispatch_queue_batch..batch_promise_resolved) r$rHfailed_dispatchrArr&r'rUcatchr)rWrX batch_promiseer_rr^rrVs,   rVcCs$|D] }||j||qdS)z Do not cache individual loads if the entire batch dispatch fails, but still reject each request so they do not hang. N)rFr5r;)rWrXerrorr[rrrr`>s  r`N)r) collectionsrcollections.abcr ImportError functoolsr threadingrr8rrr typingr r r r rrrrrrr r!r*r=r>rVr`rrrrs&      3 @