s,gBVddlZddlZddlZddlZddlmZmZmZddlmZm Z m Z m Z m Z m Z mZmZddlmZddlmZddlmZmZmZddlmZmZdd lmZdd lmZmZdd l m!Z!m"Z"m#Z#erdd l$m%Z%dd l&m'Z'ddl(m)Z)ejTdZ+GddZ,Gdde,Z-Gdde,Z.Gdde,Z/Gdde,Z0Gdde,Z1Gdde,Z2d"ddd e e3fd!Z4y)#N)datetime timedeltatimezone) TYPE_CHECKINGAnyCallableListOptionalTypeUnioncast)resolve_serializer)DEFAULT_FAILURE_TTL)AbandonedJobErrorInvalidJobOperationNoSuchJobError)Job JobStatus)Queue)BaseDeathPenaltyUnixSignalDeathPenalty)as_text backend_classcurrent_timestamp)Redis)Pipeline) Executionz rq.registryceZdZUdZeZeZdZe de fe d< d*de de dd e ed d e d d e de eef dZdZdZde defdZedefdZd+defdZd,dd de ddedefdZd-ded e fde ddefdZd.dd de dfdZd.de efd Zd/d!ed"ed#ededee f d$Z d%Z!dd de"fd&Z#d0d'ed e fd(edd fd)Z$y)1 BaseRegistryz Base implementation of a job registry, implemented in Redis sorted set. Each job is stored as a key in the registry, scored by expiration time (unix timestamp). zrq:registry:{0}.cleanupNname connectionr job_classrqueuer serializerdeath_penalty_classcL|r4|j|_|j|_|j|_n||_||_t||_|jj |j|_t|d||_t|d||_ y)Nr$)overrider') r"r#r&r key_templateformatkeyrr$r')selfr"r#r$r%r&r's e/var/www/trellinator.diamondhoofcare.com/public_html/venv/lib/python3.12/site-packages/rq/registry.py__init__zBaseRegistry.__init__'s  DI#..DO#..DODI(DO0r$r?zremr,rfetchr&delete)r-rHrIrTr#rBresult job_instances r.removezBaseRegistry.remove~s"*!5X4?? %c4>>:62 #t~~." "yyJSWSbSbc    ! r0c||n |j}|jDcgc]}|j}}|r|j|jg|Sycc}w)zRemoves job executions from registry Args: job (Job): The Job to remove from the registry pipeline (Optional[Pipeline], optional): The Redis Pipeline. Defaults to None. N)r#get_executions composite_keyrWr,)r-rHrIr# execution execution_idss r.remove_executionszBaseRegistry.remove_executionss^"*!5X4?? BEBTBTBVWY00W W ":??488r$r&rXr#rWr,r?rrIrorigin started_atended_at _exc_infosave _enqueue_jobexecute)r-rtrurHr&rZrIr%s r.requeuezBaseRegistry.requeues i 0CJJ..&&yT__Yc&dC%%dhh7% % __ % % ' 8#**DNNgqrE!CNCLCM HHJ$$S8h$OC        s ,A7D--D7)defaultNNNNN)T)rNF)NFr6)rrLFTF)%__name__ __module__ __qualname____doc__rr$rr'r*rr__annotations__rAr r rr/r4r:boolrCpropertyrOr2rErSr r\rbfloatrhr rnrprrsrr0r.r r s I0$L c3h (,+/#'@DllW%lDK( l  l  l&d+;&<=l, O O O0s00 /S /FuFx /CFPTFadF(%s +x 7K`d& =U =hz6J = +//S__E005O'6R))$((Au=  "W+ #ZwS*! ! + #Zs89J)2J3A2J)&E'J) J&"J)%J&&J))J3r`rrIrrJr<c|dkr|n t|z}|dk(rd}|j|j|j|i|S)aAdds an execution to a 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. xx (bool, optional): .... Defaults to False. Returns: result (int): The ZADD command result rrLrMrN)rrPr,r_)r-r`rIrQrJrRs r. add_executionz StartedJobRegistry.add_execution4sJQw$5$7#$= B;E}}TXX (?(?'GB}OOr0rHrrTcr||n |j}|j|j|j}|S)a}Removes job from registry and deletes it if `delete_job == True` Args: execution (Execution): The Execution to add job (Job): The Job to remove from the registry pipeline (Optional[Pipeline], optional): The Redis Pipeline. Defaults to None. delete_job (bool, optional): If should delete the job.. Defaults to False. )r#rWr,r_)r-r`rHrIrTr#rZs r.remove_executionz#StartedJobRegistry.remove_executionFs5"*!5X4?? 9+B+BC r0NN)rFr) rrrrr*r rlistr!rrOrrrr0r.rrsu L>%>XVZ^>@P{PjPUYPfiP$ + E Z ei r0rc(eZdZdZdZddeefdZy)FinishedJobRegistryz Registry of jobs that have been completed. Jobs are added to this registry after they have successfully completed for monitoring purposes. zrq:finished:{0}Nrccn||n t}|jj|jd|yzRemove 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. Nrrr#rr,r-rcrRs r.r!zFinishedJobRegistry.cleanup`.'2 8I8K ((1e>r0 chunk_sizer<c||n t}|jj|jd|d|}|Dcgc] }t |c}Scc}w)aGet's a list of job IDs that should be scheduled. Args: timestamp (Optional[int]): _description_. Defaults to None. chunk_size (int, optional): _description_. Defaults to 1000. Returns: jobs (List[str]): A list of Job ids r)rinumre)r-rcrrRjobs_to_schedulerBs r.get_jobs_to_schedulez)ScheduledJobRegistry.get_jobs_to_schedulesR#,"7Y=N=P??881eSTZd8e.>?F???sArtct|tr |j}n|}|jj |j |}|st tj|tjS)a:Returns datetime (UTC) at which job is scheduled to be enqueued Args: job_or_id (Union[Job, str]): The Job instance or Job ID Raises: NoSuchJobError: If the job was not found Returns: datetime (datetime): The scheduled time as datetime object )r) r>rr?r#r@r,rr fromtimestamprutc)r-rtrBrRs r.get_scheduled_timez'ScheduledJobRegistry.get_scheduled_timesV i %\\FF&&txx8 %%e ==r0r6r)Ni)rrrrr*r/r rr!rrrOr rArr rrrs@r.rrs&L< CE C*AU C ?Xh%7 ?(S]J^ ? @hsm @PS @_cdg_h @>E%*,=>(>r0rc*eZdZdZddeefdZdZy)CanceledJobRegistryzrq:canceled:{0}Nrcctr6)NotImplementedError)r-rcs r.rhz'CanceledJobRegistry.get_expired_job_idss!!r0cyrrr3s r.r!zCanceledJobRegistry.cleanup rr0r6)rrrr*r rrhr!rr0r.rrs$L"Xe_" r0rr%rrc0t|j|j|j|jj t |j|j|j|jj |t|j|j|j|jj t|j|j|j|jj y)zCleans StartedJobRegistry, FinishedJobRegistry and FailedJobRegistry, and DeferredJobRegistry of a queue. Args: queue (Queue): The queue to clean )r"r#r$r&)rN) rr"r#r$r&r!rrr)r%rs r.clean_registriesr's  ZZE$4$4\a\l\l gi ZZE$4$4\a\l\l g!3g4 ZZE$4$4\a\l\l gi ZZE$4$4\a\l\l gir0r6)5rloggingrrrrrtypingrrrr r r r r rq.serializersrdefaultsr exceptionsrrrrHrrr%rtimeoutsrrutilsrrrredisr redis.clientr rq.executionsr getLoggerrr rrrrrrrrrr0r.rs 22RRR-)NN><<%'   = )JJZkk`=,=&, ,^+L,+L\O><O>d  ,  G$r0