a dG2@sddlZddlZddlZddlmZmZddlmZddlm Z m Z ddl m Z ddlmZmZmZddlmZddlmZdd lmZdd lmZmZdd lmZdd lmZmZdd l m!Z!m"Z"e#e$Z%GdddZ&Gddde&eZ'Gddde&eZ(dddZ)ee*dddZ+dS)N)UnionList)events)JobSubmissionEventJobExecutionEvent)Job) BaseJobStoreJobLookupErrorConflictingIdError)MemoryJobStore) BaseScheduler)db) transactionIntegrityError)util) DjangoJobDjangoJobExecution)get_apscheduler_datetimeget_django_internal_datetimecsveZdZdZdZfddZeedddZee e e dfdd d Z ee e e dfdd d Z d dZZS)DjangoResultStoreMixinzfMixin class that adds the ability for a JobStore to store job execution results in the Django databaseNcs&t|||jt_|dSN)superstart _schedulerZ _create_lockrlockregister_event_listeners)self scheduleralias __class__X/var/www/html/django/DPS/env/lib/python3.9/site-packages/django_apscheduler/jobstores.pyrs zDjangoResultStoreMixin.start)eventcCsz|jtjkr,t|j|j|jdtj}n`|jtj krltj }d|jd}tj|j|j|jd||d}n t d|jdtjtj gdWn(t yt d|jd Yd S0|jS) a Create and return new job execution instance in the database when the job is submitted to the scheduler. :param event: JobExecutionEvent instance :return: DjangoJobExecution ID or None if the job execution could not be logged. rzExecution of job 'z7' skipped: maximum number of running instances reached! exceptionz-Don't know how to handle JobSubmissionEvent ''. Expected one of ''.Job '8' no longer exists! Skipping logging of job execution...N)coderEVENT_JOB_SUBMITTEDratomic_update_or_createrjob_idZscheduled_run_timesZSENTEVENT_JOB_MAX_INSTANCESZ MAX_INSTANCESNotImplementedErrorrloggerwarningid)clsr# job_executionstatusr%r!r!r"handle_submission_event's<       z.DjangoResultStoreMixin.handle_submission_event)r#returncCsr|jtjkr&td|jdtjdzt|j|j|jtj }Wn(t yjt d|jdYdS0|j S)z Store "successful" job execution status in the database. :param event: JobExecutionEvent instance :return: DjangoJobExecution ID or None if the job execution could not be logged. ,Don't know how to handle JobExecutionEvent 'z '. Expected 'r'r(r)N)r*rEVENT_JOB_EXECUTEDr/rr,rr-scheduled_run_timeSUCCESSrr0r1r2)r3r#r4r!r!r"handle_execution_eventUs&    z-DjangoResultStoreMixin.handle_execution_eventcCsz|jtjkr\|jr*t|j}t|j}nd|jd}d}tj|j |j|j tj ||d}n`|jtj krtj }d|jd}tj|j |j|j ||d}n$td|jd tjtjtj gd Wn(tytd|jd YdS0|jS) z Store "failed" job execution status in the database. :param event: JobExecutionEvent instance :return: DjangoJobExecution ID or None if the job execution could not be logged. r(z' raised an error!N)r% tracebackzRun time of job 'z ' was missed!r$r8r&r'r))r*rEVENT_JOB_ERRORr%strr=r-rr,rr:ERROREVENT_JOB_MISSEDZMISSEDr/r.rr0r1r2)r3r#r%r=r4r5r!r!r"handle_error_eventrsH        z)DjangoResultStoreMixin.handle_error_eventcCsF|j|jtjtjB|j|jtj|j|jtj tj BdS)z Register various event listeners. See: https://github.com/agronholm/apscheduler/blob/master/docs/modules/events.rst for details on which event class is used for each event code. N) rZ add_listenerr6rr+r.r<r9rBr>rArr!r!r"rs  z/DjangoResultStoreMixin.register_event_listeners)__name__ __module__ __qualname____doc__rr classmethodrr6rrintr<rBr __classcell__r!r!rr"rs -4rcseZdZdZejfedfdd Zej e e de fdddZ ee d d d Zej d d ZddZej e dddZej e dddZej e dddZej ddZddZddZej ddZd d!ZZS)"DjangoJobStorea5 Stores jobs in a Django database. Based on APScheduler's `MongoDBJobStore`. See: https://github.com/agronholm/apscheduler/blob/master/apscheduler/jobstores/mongodb.py :param int pickle_protocol: pickle protocol level to use (for serialization), defaults to the highest available )pickle_protocolcst||_dSr)r__init__rL)rrLrr!r"rMs zDjangoJobStore.__init__N)r-r7cCs@z$tjj|dj}|r ||ndWStjy:YdS0dSN)r2)robjectsget job_state_reconstitute_job DoesNotExist)rr-rQr!r!r" lookup_jobs zDjangoJobStore.lookup_job)r7cCst|}|j|dS)N)Znext_run_time__lte)r _get_jobs)rnowdtr!r!r" get_due_jobsszDjangoJobStore.get_due_jobscCs@z$tjjddd}t|j|jWStjy:YdS0dS)NF)Znext_run_time__isnull next_run_time)rrOfilterZearliestrrYrrSrjobr!r!r"get_next_run_timesz DjangoJobStore.get_next_run_timecCs|}|||Sr)rUZ_fix_paused_jobs_sorting)rjobsr!r!r" get_all_jobss zDjangoJobStore.get_all_jobs)r\c Cstfz8tjj|jt|jt | |j dWWdSt y^t |jYn0Wdn1st0YdS)N)r2rYrQ)ratomicrrOcreater2rrYpickledumps __getstate__rLrr r[r!r!r"add_jobs  zDjangoJobStore.add_jobc Cstlz)r rDrLrCr!r!r"__repr__6szDjangoJobStore.__repr__)rDrErFrGrbHIGHEST_PROTOCOLrIrMrZretry_on_db_operational_errorr?rrprTrrXr]r_rergrirkrnrRrUrzrJr!r!rr"rKs*      rKc@seZdZdZdS)DjangoMemoryJobStorez Adds the DjangoResultStoreMixin to the standard MemoryJobStore so that job executions can be logged to the Django database. N)rDrErFrGr!r!r!r"r|:sr|cCstdtdS)Nza'register_events' is deprecated since version 0.4.0. Please remove all references from your code.)warningswarnDeprecationWarning)rZresult_storager!r!r"register_eventsCs r)rr7csfdd}|S)a Helper decorator for job registration. Automatically fills id parameter to prevent jobs duplication. See this comment for explanation: https://github.com/jcass77/django-apscheduler/pull/9#issuecomment-342074372 Usage example:: @register_job(scheduler, "interval", seconds=1) def dummy_job(): print("I'm a job!") :param scheduler: Scheduler instance :param args, kwargs: Params, will be passed to scheduler.add_job method. See :func:`BaseScheduler.add_job` csBtdtd|jd|jj|gRi|S)NzThe 'register_job' decorator is deprecated since version 0.5.0. Please use APScheduler's add_job() method or @scheduled_job decorator instead.r2.)r}r~r setdefaultrErDre)funcargskwargsrr!r"wrapper_register_job^sz*register_job..wrapper_register_jobr!)rrrrr!rr" register_jobMs r)N),loggingrbr}typingrrZ apschedulerrZapscheduler.eventsrrZapscheduler.jobrrpZapscheduler.jobstores.baserr r Zapscheduler.jobstores.memoryr Zapscheduler.schedulers.baser djangor Z django.dbrrZdjango_apschedulerrZdjango_apscheduler.modelsrrZdjango_apscheduler.utilrr getLoggerrDr0rrKr|rcallablerr!r!r!r"s*       %|