$gK#ddlmZmZmZmZmZddlmZddlm Z ddl m Z ddl m Z eddZdZGd d eZd Zdd ZdZdZdZdS))gather ensure_futureget_event_loop iscoroutineiscoroutinefunction) namedtuple)Iterable)partial)ListLoaderz key,futurecXtt|tr|jn|SN)r isinstancer func)fns Y/var/www/html/netbox-4.1.3/venv/lib/python3.11/site-packages/graphene/utils/dataloader.pyiscoroutinefunctionorpartialrs% *R*A*AIrwwr J JJcleZdZdZdZdZ d dZedZd dZ dZ dZ dZ d Z d ZdS) DataLoaderTNcl||_|||_t|js"Jd|jt |js"t d||||_|||_|||_|pd|_ ||ni|_ g|_ dS)Nz-batch_load_fn must be coroutine. Received: {}zuDataLoader must be have a batch_load_fn which accepts Iterable and returns Future>, but got: {}.c|Sr)xs rz%DataLoader.__init__..?sr) _loop batch_load_fnrformatcallable TypeErrorbatchmax_batch_sizecache get_cache_key_cache_queue)selfrr!r"r#r$ cache_maploops r__init__zDataLoader.__init__s  $!.D +     V V : A A$BT U U V V V*++ V&''    DJ  %"0D   DJ*;{{#,#8iib  rcD|jst|_|jSr)rrr's rr)zDataLoader.loopDs!z *'))DJzrcB|"td|||}|jr|j|}|r|S|j}|jr ||j|<||||S)zZ Loads a key, returning a `Future` for the value represented by that key. NzDThe loader.load() function must be called with a value, but got: {}.) r rr$r#r%getr) create_futuredo_resolve_reject)r'key cache_key cached_resultfutures rloadzDataLoader.loadKs ;#&++  &&s++  : % KOOI66M %$$((** : ,%+DK " sF+++ rc|jt||t|jdkr/|jrt |j|dSt|dSdS)N)r1r4)r&appendr lenr!enqueue_post_future_jobr)dispatch_queue)r'r1r4s rr0zDataLoader.do_resolve_rejecths| 6c&999::: t{  q z %' 488888t$$$$$ ! rct|ts"td|t fd|DS)a% Loads multiple keys, returning a list of values >>> a, b = await my_loader.load_many([ 'a', 'b' ]) This is equivalent to the more verbose: >>> a, b = await gather( >>> my_loader.load('a'), >>> my_loader.load('b') >>> ) zNThe loader.load_many() function must be called with Iterable but got: {}.c:g|]}|Sr)r5).0r1r's r z(DataLoader.load_many..s#7773 #777r)rr r rr)r'keyss` r load_manyzDataLoader.load_manyvs[$)) #&,,  7777$77788rcf||}|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'r1r2s rclearzDataLoader.clears2 &&s++   4((( rc8|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%rDr,s r clear_allzDataLoader.clear_alls  rc||}||jvrc|j}t |t r||n||||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%r)r/r Exception set_exception set_result)r'r1valuer2r4s rprimezDataLoader.primes &&s++  DK ' 'Y,,..F%++ )$$U++++!!%(((%+DK " r)NNNNNNNr)__name__ __module__ __qualname__r!r"r#r*propertyr)r5r0rArDrFrLrrrrrs EN E ((((TX : % % %999.rrcXfd}|t|dS)Nc*KtdSr)r;)loadersrdispatchz)enqueue_post_future_job..dispatchsvr) call_soonr)r)rSrTs ` rr:r:s> NN=((**-----rr7c|tdfdtdtDS)Nr7c32K|]}||zVdSrr)r>i chunk_size iterable_objs r zget_chunks..sE    QZ'(      rr)maxranger9)rZrYs``r get_chunksr^sWQ ##J     q#l++Z88   rc|j}g|_|j}|rG|t|kr4t||}|D]}t t || dSt t ||dS)zh Given the current state of a Loader instance, perform a batch load from its current queue. N)r&r"r9r^rdispatch_queue_batch)rSqueuer"chunkschunks rr;r;s MEFM*N;.3u::55E>22 ? ?E .vu== > > > > ? ? *6599:::::rc Kd|D}||}|rt|s1t||td|S |d{V}t |t s"td|t|}t|t|kr#td||t||D]O\}}t |tr|j |5|j |PdS#t$r}t|||cYd}~Sd}~wwxYw)Ncg|] }|j Sr)r1)r>loadeds rr?z(dispatch_queue_batch..s + + +6FJ + + +rzDataLoader must be constructed with a function which accepts Iterable and returns Future>, but the function did not return a Coroutine: {}.zDataLoader must be constructed with a function which accepts Iterable and returns Future>, but the function did not return a Future of a Iterable: {}.zDataLoader must be constructed with a function which accepts Iterable and returns Future>, but the function did not return a Future of a Iterable with the same length as the Iterable of keys. Keys: {} Values: {})rrfailed_dispatchr rrr listr9ziprHr4rIrJ)rSrar@ batch_futurevaluesrfrKes rr`r`s + +U + + +D''--L   {<88     2&&&      !1#######&(++ =&..  f v;;#d)) # #& &v&&   !// 0 0MFE%++ 0 ++E2222 ((////  0 0 111vua000000001s'C1E E?$E:4E?:E?cx|D]6}||j|j|7dS)z Do not cache individual loads if the entire batch dispatch fails, but still reject each request so they do not hang. N)rDr1r4rI)rSraerrorrfs rrgrgsJ ++ VZ    ##E****++rN)r7)asynciorrrrr collectionsrcollections.abcr functoolsr typingr r robjectrr:r^r;r`rgrrrrusF#"""""$$$$$$ Hl + +KKK]]]]]]]]@...;;;*616161r+++++r