U d@sdZddlZddlmZddlmZddlmZddlm Z ddl m Z m Z m Z mZdd lmZmZd ZGd d d eZdd dZGddde jZGdddeZGdddeZdS)zModel managers.Nwraps)count)maybe_timedelta)settings) connectionsmodelsrouter transaction)now raw_deletez 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__rrB/tmp/pip-unpacked-wheel-oj8imec1/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]4}z||WStk rF||krBYqXqdS)NZexception_retry_countr)popr Exception)argskwargsZ _max_retriesretries)fun max_retriesrr_inner&s  z1transaction_retry.._outer.._innerr)rrr)rr_outer$s z!transaction_retry.._outerr)rrrrrtransaction_retrys rc@s@eZdZdZddZddZddZdd Zd d Zd d Z dS) ResultManagerz#Generic manager for celery results.cCsPd|krL|}|drL|d}|dkrLttt dS)NZmysqlzPSHOW VARIABLES WHERE variable_name IN ('tx_isolation', 'transaction_isolation');r zREPEATABLE-READ) current_enginelowerconnection_for_readcursorexecuteZfetchonewarningswarnrW_ISOLATION_REPstrip)selfr$Z isolationrrrwarn_if_repeatable_read<s    z%ResultManager.warn_if_repeatable_readcCstt|jSN)rr Z db_for_writemodelr*rrrconnection_for_writeHsz"ResultManager.connection_for_writecCs t|jSr,)rdbr.rrrr#Ksz!ResultManager.connection_for_readcCs2ztj|jdWStk r,tjYSXdS)NZENGINE)rZ DATABASESr0AttributeErrorZDATABASE_ENGINEr.rrrr!NszResultManager.current_enginecCs|jtt|dS)zGet all expired results.)Z date_done__lt)filterr rr*expiresrrrget_all_expiredTszResultManager.get_all_expiredc Cs.tj|jdt||dW5QRXdS)zDelete all expired results.using)ZquerysetN)r Zatomicr0r r5r3rrrdelete_expiredXszResultManager.delete_expiredN) rrrrr+r/r#r!r5r8rrrrr 9s r c @s0eZdZdZdZddZeddd ddZdS) TaskResultManagerz5Manager for :class:`celery.models.TaskResult` models.NcCsPz|j|dWS|jjk rJ|j|kr4|||_|j|dYSXdS)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_idNgetr-Z DoesNotExist_last_idr+)r*r:rrrget_taskcs  zTaskResultManager.get_taskrc Csf|||||||| | | | d }|| j||d\}}|sb|D]\}}t|||q@|j| d|S)arStore 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)r:defaultsr6)r7 get_or_createitemssetattrsave)r*rErDr:rAr@rBrCrFrGrHrIrJr7fieldsobjcreatedkvrrr store_resultvs(&   zTaskResultManager.store_result)NNNNNNNN)rrrrr=r>rrUrrrrr9^sr9c@s0eZdZdZdZddZeddd ddZdS) GroupResultManagerz6Manager for :class:`celery.models.GroupResult` models.NcCsPz|j|dWS|jjk rJ|j|kr4|||_|j|dYSXdS)awGet 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_idNr;)r*rWrrr get_groups  zGroupResultManager.get_groupr?rc Csb|||d}|s|j}||j||d\}}|s^|D]\} } t|| | q:|j|jd|S)N)rArDrE)rWrKr6)r0r7rLrMrNrO) r*rErDrWrAr7rPrQrRrSrTrrrstore_group_results  z%GroupResultManager.store_group_result)N)rrrrr=rXrrYrrrrrVs rV)r )rr& functoolsr itertoolsrZcelery.utils.timerZ django.confrZ django.dbrrr r utilsr r r( UserWarningrrManagerr r9rVrrrrs     %S