֒Tij<dZddlZddlZddlZddlmZmZddlmZddl m Z m Z m Z ddl mZddlmZddlmZdd lmZdd lmZdd lmZdd lmZmZdd lmZddlmZm Z ejBe"Z#GddZ$GddeZ%y)z Base class for cron tasks. N)ABCabstractmethod)date)OptionalDictAny) transaction)settings)timezone) CronExecution)ExecutionPattern) DatabaseLock)ValidationErrorConcurrentExecutionError)NotificationManager)CronjobHookContextget_global_cronjob_hook_managercLeZdZdZd dedededededeed eeee ffd Z y) ExecutionResultzResult of task execution.Nsuccessmessage error_codeskippedreason exceptiondatacn||_||_||_||_||_||_|xsi|_y)N)rrrrrrr)selfrrrrrrrs M/home/cursorai/projects/django-cronjob-utils/src/django_cronjob_utils/base.py__init__zExecutionResult.__init__s7  $  "JB )Tr"Fr"NN) __name__ __module__ __qualname____doc__boolstrr Exceptionrrr r!rrrs#cC_cuxKST]K^muvz{~@C{CvDmEr!rczeZdZUdZdZeed<dZeed<ejZ eed<dZ e ed<dZ eed <d Zeed <d Zeeed <defdZdZde fdZdZdefdZdefdZdefdZdefdZdedefdZde fdZdZ defdZ!d'deedee"ee#fde$fd Z%d!e$fd"Z&d!e$fd#Z'defd$Z(e)d%ede*fd&Z+y )(CronTaskz'Abstract base class for all cron tasks.r" task_name task_codeexecution_patternFretry_on_failurer max_retriesi, retry_delayNtimeoutexecution_datec ||_||_d|_t|_t |_|jdd|_y)z Initialize cron task. Args: execution_date: Date for which the task should be executed **options: Additional options (e.g., force, rerun) N enable_hooksT) r4options executionr_notification_managerr _hook_managerget _enable_hooks)rr4r7s rr zCronTask.__init__0sC- 26%8%:"<>$[[>r!czt|jts!tdt |jy)zBValidate task input. Override in subclasses for custom validation.z*execution_date must be a date object, got N) isinstancer4rrtypers rvalidatezCronTask.validate?s7$--t4!$NtTXTgTgOhNi"jk k5r!returncD|jjdd}|ry|jjdd}|ry|jtjk(ry|jtj k(rEt jj|j|jdj S|jtjk(ryt jj|j|jdj S)z0Check if task should run (duplicate prevention).forceFTrerun)r.r4r)r.r4 completed) r7r;r/r ALWAYSRERUN_ON_FAILUREr objectsfilterr.r4exists RATE_LIMITED)rrDrEs r should_runzCronTask.should_runDs   %0    %0   ! !%5%<%< <  ! !%5%F%F F$,,33..#224fh    ! !%5%B%B B!((//nn..0  &(  r!cBt|j|jS)z6Acquire database lock to prevent concurrent execution.)rr.r4r@s r acquire_lockzCronTask.acquire_lockesDNND,?,?@@r!ctjj|j|j|j t jdd|_|jS)z$Create execution record in database.F)r.r-r4pidrFr) r rIcreater.r-r4osgetpidr8r@s rcreate_execution_recordz CronTask.create_execution_recordisQ&..55nnnn..  6 ~~r!cZ|jr|jS|jS)zExecute with optional timeout.)r3_execute_with_signal_timeout_executer@s rexecute_with_timeoutzCronTask.execute_with_timeoutus$ <<446 6}}r!c.ttds3tjdjdj Sfd}tjtj |}tjj j }tjdtjtj ||S#tjdtjtj |wxYw)z.Execute with signal-based timeout (Unix only).SIGALRMz-Timeout not supported on this platform. Task z will run without timeout.c6tdjd)Nz#Task execution exceeded timeout of  seconds) TimeoutErrorr3)signumframers rtimeout_handlerz>CronTask._execute_with_signal_timeout..timeout_handlers!DT\\NRZ[\ \r!r) hasattrsignalloggerwarningr-rXr[alarmr3)rra old_handlerresults` rrWz%CronTask._execute_with_signal_timeout{svy) NN''AC ==? " ]mmFNNOD  T\\" 7]]_F LLO MM&..+ 6  LLO MM&..+ 6s C;Dc |j|j}t|tr?t |j dd |j dd|j ddSt|tr|St d|rt | Sd S#t$rX}|j|}tjd |jd |t dt ||| cYd }~Sd }~wwxYw)zInternal execution method.errorFrr"rrrrTzTask completed successfully)rrTask z execution failed: )rrrrN) executer4r>dictrr;r(r)get_error_coderdrr-)rrhers rrXzCronTask._executes \\$"5"56F&$'& & 7E ::"JJy"5%zz,;&/2 #'-F  3P  ,,Q/J   uT^^$44GsK L"A%   s1A)B,B>BB C>&A C93C>9C>rc.|jjS)z?Get error code from exception. Override for custom error codes.) __class__r#)rrs rrozCronTask.get_error_codes""+++r!c|jsy|jsy|jj|jk\ryy)z Check if task should be retried.FT)r0r8 retry_countr1r@s r should_retryzCronTask.should_retrys6$$~~ >> % %)9)9 9r!c |jr|jjtjd|jjdzd|j d|j d|jd y)z:Schedule retry execution. Override for custom retry logic.zScheduling retry /z for task z after r]N)r8increment_retryrdinfortr1r-r2r@s rschedule_retryzCronTask.schedule_retrysn >> NN * * ,  : :Q >?qAQAQ@RS'wt/?/?.@ J r!rhc|jr7|jxsd}|jj|j|yy)zNotify stakeholders of failure.z Unknown errorN)r8rr9notify_failure)rrh error_messages rr}zCronTask.notify_failures6 >>"NN=oM  & & 5 5dnnm T r! execution_idmetadatac td|j|j|j|j||j r|j j nd|j|xsi S)z*Create hook context for current execution.rmr) operationr.r-r4r/rrtr7r)rr.r-r4r/r8rtr7)rrrs r_create_hook_contextzCronTask._create_hook_contextsX!nnnn.."44%6:nn22!LL^  r!contextcr|jsy |jj|y#t$rwxYw)z(Execute PRE hooks before task execution.N)r<r:execute_pre_hooksr)rrs r_execute_pre_hookszCronTask._execute_pre_hookss9!!      0 0 9   s+ 6cT|jsgS|jj|S)z(Execute POST hooks after task execution.)r<r:execute_post_hooksrs r_execute_post_hookszCronTask._execute_post_hookss'!!I!!44W==r!c|j|js1td|jtj k7rdSdS|j d} |j|d} |j!5|j#|_|j$j&}|j ||j(} |j+}|j$j-|j.|j0|j2|j5|}|r=tjd |jd|Dcgc]}|j2c}|j.s|j7||j.s |j9r|j;|j.s|j<r |j<|cdddS#t$rz}t|d|jj}t|}t|di}tjd |jd |d |td ||| cYd}~Sd}~wwxYwcc}w#t>tf$r}tA|t>s|jC|nd}t|}|j$r|j$jE|||j7td |||j9r|j;d}~wwxYw#1swYyxYw#tF$rt>tf$r}| tHjJjM|#tHjN$rtA|t>s|jC|nd}t|}tQjR5tHjJjU|jV|j|jXt[j\dd ||t_j` dddY#1swYYxYwwxYwd}~wwxYw)z5Main execution method with full lifecycle management.TzAlready executedSkipped)rrN)rrdetailsrlz rejected by PRE hook: z - F)rrrr)rrkz& completed but POST hooks had errors: TIMEOUT)rr)pk) r.r-r4rQrFrrrended)1rArMrr/r rGrrr)getattrrrr#r(rdrer-rOrUr8idrrYmark_completedrrrrr}rur{rr^r>ro mark_failedrr rIr; DoesNotExistr atomicrRr.r4rSrTr now) rrrprr~ error_detailsrrhpost_hook_errorss rrunz CronTask.runs  "-1-C-CGWG^G^-^) dm  +++>   # #G ,& Z ""$!%!=!=!?#~~00 33L7K[K[3\6!668FNN11 & &#)#4#42(,'?'?'H$'#DNN#33Y6FG6F 6FGHJ ">>++F3">>d.?.?.A++-">>f.>.>$...!O%$)  L!++2F2FGJFM#Ay"5M NN''>zl#m_] #%%"   X H"%i0?I!\?Z!4!4Q!7`iJ$'FM~~22$1'13 '' % -#-)((*++-+S%$~(  i( '!))--->$ #%11?I!\?Z!4!4Q!7`iJ$'FM$++-%--44&*nn&*nn+/+>+> " &*$)$1'1"*,,.5 .- .- #" / s%G'9L* A LBI2I-&A7I2 L*' I*0A/I%I*%I*-I22LBLLLL'#L*'L**QQ M'&Q'AP?+=>}_}~  D  T    r!r,)&r&loggingrSrcabcrrdatetimertypingrrr django.dbr django.confr django.utilsr django_cronjob_utils.modelsr django_cronjob_utils.registryr django_cronjob_utils.locksrdjango_cronjob_utils.exceptionsrr"django_cronjob_utils.notificationsrdjango_cronjob_utils.hooksrr getLoggerr#rdrr,r*r!rrsf #&&! !5:3UBZ   8 $  c sc r!