AHj!ddlmZddlZddlmZmZddlmZmZddlm Z ddl m Z er ddl m Z dd lmZdd lmZdd lmZmZdd lmZmZmZmZGd dZGddeZej8dZddZddZy)) annotationsN)datetimetimezone) TYPE_CHECKINGAny)uuid4)Redis)Pipeline) BaseWorker)Job) BaseRegistryStartedJobRegistry)as_textcurrent_timestampnowparse_composite_keyceZdZdZddZddZeddZeddZedZ e ddZ dZ e dd Z e dd Zddd Zdd ZddZddZy ) Executionz)Class to represent an execution of a job.cl||_||_||_t}||_||_d|_yN)idjob_id connectionr created_atlast_heartbeat_job)selfrrr right_nows E/root/tools/cai/cai_env/lib/python3.12/site-packages/rq/executions.py__init__zExecution.__init__s4 $E #' $ cVt|tsy|j|jk(S)NF) isinstancerr)rothers r __eq__zExecution.__eq__!s"%+ww%((""r"c d|jS)Nz rq:execution:) composite_keyrs r keyz Execution.key&st11233r"c|jr |jStj|j|j|_|jS)N)rr)rr fetchrrr)s r jobz Execution.job*s7 9999 III yyr"c8|jd|jS)N:)rrr)s r r(zExecution.composite_key1s++ay))r"c<||||}|j|S)zFetch an execution from Redis.rrr)refresh)clsrrr executions r r,zExecution.fetch5s$2fD r"c\|jj|j}|std|jdt j t|dtj|_ t j t|dtj|_ y)z"Refresh execution data from Redis.z Execution z not found in Rediss created_at)tzslast_heartbeatN) rhgetallr* ValueErrorrr fromtimestampfloatrutcrr)rdatas r r2zExecution.refresh<s&&txx0z$''2EFG G"00tM7J1KPXP\P\]&44U4@Q;R5SX`XdXder"c4t|\}}||||S)z>A combination of job_id and execution_id separated by a colon.r1)r)r3r(rr execution_ids r from_composite_keyzExecution.from_composite_keyDs" 3=A l6jIIr"c"tj}|||j|j}|j ||t |j|j ||||jj|||d|S)zSave execution data to Redis.r1)ttlpipelinerrr4rArBF)rBrAxx) rhexrrsaveExecutionRegistryaddstarted_job_registry add_execution)r3r-rArBrr4s r createzExecution.createJs|W[[2cffH 32H=AAI[^iqAr   ..y8QTY^._r"Nc||n |j}|j|j|j|j |j|y)z6Save execution data to Redis and JobExecutionRegistry.N)mapping)rhsetr* serializeexpire)rrArBrs r rGzExecution.saveTsA!)!5X4?? $..*:;$((C(r"c|j|j|jj||t |j |j j||y)zDelete an execution from Redis.)r4rBrCN)deleter*rJremove_executionrHrrremove)rr-rBs r rSzExecution.delete[sO!   11D81TIPP[_jrPsr"c|j|jj|jjdS)N)rrr)rr timestamprr)s r rPzExecution.serializeas5''//335"11;;=  r"c@t|_|j|jd|jj |j |j||j |||dt|j|j|||y)zUpdate execution heartbeat.rT)rArBrErCrDN) rrrOr*rWrQrKrHrrI)rrJrArBs r heartbeatzExecution.heartbeaths"e dhh 0$2E2E2O2O2QR#&**4S8PT*UBFFQU[^iqFrr")rstrrrZrr )r%objectreturnbool)r\rZ)r\r )rrZrrZrr r\r)r(rZrr r\r)r-r rAintrBr r\rr)rAr^rBzPipeline | Noner-r rBr )r\dict)rJrrAr^rBr )__name__ __module__ __qualname____doc__r!r&propertyr*r-r( classmethodr,r2r?rLrGrSrPrYr"r rrs3%# 44 ** fJJ )t  sr"rcReZdZdZdZd dZd d dZddZddZdddZ ddd Z dd Z y)rHzbClass to represent a registry of job executions. Each job has its own execution registry. zrq:executions:{0}c`||_||_|jj||_yr)rr key_templateformatr*)rrrs r r!zExecutionRegistry.__init__ys($ $$++F3r"Ncn||n t}|jj|jd|y)zRemove expired jobs from registry. Removes jobs with an expiry time earlier than timestamp, specified as seconds since the Unix epoch. timestamp defaults to call time if unspecified. Nr)rrzremrangebyscorer*)rrWexception_handlersscores r cleanupzExecutionRegistry.cleanup~s.'2 8I8K ((1es r get_executionsz ExecutionRegistry.get_executionssX..uc:  ) pL   ioodkk^b^m^mon o pr"c|j}|D]}|j|||j|jy)zDelete the registry.)rBr-N)rrSr*)rr-rBrr4s r rSzExecutionRegistry.deletesB((* # 9I   hC  8 9!r")rrZrr )NN)rWz float | Nonernz list | None)r4rrAr^rBr r\r)r4rrBr r\r)r)ryr^rzr^r\z list[str])ryr^rzr^r\zlist[Execution]r_) rarbrcrdrjr!rprIrUr{rrSrgr"r rHrHrs2'L4 ="5\ "r"rHz rq.workercPddlm}|jj5}|j |}t j ||||_|j|j||jddd|jS#1swY|jSxYw)azPrepares execution for a job. This is called by the main Worker (not the horse) as it prepares for execution. Do not confuse this with worker.prepare_job_execution() which is called by the horse. Args: worker: The worker preparing the execution job: The job to prepare execution for Returns: Execution: The created Execution object r ) WorkerStatusrBN) worker.baserrrBget_heartbeat_ttlrrLr4 set_stateBUSYexecute)workerr-rrB heartbeat_ttls r prepare_executionrs*    # # %005 $++C+R**X>        s ABB%ctjd|j|jd||j%|jj ||d|_yy)a@Cleans up the execution of a job. It will remove the job execution record from the StartedJobRegistry and delete the Execution object. Args: worker: The worker to clean up execution for job: The job whose execution is being cleaned up pipeline: Redis pipeline to use for the cleanup zCleaning up execution of job %sNr)r-rB)loggerdebugrset_current_job_idr4rS)rr-rBs r cleanup_executionrs\ LL2CFF; dX6 #C(;$r")rr r-r r\r)rr r-r rBr r\None) __future__rloggingrrtypingrruuidrredisr redis.clientr rr r-r registryrrutilsrrrrrrH getLoggerrrrrgr"r rsf"'%%'6GGZsZsz>" >"B   ; '.  r"