AHj*ddlmZddlZddlmZmZddlmZmZddlm Z ddl m Z ddl m Z dd lmZdd lmZdd lmZdd lmZmZd ZGddZy)) annotationsN) b64decode b64encode)datetimetimezone)Enum)Any)Redis)#UNSERIALIZABLE_RETURN_VALUE_PAYLOAD)Job)resolve_serializer)decode_redis_hashnowc d|SNz rq:results:)job_ids B/root/tools/cai/cai_env/lib/python3.12/site-packages/rq/results.pyget_keyrs  !!ceZdZGddeZ d ddZdZdZdZe d ddZ e d dd Z e d dd Z e d dd Z e ddd Ze dd Ze ddZe dddZe dd dZe d!d"dZe dZddZd#dZy)$Resultc eZdZdZdZdZdZdZy) Result.Typer N)__name__ __module__ __qualname__ SUCCESSFULFAILEDSTOPPEDRETRIEDMAX_RETRIES_EXCEEDEDrrrTypers  rr(Nc ||_||_||_|r|n t|_t | |_||_||_||_ ||_ | |_ | |_ | |_ yN) return_value exc_stringtyper created_atr serializer connectionrid worker_name execution_idexecution_started_atexecution_ended_at) selfrr-r0r1r.r+r,r2r/r3r4r5s r__init__zResult.__init__sh)$ (2*,Z8$ &($8!"4rcnd|jd|j|jjdS)Nz Result(id=z, type=))r1r(r-namer6s r__repr__zResult.__repr__9s/DGG9GDIIdii,@,E,E+FaHHrcT |j|jk(S#t$rYywxYw)NF)r1AttributeError)r6others r__eq__z Result.__eq__<s, 77ehh& &  s  ''c,t|jSr*)boolr1r;s r__bool__zResult.__bool__BsDGG}rc ||j||j||||j|| |  } | j||| S)N) rr-r0r+r,r2r/r3r4r5ttlpipeline)r1r0r/save) clsjobr-rFr+r,r2rGr3r4r5results rcreatez Result.createEsO66~~%!#~~%!51    h / rc ||j|jj|j|||j||| } | j ||| S)N) rr-r0r,r2r/r3r4r5rE)r1r(r$r0r/rH) rIrJrFr,r2rGr3r4r5rKs rcreate_failurezResult.create_failurebsT66~~!#~~%!51    h / rc ^|j||jj||||||| SN)r+r2rGr3r4r5)rLr(r& rIrJrFr+r2r3r4r5rGs rcreate_retriedzResult.create_retried|s?zz  HH   %#%!51  rc ^|j||jj||||||| SrP)rLr(r'rQs rcreate_max_retries_exceededz"Result.create_max_retries_exceededs?zz  HH ) ) %#%!51  rc |jj|j|jdd}g}|D]L\}}|j |j |j|j ||j|N|S)zReturns all results for job+-r0r/)r0 xrevrangerr1appendrestoredecode)rIrJr/responseresults result_idpayloads rallz Result.alls}>>++CKK,?cJ"*  Iw NN CFFI$4$4$6CNNgq r   rcj|jj|j|jS)z!Returns the number of job results)r0xlenrr1rIrJs rcountz Result.counts&~~""3;;svv#677rcl|jj|j|jy)zDelete all job resultsN)r0deleterr1rds r delete_allzResult.delete_alls# ckk#&&12rc tjt|jdddz tj }t |}t|}|jd}|(|jt|j}|jd}|r,tjt|j}|jdr |jddjnd } |jd } | r| j} |jd } | r.tjt| tj } |jd } | r.tjt| tj } t|tj!t|d |||||| | | |  S)z/Create a Result object from given Redis payloadrWr)tzr+r,r2rr3r4r5r-) r0r1r.r+r,r2r3r4r5)r fromtimestampintsplitrutcrrgetloadsrr\zlib decompressfloatrr() rIrr_r`r0r/r.r+r,r2r3r4r5s rr[zResult.restores++C 0DQ0G,H4,OT\T`T`a #G, ( 3 {{>2  #%++Il6I6I6K,LML[[. :)>?FFHJBI++mB\gkk-5<<>bd {{>2 '..0L&{{+AB #+#9#9%@T:UZbZfZf#g $[[)=> !)!7!7>P8QV^VbVb!c   KKGFO, -!!%!#%!51  rcy)afFetch a result that matches a given job ID. The current sorted set based implementation does not allow us to fetch a given key by ID so we need to iterate through results, deserialize the payload and look for a matching ID. Future Redis streams based implementation may make this more efficient and scalable. Nr)rIrJr/s rfetchz Result.fetchsrc|rS|dz}|jj|j|jdi|}|sy|d}|d}|d\}}nC|jj |j|jdd d }|sy|d\}}|j |j|j ||j| }|S) zReturns the latest result for given job instance or ID. If a non-zero timeout is provided, block for a result until timeout is reached. rjz0-0)blockNrr rVrW)rerX)r0xreadrr1rYr[r\) rIrJr/timeout timeout_msr]r_r`ress r fetch_latestzResult.fetch_latests !4J~~++S[[-@%,HPZ+[H{H{H!)" Iw~~// CFF0CS#UV/WH!)! Iwkk#&&)"2"2"4g#..eokp rc d|Srr)rIrs rrzResult.get_keysVH%%rcN|j|j}||n |j}|j||j d}||j |_|4|dk(r|j||j S|j|||j S)zSave result data to Redis )maxlenrz) rrr0xadd serializer\r1persistexpire)r6rFrGkeyr0rKs rrHz Result.savesll4;;'!)!5X4?? dnn&6rB  mmoDG ?by""3'ww!!#s+wwrc|jj|jd}|jGt t j |jjj|d< |jj|j}|jt |j|d<|jr|j|d<|jr|jj|d<|j r|j j|d<|S#|jjt}YxYw)N)r-r2r,r+r3r4r5)r-valuer2r,rrscompressencoder\r/dumpsr+r r3r4 timestampr5)r6data serializeds rrzResult.serialize-s(, HXHXY ?? &!*4==9O9O9Q+R!S!Z!Z!\D  T..t/@/@AJ    (#,Z#8#?#?#AD   #'#4#4D  $ $+/+D+D+N+N+PD' (  " ")-)@)@)J)J)LD% & ! T../RSJs 8%D44!E) NNNNrlNNNN)rstrr-r(r0r r1 str | Noner.datetime | Noner+z Any | Noner,rr2rr3rr4rr5r)NNrlNNNN)r3rr4rr5rreturnr)rlNNNNr*)r3rr4rr5rrr)rJr )rJr rrn)rJr rNone) rrr_rr`dictr0r rr)rJr r Result | None)Nr)rJr r|rnrr)rzdict[str, Any])r r!r"rr(r7r<r@rC classmethodrLrNrRrTrarerhr[rwrrrHrrrrrrse!t!&*#'!%#'04.2555 5  5 $ 5!555!5.5,58I  #'04.2!., 8 #'04.2!., 2   ' %    .   ' %    .  8833+ + Z  4&&"rr) __future__rrsbase64rrrrenumrtypingr redisr defaultsr rJr serializersrutilsrrrrrrrrs7" ''9+)"ppr