o ͜iK@sdZddlZddlZddlZddlmZzddlmZWney+ddl mZYnwddl m Z m Z ddl mZmZddlmZddlmZdd lmZdd lmZdd lmZmZdd lmZmZmZmZm Z dd l!m"Z"ddl#m$Z$m%Z%ddl&m'Z'm(Z(ddl)m*Z*m+Z+ddl,m-Z-ddl.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4ddl5m6Z6m7Z7m8Z8dZ9dZ:dZ;ee<Z=e=j>e=j?e=j@Z>Z?Z@GdddeZAGdddeZBdS)zBeat Scheduler Implementation.N)Finalize)ZoneInfo) current_app schedules) ScheduleEntry Scheduler) get_logger)maybe_make_aware)settings)ObjectDoesNotExist)close_old_connections transaction)CaseF IntegerFieldQWhen)Cast) DatabaseErrorInterfaceError) safe_reprsafe_str)dumpsloads)clocked)ClockedScheduleCrontabScheduleIntervalSchedule PeriodicTask PeriodicTasks SolarSchedule)NEVER_CHECK_TIMEOUT aware_nownowi,z;Cannot add entry %r to database schedule: %r. Contents: %r c@seZdZdZejedfejedfej e dfe e dffZ gdZddd Zd d Zd d ZddZddZeZddZeddZedddZe dddZe  d ddZddZdS)! ModelEntryz(Scheduler entry taken from database row.crontabintervalsolarr) last_run_attotal_run_count no_changesNc Cst|pt|_|j|_|j|_z|j|_Wn|jy+td|j| |Ynwzt |j p2d|_ t |j p:d|_ Wnt y]}ztd|j|| |WYd}~nd}~wwi|_dD]}t||}|duroqc||j|<qct|ddrt|d|jd<t |jpd}|j|d <||jd <|j|_||_|js|jp||_|jjr|jtjd d |_|j|_dS) zInitialize the model entry.z4Disabling schedule %s that was removed from databasez[]z{}z:Removing schedule %s for argument deseralization error: %rN)queueexchange routing_keypriorityexpires_expiresperiodic_task_nameheadersi*)days)r_get_current_objectappnametaskschedule DoesNotExistloggererror_disablerargskwargs ValueError exceptionoptionsgetattrr4r+modelr* date_changed _default_now start_timedatetime timedelta)selfrEr7excoptionvaluer4rOX/var/www/html/philips/venv/lib/python3.10/site-packages/django_celery_beat/schedulers.py__init__7sV        zModelEntry.__init__cCsd|_d|_|dS)NTF)r,enabledsave)rKrErOrOrPr>ks zModelEntry._disablecCs&|jjs tddS|jjdur@|}ttddr t|}||jjkr@|j }|j |}||}t | }td|S|jjdur\|}||jjkr\||jtdtS|jjr|jjr|jjdkrd|j_d|j_d|j_|jtdtS|jj}t|j|}|j|S)NFg@DJANGO_CELERY_BEAT_TZ_AWARETr)rErRr schedstaterHrGrDr r tzinfodue_start_timemathceil total_secondsr2r>r"one_offr+r,rSr7timezoner* astimezoner:is_due)rKr$ current_tzrHtime_remainingdelaytzlast_run_at_in_tzrOrOrPr^ps6               zModelEntry.is_duecCs.ttddrtj|jj}|Stj}|S)NrTT)rDr rIr$r7r\utcnow)rKr$rOrOrPrGs  zModelEntry._default_nowcCs0||j_|jjd7_d|j_||jS)NrT)rGrEr*r+r, __class__rKrOrOrP__next__s  zModelEntry.__next__cCsDt|jjj|jjd}|jD] }t||t|j|q|dS)N)pk) typerE_default_managergetrh save_fieldssetattrrDrS)rKobjfieldrOrOrPrSs  zModelEntry.savecCsT|jD]\}}}t|}t||r!||}|||fSqtd|d)NzCannot convert schedule type z to model)model_schedulesrmaybe_schedule isinstance from_schedulerSrA)clsr: schedule_type model_type model_fieldmodel_schedulerOrOrPto_model_schedules     zModelEntry.to_model_schedulecKs,tjj||jdi|d\}}|||dS)N)r8defaultsr7rO)rrjupdate_or_create_unpack_fields)rtr8r7entryrncreatedrOrOrP from_entrys zModelEntry.from_entryc Ksbdd|jD}||\}} ||| <|j|ft|pgt|p id|jdi|p*i|S)NcSsi|]\}}}|dqSNrO).0_rwrOrOrP  z-ModelEntry._unpack_fields..)r?r@rO)rpryupdater_unpack_options) rtr:r?r@relativerCr~entry_schedulesrxrwrOrOrPr}s  zModelEntry._unpack_fieldscKs||||t|pi|dS)N)r-r.r/r0r4expire_seconds)r)rtr-r.r/r0r4rr@rOrOrPrs zModelEntry._unpack_optionscCs(dt|j|jt|jt|j|jS)Nz!)formatrr8r9rr?r@r:rfrOrOrP__repr__s zModelEntry.__repr__r)NNNN)NNNNNN)__name__ __module__ __qualname____doc__rr'rr:rr)r!rrrprlrQr>r^rGrgnextrS classmethodryrr}rrrOrOrOrPr&,s6    4,     r&cseZdZdZeZeZeZ dZ dZ dZ dZ dZddZddZd d Zd d Zd dZddZddZddZddZddZddZddZddZfdd Zed!d"ZZS)#DatabaseSchedulerzDatabase-backed Beat Scheduler.NTFcOsPt|_tj|g|Ri|t||jdd|_|dp$|jj j p$t |_ dS)z"Initialize the database scheduler.r%) exitpriority max_intervalN) set_dirtyrrQrsync _finalizerkr7confbeat_max_loop_intervalDEFAULT_MAX_INTERVALrrKr?r@rOrOrPrQs zDatabaseScheduler.__init__cCs ||j||jjjdSr)install_default_entriesr:update_from_dictr7r beat_schedulerfrOrOrPsetup_schedules z DatabaseScheduler.setup_schedulec CsJtdi}|D]}z |j||jd||j<Wq ty"Yq w|S)Nz-DatabaseScheduler: Fetching database scheduler{)debugenabled_modelsEntryr7r8rA)rKsrErOrOrPall_as_schedules  z!DatabaseScheduler.all_as_schedulecCs t|S)zReturn list of enabled periodic tasks. Allows overriding how the list of periodic tasks is fetched without duplicating the filtering/querying logic. )listenabled_models_qsrfrOrOrPrs z DatabaseScheduler.enabled_modelscCs@ttjtd}td|d}|}||B}|jj |S)N)secondsF)clocked__isnullclocked__clocked_time__gt) r$rIrJSCHEDULE_SYNC_MAX_INTERVALr_get_crontab_exclude_queryModelobjectsrRexclude)rKnext_schedule_syncexclude_clock_tasks_queryexclude_cron_tasks_query exclude_queryrOrOrPrsz#DatabaseScheduler.enabled_models_qscst}|jfddtddD}|dg7}d}tjj|d}|jtdtt fd d Dd t d id }|j |d j ddd}tddt|d@}|S)a$ Build a query to exclude crontab tasks based on their hour value, adjusted for timezone differences relative to the server. This creates an annotation for each crontab task that represents the server-equivalent hour, then filters on that annotation. csg|]}|dqS)rO)roffset) server_hourrOrP 8rz@DatabaseScheduler._get_crontab_exclude_query..z^\d+$) hour__regexhourcs.g|]}t|td|dddqS)hour_intr)r\then)rr_get_timezone_offset)r timezone_namerfrOrPrNs defaultr)rr)server_hour__inidTflatF)crontab__isnull)crontab__id__in)r#rrangerrfilterannotaterrr_get_unique_timezone_namesrr values_listr)rK server_timehours_to_includenumeric_hour_patternnumeric_hour_tasksannotated_tasksexcluded_hour_task_idsrrO)rKrrPr*s:       z,DatabaseScheduler._get_crontab_exclude_querycCstjjdddS)z?Get a list of all unique timezone names used in CrontabScheduler\Tr)rrrdistinctrfrOrOrPrjs z,DatabaseScheduler._get_unique_timezone_namesc Cst}t|jtr |j}ntt|j}t|tr|j}t|}tdddddd}|j|d}|j|d}| | }t |d} | S)z Args: timezone_name: The name of the timezone or a ZoneInfo object Returns: int: The hour offset ir r)rVi) r#rrrVrstrkeyrIreplace utcoffsetrZint) rKrr server_tz target_tzfixed_dtdt1dt2offset_seconds offset_hoursrOrOrPrps     z&DatabaseScheduler._get_timezone_offsetc CsztztWn tjyYnw|j|j}}Wn&ty8}z t d|WYd}~dSd}~wt yEt dYdSwz|rW||rN|n|kr]W||_dSW||_dSW||_dS||_w)NzDatabase gave error: %rFzYDatabaseScheduler: InterfaceError in schedule_changed(), waiting to retry in next call...T) r r commitTransactionManagementError_last_timestampChanges last_changerr<rBrwarning)rKlasttsrLrOrOrPschedule_changeds:    z"DatabaseScheduler.schedule_changedcCst|}|j|j|Sr)rraddr8)rKr~ new_entryrOrOrPreserveszDatabaseScheduler.reservec Csttjr tdt}t}zsz0t|jr@|j}z|j | | |Wnt t tfy<| |Ynw|jsWn$tyY}z td|WYd}~nd}~wtyetdYnwW|j|O_dSW|j|O_dSW|j|O_dS|j|O_w)NzWriting entries...zDatabase error while sync: %rzMDatabaseScheduler: InterfaceError in sync(), waiting to retry in next call...)r< isEnabledForloggingDEBUGrrr rpop _schedulerSrKeyError TypeErrorr rrBrr)rK_tried_failedr8rLrOrOrPrs<   "zDatabaseScheduler.syncc Csi}|D]5\}}z|jj|fd|ji|}|jjr |||<Wqty;}ztt |||WYd}~qd}~ww|j |dS)Nr7) itemsrrr7rErR Exceptionr<rBADD_ENTRY_ERRORr:r)rKmappingrr8 entry_fieldsr~rLrOrOrPrs  z"DatabaseScheduler.update_from_dictc Cs>i}|jjjr|ddtdddddid||dS)Nzcelery.backend_cleanup04*ri)r9r:rC)r7rresult_expires setdefaultrr'r)rKdataentriesrOrOrPrs  z)DatabaseScheduler.install_default_entriescs"|jrd|_dStj|i|S)NF)_heap_invalidatedsuperschedules_equalrrerOrPrsz!DatabaseScheduler.schedules_equalcCsd}}tj}|jrtdd}}d|_||_n |r(tdd}||_|s>||j}|tkr>tdd}||_|rg| | |_ |sQg|_ d|_ ttjrgtdddd |j D|j S) NFzDatabaseScheduler: initial readTz$DatabaseScheduler: Schedule changed.z4DatabaseScheduler: Forcing full sync after 5 minuteszCurrent schedule: %s css|]}t|VqdSr)repr)rr~rOrOrP s z-DatabaseScheduler.schedule..)rIr$ _initial_readr_last_full_syncrinforZrrrr_heaprr<rrrjoinvalues)rKinitialr current_timetime_since_last_syncrOrOrPr:s@    zDatabaseScheduler.schedule) rrrrr&rrrr rrrr rr rQrrrrrrrrrrrrrpropertyr: __classcell__rOrOrrPrs4  @$ r)CrrIrrXmultiprocessing.utilrzoneinfor ImportErrorbackports.zoneinfoceleryrr celery.beatrrcelery.utils.logrcelery.utils.timer django.confr django.core.exceptionsr django.dbr r django.db.modelsrrrrrdjango.db.models.functionsrdjango.db.utilsrrkombu.utils.encodingrrkombu.utils.jsonrrclockedschedulermodelsrrrrr r!utilsr"r#r$rrrrr<rr rr&rrOrOrOrPsB         A