s,gvddlmZmZmZddlmZmZmZddlmZddl m Z er ddl m Z ddl mZGdd eZy ) )datetime timedeltatimezone) TYPE_CHECKINGListOptional)Redis)now)Queue) BaseWorkerceZdZdedefdZededefdZdedefdZdede fdZ dede e fd Z dede fd Zdeefd Zdedd fd ZddZy )IntermediateQueue queue_key connectioncL||_|j||_||_y)N)rget_intermediate_queue_keykeyr)selfrrs o/var/www/trellinator.diamondhoofcare.com/public_html/venv/lib/python3.12/site-packages/rq/intermediate_queue.py__init__zIntermediateQueue.__init__s""229=$returnc |dS)zReturns the intermediate queue key for a given queue key. Args: key (str): The queue key Returns: str: The intermediate queue key z :intermediate)clsrs rrz,IntermediateQueue.get_intermediate_queue_keysM**rjob_idc$|jd|S)zReturns the first seen key for a given job ID. Args: job_id (str): The job ID Returns: str: The first seen key z :first_seen:)rrrs rget_first_seen_keyz$IntermediateQueue.get_first_seen_keys((<x00rct|jj|j|t j ddS)zSets the first seen timestamp for a job. Args: job_id (str): The job ID timestamp (float): The timestamp TiQ)nxex)boolrsetr r timestamprs rset_first_seenz IntermediateQueue.set_first_seen*s=DOO''(?(?(GIZ_chq'rssrc|jj|j|}|r.tjt |t jSy)zReturns the first seen timestamp for a job. Args: job_id (str): The job ID Returns: Optional[datetime]: The timestamp )tzN)rgetr r fromtimestampfloatrutc)rrr&s rget_first_seenz IntermediateQueue.get_first_seen4sEOO''(?(?(GH ))% *:x||L Lrc^|j|}|syt|z tdkDS)aReturns whether a job should be cleaned up. A job in intermediate queue should be cleaned up if it has been there for more than 1 minute. Args: job_id (str): The job ID Returns: bool: Whether the job should be cleaned up Fr )minutes)r.r r)rr first_seens rshould_be_cleaned_upz&IntermediateQueue.should_be_cleaned_upBs1((0 uz!Ia$888rc|jj|jddDcgc]}|jc}Scc}w)zlReturns the job IDs in the intermediate queue. Returns: List[str]: The job IDs r)rlrangerdecoders r get_job_idszIntermediateQueue.get_job_idsRs6 /3oo.D.DTXXqRT.UVF VVVsANcR|jj|jd|y)zgRemoves a job from the intermediate queue. Args: job_id (str): The job ID r N)rlremrrs rremovezIntermediateQueue.removeZs TXXq&1rc,|j}|D]}|j|}||jvs#|s|j|7|j |rI|j |s[|j ||d|j|y)Nz$Job was stuck in intermediate queue.) exc_string)r7 fetch_jobstarted_job_registryr:r'r2handle_job_failure)rworkerqueuejob_idsrjobs rcleanupzIntermediateQueue.cleanupbs""$ (F//&)CU777KK'&&v.,,V4--c5Ek-lKK'! (r)r@r rAr rN)__name__ __module__ __qualname__strr r classmethodrr r$r'rrr.r2rr7r:rDrrrrr s%#%5%  +3 +3 + + 1 1 1tStTt S Xh-? 93949 WT#YW2S2T2(rrN)rrrtypingrrrredisr rq.utilsr rAr r@r objectrrrrrNs.2200"h(h(r