]jzddlZddlmZmZddlmZmZddlmZddlm Z m Z ddl m Z ddl mZdd lmZdd lmZdd lmZmZd ZGd dZy)N) b64decode b64encode)datetimetimezone)Enum)AnyOptional)Redis)#UNSERIALIZABLE_RETURN_VALUE_PAYLOAD)Job)resolve_serializer)decode_redis_hashnowc d|zSNz rq:results:%s)job_ids 4/root/env/lib/python3.12/site-packages/rq/results.pyget_keyrs V ##ceZdZGddeZ d!dedededeedeed ee d eed efd Z d Z dZ dZ ed"d#dZed$d#dZed$d#dZed%defdZededefdZededdfdZed%dededededdf dZed%dededfdZed&dedededfdZedZd%dZdeee ffd Zy)'ResultceZdZdZdZdZdZy) Result.Typer N)__name__ __module__ __qualname__ SUCCESSFULFAILEDSTOPPEDRETRIEDrrrTypers rr&Nrtype connectionid created_at return_value exc_string worker_namec ||_||_||_|r|n t|_t | |_||_||_||_ ||_ yN) r+r,r'rr*r serializerr(rr)r-) selfrr'r(r)r*r+r,r-r0s r__init__zResult.__init__sP)$ (2*,Z8$ &rcnd|jd|j|jjdS)Nz Result(id=z, type=))r)r&r'namer1s r__repr__zResult.__repr__0s/DGG9GDIIdii,@,E,E+FaHHrcT |j|jk(S#t$rYywxYw)NF)r)AttributeError)r1others r__eq__z Result.__eq__3s, 77ehh& &  s  ''c,t|jSr/)boolr)r6s r__bool__zResult.__bool__9sDGG}rreturnc ||j||j||||j}|j|||S)N)rr'r(r+r,r-r0ttlpipeline)r)r(r0save) clsjobr'rBr+r,r-rCresults rcreatez Result.create<sD66~~%!#~~   h / rc||j|jj|j|||j}|j |||S)N)rr'r(r,r-r0rA)r)r&r#r(r0rD)rErFrBr,r-rCrGs rcreate_failurezResult.create_failureJsI66~~!#~~    h / rc||j|jj|j||j}|j |||S)N)rr'r(r-r0rA)r)r&r%r(r0rD)rErFrBr-rCrGs rcreate_retriedzResult.create_retriedWsH66!!~~#~~    h / rrFc |jj|j|jdd}g}|D]L\}}|j |j |j|j ||j|N|S)zReturns all results for job+-r(r0)r( xrevrangerr)appendrestoredecode)rErFr0responseresults result_idpayloads rallz Result.allcs}>>++CKK,?cJ"*  Iw NN CFFI$4$4$6CNNgq r   rcj|jj|j|jS)z!Returns the number of job results)r(xlenrr)rErFs rcountz Result.countps&~~""3;;svv#677rcl|jj|j|jy)zDelete all job resultsN)r(deleterr)r\s r delete_allzResult.delete_allus# ckk#&&12rrWrXc rtjt|jdddz tj }t |}t|}|jd}|(|jt|j}|jd}|r,tjt|j}|jdr |jddjnd } t|tjt|d |||||| S) z/Create a Result object from given Redis payloadrOr)tzr+r,r-rr')r(r)r*r+r,r-)r fromtimestampintsplitrutcrrgetloadsrrTzlib decompressrr&) rErrWrXr(r0r*r+r,r-s rrSzResult.restorezs++C 0DQ0G,H4,OT\T`T`a #G, ( 3 {{>2  #%++Il6I6I6K,LML[[. :)>?FFHJBI++mB\gkk-5<<>bd   KKGFO, -!!%!#  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)rErFr0s rfetchz Result.fetchsrtimeoutc|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. rbz0-0)blockNrr rNrO)r]rP)r(xreadrr)rQrSrT) rErFr0ro timeout_msrUrWrXress r fetch_latestzResult.fetch_latests !4J~~++S[[-@%,HPZ+[H{H{H!)" Iw~~// CFF0CS#UV/WH!)! Iwkk#&&)"2"2"4g#..eokp rc d|zSrr)rErs rrzResult.get_keys ''rcN|j|j}||n |j}|j||j d}||j |_|4|dk(r|j||j S|j|||j S)zSave result data to Redis )maxlenrr) rrr(xadd serializerTr)persistexpire)r1rBrCkeyr(rGs rrDz Result.savesll4;;'!)!5X4?? dnn&6rB  mmoDG ?by""3'ww!!#s+wwrc|jj|jd}|jGt t j |jjj|d< |jj|j}|jt |j|d<|S#|jjt}YMxYw)N)r'r-r,r+) r'valuer-r,rrkcompressencoderTr0dumpsr+r )r1data serializeds rr|zResult.serializes(, HXHXY ?? &!*4==9O9O9Q+R!S!Z!Z!\D  T..t/@/@AJ    (#,Z#8#?#?#AD   T../RSJs 8%C!C*)NNNNrdN)NNrdN)r?r)rdNr/)Nr)rr r!rr&strr r rrr2r7r;r> classmethodrHrJrLr rYrfr]r`dictrSrnrvrrDr|rrrrrst!)-&*$(''' ' SM ' X& 'sm'SM'',I        c  88883S3T33 S S 4 U hp  <  (1C  sc(S[J\4(("4S>rr)rkbase64rrrrenumrtypingrr redisr defaultsr rFr serializersrutilsrrrrrrrrs4 '' 9+)$QQr