AHjm.UddlmZddlZddlZddlZddlZddlZddlZddlm Z ddl m Z ddl m Z ddl mZmZddlmZddlmZdd lmZmZdd lmZd d lmZd d lmZmZmZd dlm Z d dl!m"Z"d dl#m$Z$d dl%m&Z&d dl'm(Z(d dl)m*Z*d dl+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1de2d< edjZ3dZ5dZ6Gdde7e Z8GddZ9dZ:y#e4$reZ3Y'wxYw)) annotationsN)Iterable)datetime)Enum)Process get_context) BaseProcess)uuid4)ConnectionPoolRedis)Pipeline)parse_connection)DEFAULT_LOGGING_DATE_FORMATDEFAULT_LOGGING_FORMAT!DEFAULT_SCHEDULER_FALLBACK_PERIOD)SchedulerNotFound)Job)setup_loghandlers)Queue)ScheduledJobRegistry)resolve_serializer)current_timestampdecode_redis_hashnow parse_names utcformatutcparseztype[BaseProcess] ForkProcessforkzrq:scheduler:%szrq:scheduler-lock:%sceZdZdZdZdZy)SchedulerStatusstartedworkingstoppedN)__name__ __module__ __qualname__STARTEDWORKINGSTOPPEDD/root/tools/cai/cai_env/lib/python3.12/site-packages/rq/scheduler.pyr"r"'sGGGr-r"c(eZdZeZdej eeddf ddZ e dZ e dZ e dZ e ddZe dZdd Zddd Zedd Zd d Zd!d Zdd"dZd#dZd$dZed%dZdZdZd&dZdZdZdZdZ dZ!y)' RQSchedulerrNc Btt||_t|_g|_d|_t |\|_|_|_ t||_ |xstj|_tj |_t%|_d|_d|_d|_||_d|_|j2j4|_d|_t;j<t>|_ tC|t>||y)NrF)levelname log_format date_format)"setr _queue_names_acquired_locks_scheduled_job_registrieslock_acquisition_timer_connection_class _pool_class _pool_kwargsr serializerr hexr3socket gethostnamehostnamer created_atpidlast_heartbeat _connectioninterval_stop_requestedStatusr+_status_processlogging getLoggerr&logr) selfqueues connectionrG logging_levelr5r4r>r3s r.__init__zRQScheduler.__init__4s F 34),EG&%)"FVWaFbC 0$2C,Z8, #//1 $'E/3  ${{**  $$X.!#  r-c|jr |jS|jtdd|ji|j|_|jS)Nconnection_class)connection_poolr,)rFr;r r<r=rOs r.rQzRQScheduler.connectionZsZ   ## #11*bDE $$T* + *,&#3399:JK%-\\^"   J== (>(>(@ r-cg|_|s |j}|D]=}|jjt||j|j ?y)z(Prepare scheduled job registries for userQr>N)r9r8appendrrQr>)rO queue_namesr3s r.prepare_registrieszRQScheduler.prepare_registriessR)+&..K D  * * 1 1$TdooRVRaRab  r-ct|zS)z,Returns scheduler key for a given queue name)SCHEDULER_LOCKING_KEY_TEMPLATE)clsr3s r.rjzRQScheduler.get_locking_keys.44r-c|jJ|j|jt|jdj |j t|jt|jdS)zBSerialize this scheduler's metadata for storage in its Redis hash.,)r3rBrDrPrCrE) rEr3rBstrrDrir7rrCrWs r.to_dictzRQScheduler.to_dictsc""...II txx=hht001#DOO4'(;(;<   r-cdt|d}|d|_|d|_t|d|_|j drt |djdn t |_t|d|_ |j d rt|d |_ y d |_ y ) z6Restore this scheduler's metadata from its Redis hash.T) decode_valuesr3rBrDrPr|rCrEN) rr3rBintrDgetr6splitr7rrCrE)rOraw_dataobjs r.restorezRQScheduler.restores=K J s5z?=@WWX=NCH 3 3C 89TWTY"3|#45ADIYAZhs+;'<=`dr-c||n|jj}|j|j|j |j |j|j dz||jyy)z/Save this scheduler's metadata hash with a TTL.N)mappingre)rQpipelinehsetr_r~expirerGexecute)rOrrQs r.savezRQScheduler.savesi!)!5X4??;S;S;U $,,.9$((DMMB$67       r-c|jjd|jtj|_t |_|jy)zRegister this scheduler's birth by writing its metadata hash. Idempotent: re-registering the same name (e.g. when the worker restarts a crashed scheduler process) overwrites the existing hash rather than erroring. zScheduler %s: registering birthN) rNrhr3osgetpidrDrrErrWs r.register_birthzRQScheduler.register_births; 8$))D99;!e r-c|jjd|jt|jj |j S)zRegister this scheduler's death by deleting its metadata hash. Returns: True if the scheduler metadata existed and was deleted, False if it was already absent. zScheduler %s: registering death)rNrhr3boolrQdeleter_rWs r.register_deathzRQScheduler.register_deaths9 8$))DDOO**488455r-c|jt|z}|std|d|g||}|j||S)zrr fetch_many _enqueue_jobshould_enqueue_at_frontremoverr)) rOregistry timestampjob_idsqueuerjobsjobjob_ids r.enqueue_scheduled_jobsz"RQScheduler.enqueue_scheduled_jobss?{{** --$2F2F  # # %66 #H)+I33I>G(--DOOPTP_P_`E))+ #x~~g$//VZVeVefkC**3CLgLgLi*jk&?FOOFXO>?  " # # #({{**  # #s74E ,AE  E* ctjtj|jtjtj|jy)zUInstalls signal handlers for handling SIGINT and SIGTERM gracefully. N)signalSIGINT request_stopSIGTERMrWs r._install_signal_handlersz$RQScheduler._install_signal_handlerss4  fmmT%6%67 fnnd&7&78r-cd|_y)z8Toggle self._stop_requested that's checked on every loopTN)rH)rOsignumframes r.rzRQScheduler.request_stops #r-c:|jjddj|jt |_|j j5}|j|jdt|j |j|j|jdz|jD]0}|j|j||jdz2|jdddy#1swYyxYw)z?Refresh the TTL on the scheduler's metadata hash and its locks.z!Scheduler sending heartbeat to %srdrEreN)rNrhrirZrrErQrrr_rrrGr8rjr)rOrr3s r. heartbeatzRQScheduler.heartbeats :DIIdFYFYrs" $0/'!)ll)'*+^^ f%--K+!7c4 D&D&NIiKsCCC