jgvdZddlZddlmZddlmZddlmZddlm Z ddl m Z m Z m Z mZdd lmZd ZGd d eZdd ZGdde j*ZGddeZGddeZy)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. ceZdZdZy)TxIsolationWarningzAWarning emitted if the transaction isolation level is suboptimal.N)__name__ __module__ __qualname____doc__Y/var/www/html/Stanley/venv/lib/python3.12/site-packages/django_celery_results/managers.pyrrsKrrcfd}|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. c2tfd}|S)Nc|jd}tdD] } |i|cSy#t$r ||k\rY!wxYw)Nexception_retry_countr)popr Exception)argskwargs _max_retriesretriesfun max_retriess r_innerz1transaction_retry.._outer.._inner%sZ!::&={KL 8 ///$! ,./ s/AAr)r!r#r"s` r_outerz!transaction_retry.._outer#s s    rr)r"r$s` rtransaction_retryr%s$ Mrc4eZdZdZdZdZdZdZdZdZ y) ResultManagerz#Generic manager for celery results.c8d|jjvrz|jj}|j drJ|j d}|dk(r1t jttjyyyy)NmysqlzPSHOW VARIABLES WHERE variable_name IN ('tx_isolation', 'transaction_isolation');r zREPEATABLE-READ) current_enginelowerconnection_for_readcursorexecutefetchonewarningswarnrW_ISOLATION_REPstrip)selfr- isolations rwarn_if_repeatable_readz%ResultManager.warn_if_repeatable_read;s d))+113 3--/668F~~KL"OO-a0  11MM"4_5J5J5L"MN2L 4rcNttj|jSN)rr db_for_writemodelr4s rconnection_for_writez"ResultManager.connection_for_writeGs6..tzz:;;rc(t|jSr8)rdbr;s rr,z!ResultManager.connection_for_readJs477##rc tj|jdS#t$rtjcYSwxYw)NENGINE)r DATABASESr>AttributeErrorDATABASE_ENGINEr;s rr*zResultManager.current_engineMs< ,%%dgg.x8 8 ,++ + ,s ">>cN|jtt|z S)zGet all expired results.) date_done__lt)filterr rr4expiress rget_all_expiredzResultManager.get_all_expiredSs{{1I)I{JJrctj|j5|j|j dddy#1swYyxYw)zDelete all expired results.usingN)r atomicr>rIdeleterGs rdelete_expiredzResultManager.delete_expiredWs7   dgg .   ) 0 0 2/ . .s A  AN) rrrrr6r<r,r*rIrOrrrr'r'8s%- O<$, K3rr'cBeZdZdZdZdZed ddZy)TaskResultManagerz0Manager for :class:`~.models.TaskResult` models.Nc |j|S#|jj$r;|j|k(r|j ||_|j|cYSwxYw)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_idgetr: DoesNotExist_last_idr6)r4rSs rget_taskzTaskResultManager.get_taskbsa /88G8, ,zz&& /}}',,.#DM::g:. .  /AA('A(r"c |||||||| | | | d }|j| j||\}}|s7|jD]\}}t||||j | |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)rSdefaultsrK)rL get_or_createitemssetattrsave)r4rbrarSr^r]r_r`rcrdrerfrgrLfieldsobjcreatedkvs r store_resultzTaskResultManager.store_resulttsH" 0("4""&  zz%(66w@F7H W 1Q"' HH5H ! r)NNNNNNNN)rrrrrWrXr%rrrrrrQrQ]s6:H/$1%+/(,AE(, 5&5rrQc<eZdZdZdZdZed ddZy)GroupResultManagerz1Manager for :class:`~.models.GroupResult` models.Nc |j|S#|jj$r;|j|k(r|j ||_|j|cYSwxYw)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_idrT)r4rvs r get_groupzGroupResultManager.get_groupsa 188X8. .zz&& 1}}(,,.$DM::x:0 0  1rYrZr[c|||d}|s |j}|j|j||\}}|sA|jD]\} } t || | |j |j|S)N)r^rarb)rvrhrK)r>rLrirjrkrl) r4rbrarvr^rLrmrnrorprqs rstore_group_resultz%GroupResultManager.store_group_results 0(  GGEzz%(66@F7H W 1Q"' HH477H # rr8)rrrrrWrwr%ryrrrrtrts+;H1$1%37&rrt)r )rr0 functoolsr itertoolsrcelery.utils.timer django.confr django.dbrr r r utilsr r2 UserWarningrr%Managerr'rQrtrrrrsf- >>LL>"3FNN"3JM M`))r