o wvXj@sdZddlmZmZddlZddlmZddlmZddl m Z m Z m Z ddl m Z ddlmZdd lmZd d lmZzdd lmZWneySdd lmZYnwd ZGdddeZdddZGddde jZdS)zModel managers.)absolute_importunicode_literalsNwraps)count) connectionsrouter transaction)models)settings)items)now)maybe_timedeltaz 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__rr[/var/www/html/myproject/venv/lib/python3.10/site-packages/django_celery_results/managers.pyrsrcsfdd}|S)a Decorate 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)rr r)rr_outer-s z!transaction_retry.._outerr)rr"rr!rtransaction_retry"s r#c@sfeZdZdZdZddZedd   dddZd d Zd d Z d dZ ddZ ddZ ddZ dS)TaskResultManagerz5Manager for :class:`celery.models.TaskResult` models.NcCsJz|j|dWS|jjy$|j|kr|||_|j|dYSw)auGet 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_id)getmodel DoesNotExist_last_idwarn_if_repeatable_read)selfr%rrrget_taskGs  zTaskResultManager.get_taskr!c  Csd|||||||| | | d } || j|| d\}}|s0t| D] \}}t|||q|j| d|S)a1Store 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. 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. Keyword Arguments: ----------------- 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). 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_type task_name task_args task_kwargsworker)r%defaults)using)r9 get_or_creater setattrsave)r+r3r2r%r/r.r0r1r4r5r6r7r9fieldsobjcreatedkvrrr store_resultZs&$   zTaskResultManager.store_resultcCs\d|vr(|}|dr*|d}|dkr,ttt dSdSdSdS)NmysqlzPSHOW VARIABLES WHERE variable_name IN ('tx_isolation', 'transaction_isolation');r zREPEATABLE-READ) current_enginelowerconnection_for_readcursorexecutefetchonewarningswarnrW_ISOLATION_REPstrip)r+rG isolationrrrr*s   z)TaskResultManager.warn_if_repeatable_readcCstt|jSN)rr db_for_writer'r+rrrconnection_for_writesz&TaskResultManager.connection_for_writecCs t|jSrO)rdbrQrrrrFs z%TaskResultManager.connection_for_readcCs,z tj|jdWStytjYSw)NENGINE)r DATABASESrSAttributeErrorDATABASE_ENGINErQrrrrDs   z TaskResultManager.current_enginecCs|jtt|dS)zGet all expired task results.) date_done__lt)filterrrr+expiresrrrget_all_expiredsz!TaskResultManager.get_all_expiredcCs<t||WddS1swYdS)zDelete all expired results.N)r atomicr\deleterZrrrdelete_expireds "z TaskResultManager.delete_expired)NNNNNNN)rrrrr)r,r#rBr*rRrFrDr\r_rrrrr$Bs 6  r$)r )r __future__rrrJ functoolsr itertoolsr django.dbrrr r django.confr celery.fiver utilsrcelery.utils.timer ImportErrorcelery.utils.timeutilsrL UserWarningrr#Managerr$rrrrs&