aj2#ddlmZmZmZmZmZddlmZddlm Z ddl m Z ddl m Z eddZdZGd d eZd Zdd Zd ZdZdZy))gather ensure_futureget_event_loop iscoroutineiscoroutinefunction) namedtuple)Iterable)partial)ListLoaderz key,futurecXtt|tr|jS|SN)r isinstancer func)fns f/var/www/html/myl_app/scheduler-service/venv/lib/python3.12/site-packages/graphene/utils/dataloader.pyiscoroutinefunctionorpartialrs" *R*Arww JJr JJcdeZdZdZdZdZ d dZedZd dZ dZ dZ dZ d Z d Zy) DataLoaderTNc^||_|||_t|js Jdj|jt |jst dj||||_|||_|||_|xsd|_ ||ni|_ g|_ y)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 : A A$BTBT U V **+V&'    DJ  %"0D   DJ*;{#,#8ib $& rcP|jst|_|jSr)rrr's rr)zDataLoader.loopCszz')DJzzrcD|tdj||j|}|jr|jj |}|r|S|j j}|jr||j|<|j|||S)zZ Loads a key, returning a `Future` for the value represented by that key. zDThe 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.loadJs ;#&+  &&s+  :: KKOOI6M$$((* ::%+DKK " sF+ rc|jjt||t|jdk(r/|jrt |j |yt|yy)N)r1r4)r&appendr lenr!enqueue_post_future_jobr)dispatch_queue)r'r1r4s rr0zDataLoader.do_resolve_rejectgsR 6c&9: t{{ q zz' 48t$ !rct|tstdj|t |Dcgc]}|j |c}Scc}w)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: {}.)rr r rrr5)r'keysr1s r load_manyzDataLoader.load_manyusL$)#&,  $73 #7887sAc`|j|}|jj|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.clears+ &&s+   4( rc:|jj|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%rAr,s r clear_allzDataLoader.clear_alls  rc|j|}||jvr\|jj}t |t r|j |n|j|||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.primeso &&s+  DKK 'YY,,.F%+$$U+!!%(%+DKK " r)NNNNNNNr)__name__ __module__ __qualname__r!r"r#r*propertyr)r5r0r>rArCrIrrrrrs` EN E ''R : %9.rrcFfd}|jt|y)Nc$Ktywr)r;)loadersrdispatchz)enqueue_post_future_job..dispatchsvs ) call_soonr)r)rPrQs ` rr:r:s NN=(*-rc^tdfdtdtDS)Nr7c3.K|] }||zywrr).0i chunk_size iterable_objs r zget_chunks..s$   QZ( sr)maxranger9)rXrWs``r get_chunksr\s/Q #J q#l+Z8 rc|j}g|_|j}|r7|t|kr)t||}|D]}t t ||yt t ||y)zh Given the current state of a Loader instance, perform a batch load from its current queue. N)r&r"r9r\rdispatch_queue_batch)rPqueuer"chunkschunks rr;r;sl MMEFM**N.3u:5E>2 ?E .vu= > ? *659:rc K|Dcgc]}|j}}|j|}|r t|s%t||t dj |S |d{}t |tst dj |t|}t|t|k7rt dj ||t||D]L\}}t |tr|jj|2|jj|Nycc}w7#t$r}t|||cYd}~Sd}~wwxYww)NzDataLoader 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: {})r1rrfailed_dispatchr rrr listr9ziprEr4rFrG)rPr_loadedr= batch_futurevaluesrHes rr^r^sK%* +6FJJ +D +''-L {<8   2&&    !1##&(+=&.  f v;#d) #& &v&  !/ 0MFE%+ ++E2 ((/  0[ ,($> 1vua001sSED-AE D4%D2&CD4,E2D44 E= E E EEEc~|D]8}|j|j|jj|:y)z Do not cache individual loads if the entire batch dispatch fails, but still reject each request so they do not hang. N)rAr1r4rF)rPr_errorrfs rrcrcs5 + VZZ  ##E*+rN)r7)asynciorrrrr collectionsrcollections.abcr functoolsr typingr r robjectrr:r\r;r^rcrrrrrsX#$ Hl +K\\~.;*61r+r