o iv@sddlZddlmZmZddlmZmZddlmZddlm Z m Z ddl m Z ddl mZdd lmZdd lmZdd lmZmZd d ZGdddZdS)N) b64decode b64encode)datetimetimezone)Enum)AnyOptional)Redis)#UNSERIALIZABLE_RETURN_VALUE_PAYLOAD)Job)resolve_serializer)decode_redis_hashnowcCs d|SNz rq:results:)job_idrrA/home/app/Keep/.python/lib/python3.10/site-packages/rq/results.pyget_key rc@sxeZdZGdddeZ      d3dedededeed eed ee d eed efd dZ ddZ ddZ ddZ ed4d5ddZed6d5ddZed6d5ddZed7defddZededefdd Zededdfd!d"Zed7ded#ed$ededdf d%d&Zed7dededfd'd(Zed8ded*ededfd+d,Zed-d.Zd7d/d0Zdeee ffd1d2ZdS)9Resultc@seZdZdZdZdZdZdS)z Result.Typer N)__name__ __module__ __qualname__Z SUCCESSFULFAILEDZSTOPPEDRETRIEDrrrrTypes rNrtype connectionid created_at return_value exc_string worker_namec CsH||_||_||_|r |nt|_t| |_||_||_||_ ||_ dSN) r%r&r!rr$r serializerr"rr#r') selfrr!r"r#r$r%r&r'r)rrr__init__s   zResult.__init__cCsd|jd||jjdS)Nz Result(id=z, type=))r#rr!namer*rrr__repr__0szResult.__repr__cCs$z|j|jkWStyYdSw)NF)r#AttributeError)r*otherrrr__eq__3s  z Result.__eq__cCs t|jSr()boolr#r.rrr__bool__9rzResult.__bool__returnc Cs.||j||j||||jd}|j||d|S)N)rr!r"r%r&r'r)ttlpipeline)r#r"r)save) clsjobr!r7r%r&r'r8resultrrrcreate<s z Result.createcCs0||j|jj|j|||jd}|j||d|S)N)rr!r"r&r'r)r6)r#rrr"r)r9)r:r;r7r&r'r8r<rrrcreate_failureJszResult.create_failurecCs.||j|jj|j||jd}|j||d|S)N)rr!r"r'r)r6)r#rrr"r)r9)r:r;r7r'r8r<rrrcreate_retriedWszResult.create_retriedr;c CsP|j||jdd}g}|D]\}}||j|j|||j|dq|S)zReturns all results for job+-r"r))r" xrevrangerr#appendrestoredecode)r:r;r)responseresults result_idpayloadrrrallcs z Result.allcCs|j||jS)z!Returns the number of job results)r"xlenrr#r:r;rrrcountpsz Result.countcCs|j||jdS)zDelete all job resultsN)r"deleterr#rMrrr delete_alluszResult.delete_allrIrJc Cstjt|dddtjd}t|}t|}|d}|dur+| t | }|d}|r;t t | }|drH|dd  nd } t|tt|d |||||| d S) z/Create a Result object from given Redis payloadrAr)tzr%Nr&r'r r!)r"r#r$r%r&r')r fromtimestampintsplitrutcrr getloadsrrFzlib decompressrr) r:rrIrJr"r)r$r%r&r'rrrrEzs("  zResult.restorecCsdS)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)r:r;r)rrrfetchs z Result.fetchrtimeoutc Cs|r'|d}|jj||jdi|d}|sdS|d}|d}|d\}}n|jj||jdd dd }|s9dS|d\}}|j|j|||j|d }|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. rQz0-0)blockNrr r@rA)rNrB)r"Zxreadrr#rCrErF) r:r;r)r]Z timeout_msrGrIrJresrrr fetch_latests zResult.fetch_latestcCs d|Srr)r:rrrrrs zResult.get_keycCsv||j}|dur |n|j}|j||dd}|dur"||_|dur8|dkr2|||jS||||jS)zSave result data to RedisN )maxlenr_) rrr"Zxadd serializerFr#ZpersistZexpire)r*r7r8keyr"r<rrrr9s    z Result.savecCs||jj|jd}|jdurtt|j|d<z |j |j }Wn |j t }Y|j durr?r rKrUrNrPdictrEr\rarr9rdrrrrrsd      "   r)rZbase64rrrrenumrtypingrrZredisr defaultsr r;r Z serializersr utilsrrrrrrrrs