o /i@sddlmZmZddlmZmZmZddlmZddlm Z er&ddl m Z ddl m Z ddlmZmZdd lmZmZmZmZGd d d ZGd d d eZdS))datetimetimezone) TYPE_CHECKINGAnyOptional)uuid4)Redis)Pipeline)Job) BaseRegistryStartedJobRegistry)as_textcurrent_timestampnowparse_composite_keyc @seZdZdZdededefddZdedefd d Z e defd d Z e de fd dZ e ddZededededdfddZddZedededdfddZede deddddfddZd(dededfdd Zde ddfd!d"Zdefd#d$Zd%ededdfd&d'ZdS)) Executionz)Class to represent an execution of a job.idjob_id connectioncCs.||_||_||_t}||_||_d|_dSN)rrrr created_atlast_heartbeat_job)selfrrr right_nowrM/var/www/html/flask_server/venv/lib/python3.10/site-packages/rq/executions.py__init__s zExecution.__init__otherreturncCst|tsdS|j|jkS)NF) isinstancerr)rrrrr__eq__s  zExecution.__eq__cCs d|jS)Nz rq:execution:) composite_keyrrrrkey!s z Execution.keycCs&|jr|jStj|j|jd|_|jS)N)rr)rr fetchrrr$rrrjob%sz Execution.jobcCs|jd|jS)N:)rrr$rrrr#,szExecution.composite_keycCs||||d}||S)zFetch an execution from Redis.rrr)refresh)clsrrr executionrrrr&0szExecution.fetchcCs\|j|j}|std|jdtjt|dtj d|_ tjt|dtj d|_ dS)z"Refresh execution data from Redis.z Execution z not found in Rediss created_at)tzslast_heartbeatN) rhgetallr% ValueErrorrr fromtimestampfloatrutcrr)rdatarrrr*7s zExecution.refreshr#cCst|\}}||||dS)z>A combination of job_id and execution_id separated by a colon.r))r)r+r#rr execution_idrrrfrom_composite_key?s zExecution.from_composite_keyr'ttlpipeliner cCsZtj}|||j|jd}|j||dt|j|dj|||d|jj|||dd|S)zSave execution data to Redis.r))r6r7rrr,r6r7F)r7r6xx) rhexrrsaveExecutionRegistryaddstarted_job_registry add_execution)r+r'r6r7rr,rrrcreateEs zExecution.createNcCs8|dur|n|j}|j|j|d||j|dS)z6Save execution data to Redis and JobExecutionRegistry.N)mapping)rhsetr% serializeexpire)rr6r7rrrrr<OszExecution.savecCs:||j|jj||dt|j|jdj||ddS)zDelete an execution from Redis.)r,r7r8N)deleter%r?remove_executionr=rrremove)rr'r7rrrrFVs zExecution.deletecCs|j|j|jdS)N)rrr)rr timestamprr$rrrrD\szExecution.serializer?cCs\t|_||jd|j||j||j|||ddt|j|dj |||ddS)zUpdate execution heartbeat.rT)r6r7r:r8r9N) rrrCr%rIrEr@r=rr>)rr?r6r7rrr heartbeatcs zExecution.heartbeatr)__name__ __module__ __qualname____doc__strrrobjectboolr"propertyr%r r'r# classmethodr&r*r5intrArr<rFdictrDr rJrrrrrs*   rc@seZdZdZdZdedefddZd dee d ee fd d Z d e d e dddefddZd e dddefddZd!de de de efddZd!de de de e fddZdeddfddZdS)"r=zbClass to represent a registry of job executions. Each job has its own execution registry. zrq:executions:{0}rrcCs||_||_|j||_dSr)rr key_templateformatr%)rrrrrrrtszExecutionRegistry.__init__NrIexception_handlerscCs(|dur|nt}|j|jd|dS)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%)rrIrXscorerrrcleanupyszExecutionRegistry.cleanupr,r6r7r r cCs8t|}||j|j|di||j|ddS)aRegister an execution to registry with expiry time of now + ttl, unless it's -1 which is set to +inf Args: execution (Execution): The Execution to add ttl (int, optional): The time to live. Defaults to 0. pipeline (Optional[Pipeline], optional): The Redis Pipeline. Defaults to None. Returns: result (int): The ZADD command result <N)rzaddr%rrE)rr,r6r7rZrrrr>s zExecutionRegistry.addcCs||j|jS)z"Remove an execution from registry.)zremr%r)rr,r7rrrrHszExecutionRegistry.removerstartendcCs$|dd|j|j||DS)&Returns all executions IDs in registrycSsg|]}t|qSr)r).0rrrr sz7ExecutionRegistry.get_execution_ids..)r[rzranger%)rr`rarrrget_execution_idssz#ExecutionRegistry.get_execution_idscCs8|||}g}|D]}|tj||j|jdq |S)rbr))rfappendrr&rr)rr`ra execution_ids executionsr4rrrget_executionss z ExecutionRegistry.get_executionsr'cCs0|}|D] }|j||dq||jdS)zDelete the registry.)r7r'N)rjrFr%)rr'r7rir,rrrrFszExecutionRegistry.delete)NN)rr_)rKrLrMrNrVrOrrrr1listr[rrTrr>rHrfrjr rFrrrrr=ms  r=N)rrtypingrrruuidrredisr redis.clientr r'r registryr r utilsrrrrrr=rrrrs    ]