]jX HddlmZmZmZmZmZddlmZddlm Z ddl m Z ddl m Z ddlmZddlmZddlmZmZmZmZmZmZdd lmZdd lmZd d lmZmZm Z d d l!m"Z"d dl#m$Z$dZ%ddZ&dZ' ddeeeeeeefdeefdZ(deee)e)fdeefdZ*dZ+dZ,dZ-y))castOptionalListTupleUnion)ImproperlyConfigured) connections)SentinelConnectionPool)send_stop_job_command) Execution)Job)DeferredJobRegistryFailedJobRegistryFinishedJobRegistryScheduledJobRegistryStartedJobRegistryclean_registries)Worker)clean_worker_registry)get_connectionget_queue_by_index get_scheduler) QUEUES_LIST) to_localtimec" t|j}y#t$r\ddlm}|j j |j|j}|t|jcYSdcYSt$r }Yd}~yd}~wwxYw)aChecks whether there's a scheduler-lock on a particular queue, and returns the PID. It Only works with RQ's Built-in RQScheduler. When RQ-Scheduler is available returns False If not, it checks the RQ's RQScheduler for a scheduler lock in the desired queue Note: result might have some delay (1-15 minutes) but it helps visualizing whether the setup is working correctly Fr) RQSchedulerN) rnamer rq.schedulerr connectiongetget_locking_keyintdecode Exception)queue schedulerrpides 9/root/env/lib/python3.12/site-packages/django_rq/utils.pyget_scheduler_pidr+sz  !%**-  >,"";#>#>uzz#JK$'Os3::< ==    sAB8B<B Bc :g}ttD]\}}t|}|j}|jj }|rt |t||j|jd}|r |j|jdnd}|r%t|jjd} nd} |jdd|jdd|jdd|j |j"| ||t%|d} t'|j }t)j"| | d <t+|j |} t-|j |} t/|j |} t1|j |}t3|j |}t5| | d <t5| | d <t5| | d <t5|| d<t5|| d<|j7| d|iS)Nrzutf-8z%Y-%m-%d, %H:%M:%S-connection_pool parser_classretry)rjobsoldest_job_timestampindexconnection_kwargs scheduler_pid)r&workers finished_jobs started_jobs deferred_jobs failed_jobsscheduled_jobsqueues) enumeraterrr r.r4rrlindexkey fetch_jobr$r enqueued_atstrftimepoprcountr+rrrrrrrlenappend)run_maintenance_tasksr<r3configr&r r4 last_job_idlast_jobr2 queue_datafinished_job_registrystarted_job_registrydeferred_job_registryfailed_job_registryscheduled_job_registrys r*get_statisticsrQ0s F";/0" v"5)%% &66HH U # !% ( !'' 15 CN5??;#5#5g#>?TX #/0D0D#E#N#NOc#d #&  /6nd3gt,JJKK$8!2.u5  $EJJ/ & 5 9 9 3EJJ K1%**jI 3EJJ K/ JG!5ejj*!M&)*?&@ ?#%()=%> >"&)*?&@ ?#$'(;$< =!'*+A'B #$ j!a0"d f c Ji}ttD]\}}t|}t|jj t rD|jj jjd}|j j}n |jj j}|jddd|jddd|jdd}||vs t|d }|j|d ||<d |iS#t$rYwxYw) NrhostNOHOST:porti/dbr)rDr3 schedulers)r=rr isinstancer r.r sentinel_manager sentinelsr4r!rrDr)rZr3rHr&first_sentinelr conn_keyr's r*get_scheduler_statisticsr`gs#J";/ v#5) e&&668N O"--==NNXXYZ[N'77IIJ))99KKJ nnVX67qPT9U8VVWXbXfXfgkmnXoWpq : % )&.9 &__."' 8$( * %%(  s+$D D"!D"Nregistryreturnctj||j|j}g}t |D]0\}}||s |j || |j |2|S)zFetch jobs in bulk from Redis. 1. If job data is not present in Redis, discard the result 2. If `registry` argument is supplied, delete empty jobs from registry )r serializer)r fetch_manyr rdr=removerF)r&job_idsrar1 valid_jobsijobs r*get_jobsrksl" >>'e.>.>5K[K[ \DJD/#3 ; +   c " # rRcomposite_keyscg}|D]7\}} |jtj|||j9|S#t$rYGwxYw)znFetch executions in bulk from Redis. 1. If execution data is not present in Redis, discard the result )idjob_idr )rFr fetchr ValueError)r&rl executionsrorns r*get_executionsrss^J$     iooFuO_O_` a    s1A A  A ct|ttfr|n|g}g}g}|D]*} t|j||j |,||fS#t $r|j |YMwxYwN)r[listtupler r r%rF)r&rgstopped_job_idsfailed_to_stop_job_idsros r* stop_jobsrzs#GdE];g'GO'  !%"2"2F ; v& ' 2 22   " ) )& 1  sAA0/A0cVtjD]}|jyru)r allclose)cs r*reset_db_connectionsrs! __   rRc Pddl}|jdd|jdd|jjj |jj j |jjjgd}|j|fi|y)a Configure the Sentry client. The **options kwargs are passed straight from the command invocation - options relevant to Sentry configuration are extracted. In addition to the 'debug' and 'ca_certs' options, which can be passed in as command options, we add the RqIntegration and DjangoIntegration to the config. Raises ImportError if the sentry_sdk is not available. rN sentry_debugFsentry_ca_certs)debugca_certs integrations) sentry_sdkr!rredisRedisIntegrationrq RqIntegrationdjangoDjangoIntegrationinit) sentry_dsnoptionsrsentry_optionss r*configure_sentryrs^U3KK 148  # # ) ) : : <  # # & & 4 4 6  # # * * < < > NJOOJ1.1rR)Fru).typingrrrrrdjango.core.exceptionsr django.dbr redis.sentinelr rq.commandr rq.executionsr rq.jobr rq.registryrrrrrr rq.workerrrq.worker_registrationrr<rrrsettingsrtemplatetags.django_rqrr+rQr`rkstrrsrzrrrRr*rs557!1,#8EE!0.4n&H         #Y: $uS#X*? DO  3 2rR