o fv@sdZddlZddlmZddlmZddlmZddlm Z ddl m Z m Z m Z mZdd lmZd ZGd d d eZdd dZGddde jZGdddeZGdddeZdS)zModel managers.Nwraps)count)maybe_timedelta)settings) connectionsmodelsrouter transaction)nowz Polling 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. c@seZdZdZdS)TxIsolationWarningzAWarning emitted if the transaction isolation level is suboptimal.N)__name__ __module__ __qualname____doc__rrQ/var/www/html/venv/lib/python3.10/site-packages/django_celery_results/managers.pyr sr csfdd}|S)zDecorate a function to retry database operations. For functions doing database operations, adding retrying if the operation fails. Keyword Arguments: max_retries (int): Maximum number of retries. Default one retry. cstfdd}|S)Nc sN|d}tdD]}z |i|WSty$||kr"Yq wdS)Nexception_retry_countr)popr Exception)argskwargs _max_retriesretries)fun max_retriesrr_inner%s   z1transaction_retry.._outer.._innerr)rrr)rr_outer#s z!transaction_retry.._outerr)rrrrrtransaction_retrys r c@s@eZdZdZddZddZddZdd Zd d Zd d Z dS) ResultManagerz#Generic manager for celery results.cCs\d|vr(|}|dr*|d}|dkr,ttt dSdSdSdS)NmysqlzPSHOW VARIABLES WHERE variable_name IN ('tx_isolation', 'transaction_isolation');r zREPEATABLE-READ) current_enginelowerconnection_for_readcursorexecutefetchonewarningswarnr W_ISOLATION_REPstrip)selfr& isolationrrrwarn_if_repeatable_read;s   z%ResultManager.warn_if_repeatable_readcCstt|jSN)rr db_for_writemodelr-rrrconnection_for_writeGsz"ResultManager.connection_for_writecCs t|jSr0)rdbr3rrrr%Js z!ResultManager.connection_for_readcCs,z tj|jdWStytjYSw)NENGINE)r DATABASESr5AttributeErrorDATABASE_ENGINEr3rrrr#Ms   zResultManager.current_enginecCs|jtt|dS)zGet all expired results.) date_done__lt)filterr rr-expiresrrrget_all_expiredSszResultManager.get_all_expiredcCsBtj|jd||WddS1swYdS)zDelete all expired results.usingN)r atomicr5r>deleter<rrrdelete_expiredWs"zResultManager.delete_expiredN) rrrrr/r4r%r#r>rCrrrrr!8s  r!c@s8eZdZdZdZddZedd    d ddZdS) TaskResultManagerz0Manager for :class:`~.models.TaskResult` models.NcCJz|j|dWS|jjy$|j|kr|||_|j|dYSw)a[Get result for task by ``task_id``. Keyword Arguments: exception_retry_count (int): How many times to retry by transaction rollback on exception. This could happen in a race condition if another worker is trying to create the same task. The default is to retry once. )task_idgetr2 DoesNotExist_last_idr/)r-rFrrrget_taskb  zTaskResultManager.get_taskrc Csf|||||||| | | | d }|| j||d\}}|s1|D] \}}t|||q |j| d|S)aFStore the result and status of a task. Arguments: content_type (str): Mime-type of result and meta content. content_encoding (str): Type of encoding (e.g. binary/utf-8). task_id (str): Id of task. periodic_task_name (str): Celery Periodic task name. task_name (str): Celery task name. task_args (str): Task arguments. task_kwargs (str): Task kwargs. result (str): The serialized return value of the task, or an exception instance raised by the task. status (str): Task status. See :mod:`celery.states` for a list of possible status values. worker (str): Worker that executes the task. using (str): Django database connection to use. traceback (str): The traceback string taken at the point of exception (only passed if the task failed). meta (str): Serialized result meta data (this contains e.g. children). Keyword Arguments: exception_retry_count (int): How many times to retry by transaction rollback on exception. This could happen in a race condition if another worker is trying to create the same task. The default is to retry twice. ) statusresult tracebackmetacontent_encoding content_typeperiodic_task_name task_name task_args task_kwargsworker)rFdefaultsr?)r@ get_or_createitemssetattrsave)r-rSrRrFrOrNrPrQrTrUrVrWrXr@fieldsobjcreatedkvrrr store_resultts($   zTaskResultManager.store_result)NNNNNNNN)rrrrrJrKr rcrrrrrD]srDc@s2eZdZdZdZddZedd d ddZdS) GroupResultManagerz1Manager for :class:`~.models.GroupResult` models.NcCrE)a]Get result for group by ``group_id``. Keyword Arguments: exception_retry_count (int): How many times to retry by transaction rollback on exception. This could happen in a race condition if another worker is trying to create the same task. The default is to retry once. )group_idrG)r-rerrr get_grouprLzGroupResultManager.get_grouprMrc Csb|||d}|s |j}||j||d\}}|s/|D] \} } t|| | q|j|jd|S)N)rOrRrS)rerYr?)r5r@rZr[r\r]) r-rSrRrerOr@r^r_r`rarbrrrstore_group_results  z%GroupResultManager.store_group_resultr0)rrrrrJrfr rgrrrrrdsrd)r )rr) functoolsr itertoolsrcelery.utils.timer django.confr django.dbrrr r utilsr r+ UserWarningr r Managerr!rDrdrrrrs      %P