o q~9S@sTddlmZmZddlZddlmZddlmZddlm Z z ddlm Z m Z Wn e y5dZ Z Ynwddlm Z ddlmZdd lmZdd lmZd d lmZmZmZd d lmZGdddeZd"ddZddZGdddeZGddde jZ Gddde Z!Gddde Z"Gddde!Z#Gddde!Z$Gd d!d!e Z%dS)#)absolute_importunicode_literalsNwraps)count) connection) connectionsrouter)models)QuerySet)settings)maybe_timedelta)commit_on_success get_querysetrollback_unless_managed)nowc@s eZdZdS)TxIsolationWarningN)__name__ __module__ __qualname__rrC/var/www/html/env/lib/python3.10/site-packages/djcelery/managers.pyrsrcsfdd}|S)zDecorator for methods doing database operations. If the database operation fails, it will retry the operation at most ``max_retries`` times. cstfdd}|S)Nc sl|d}tdD])}z |i|WSty3||kr"ztWn ty0YnwYq wdS)Nexception_retry_countr)popr Exceptionr)argskwargs _max_retriesretries)fun max_retriesrr_inner%s     z1transaction_retry.._outer.._innerr)r r"r!)r r_outer#sz!transaction_retry.._outerr)r!r$rr#rtransaction_retrys r%cs"fdd|DS)Ncsg|] \}}t||qSr)setattr).0 attr_name attr_valueobjrr =sz*update_model_with_dict..)itemssave)r+fieldsrr*rupdate_model_with_dict<s r0c@eZdZddZdS)ExtendedQuerySetcKs@|jdi|\}}|st|di}||t|||S)Ndefaultsr) get_or_createdictrupdater0)selfrr+createdr/rrrupdate_or_createEs   z!ExtendedQuerySet.update_or_createN)rrrr9rrrrr2C r2c@s8eZdZddZeZddZddZddZd d Zd S) ExtendedManagercCs t|jSN)r2modelr7rrrrRs zExtendedManager.get_querysetcKst|jdi|S)Nr)rr9)r7rrrrr9Vsz ExtendedManager.update_or_createcCstr tt|jStSr<)rr db_for_writer=rr>rrrconnection_for_writeYsz$ExtendedManager.connection_for_writecCstrt|jStSr<)rdbrr>rrrconnection_for_read^s z#ExtendedManager.connection_for_readcCs,z tj|jdWStytjYSw)NENGINE)r DATABASESrAAttributeErrorDATABASE_ENGINEr>rrrcurrent_enginecs   zExtendedManager.current_engineN) rrrr get_query_setr9r@rBrGrrrrr;Ps r;c@seZdZddZddZdS) ResultManagercCs|jtt|dS)zGet all expired task results.) date_done__lt)filterrr )r7expiresrrrget_all_expiredlszResultManager.get_all_expiredcCsd|jj}t!||jdd|}|d|dWddS1s+wYdS)z#Delete all expired taskset results.Thidden(DELETE FROM {0.db_table} WHERE hidden=%sTN) r=_metarrMr6r@cursorexecuteformat)r7rLmetarSrrrdelete_expiredps "zResultManager.delete_expiredN)rrrrMrWrrrrrIjs rIc@r1)PeriodicTaskManagercC |jddS)NT)enabledrKr>rrrrZ~ zPeriodicTaskManager.enabledN)rrrrZrrrrrX|r:rXc@s:eZdZdZdZddZedd d ddZd d ZdS) TaskManagerz/Manager for :class:`celery.models.Task` models.NcCsJz|j|dWS|jjy$|j|kr|||_|j|dYSw)aBGet task meta for task by ``task_id``. :keyword exception_retry_count: How many times to retry by transaction rollback on exception. This could theoretically happen in a race condition if another worker is trying to create the same task. The default is to retry once. )task_id)getr= DoesNotExist_last_idwarn_if_repeatable_read)r7r^rrrget_tasks  zTaskManager.get_taskr#cCs|j||||d|iddS)a+Store the result and status of a task. :param task_id: task id :param result: The return value of the task, or an exception instance raised by the task. :param status: Task status. See :meth:`celery.result.AsyncResult.get_status` for a list of possible status values. :keyword traceback: The traceback at the point of exception (if the task failed). :keyword children: List of serialized results of subtasks of this task. :keyword exception_retry_count: How many times to retry by transaction rollback on exception. This could theoretically happen in a race condition if another worker is trying to create the same task. The default is to retry twice. children)statusresult tracebackrV)r^r3r9)r7r^rgrfrhrerrr store_resultszTaskManager.store_resultcCsXd|vr&|}|dr(|d}|dkr*ttddSdSdSdS)NmysqlzSELECT @@tx_isolationrzREPEATABLE-READzPolling results with transaction isolation level repeatable-read within the same transaction may give outdated results. Be sure to commit the transaction for each poll iteration.) rGlowerrBrSrTfetchonewarningswarnr)r7rS isolationrrrrbs    z#TaskManager.warn_if_repeatable_read)NN) rrr__doc__rarcr%rjrbrrrrr]s r]c@s2eZdZdZddZddZedddd Zd S) TaskSetManagerz2Manager for :class:`celery.models.TaskSet` models.cCs(z|j|dWS|jjyYdSw)z,Get the async result instance by taskset id.) taskset_idN)r_r=r`)r7rsrrrrestore_tasksets zTaskSetManager.restore_tasksetcCs||}|r |dSdS)zDelete a saved taskset result.N)rtdelete)r7rssrrrdelete_tasksets  zTaskSetManager.delete_tasksetrdr#cCs|j|d|idS)zStore the async result instance of a taskset. :param taskset_id: task set id :param result: The return value of the taskset rg)rsr3ri)r7rsrgrrrrjs zTaskSetManager.store_resultN)rrrrqrtrwr%rjrrrrrrs rrc@s0eZdZddZefddZddZddZd S) TaskStateManagercCrY)NFrNr[r>rrractiver\zTaskStateManager.activecCs|j||t|dS)N) state__in tstamp__lte)rKr )r7statesrLnowfunrrrexpireds zTaskStateManager.expiredcCs |dur|||jddSdS)NTrN)r~r6)r7r|rLrrrexpire_by_statessz!TaskStateManager.expire_by_statescCsRt|jj}|}|d|dWddS1s"wYdS)NrPrQ)rr=rRr@rSrTrU)r7rVrSrrrpurges "zTaskStateManager.purgeN)rrrryrr~rrrrrrrxs   rx)r)& __future__rrrn functoolsr itertoolsr django.dbrrr ImportErrorr django.db.models.queryr django.confr celery.utils.timeutilsr rArrrutilsr UserWarningrr%r0r2Managerr;rIrXr]rrrxrrrrs4             B