o Mf @s8dZddlZddlmZddlmZddlmZddlmZddl m Z ddl m Z dd l mZdd lmZdd lmZmZmZmZdd lmZdd lmZeZeeZddZddZddZ ddZ!ddZ"eej#dddddddeej$ddd d!dd"d#e ej$d$d%d&dd"de!ej#d'd(d)dd"dgZ%dS)*zThis file and its contents are licensed under the Apache License 2.0. Please see the included NOTICE for copyright information and LICENSE for a copy of the license. N)datetime)AllPermissions)start_job_async_or_sync) load_func)evaluate_predictions)settings)Project)update_tasks_counters) AnnotationAnnotationDraft PredictionTask) WebhookAction)emit_webhooks_for_instancecKs&t||dt|ddS)z|Retrieve predictions by tasks ids :param project: project instance :param queryset: filtered tasks db queryset z Retrieved predictionsprocessed_itemsdetail)rcountstr)projectquerysetkwargsri/var/www/html/analyze/labelStudio/lib/python3.10/site-packages/label_studio/data_manager/actions/basic.pyretrieve_tasks_predictionssrcKst|d}t|}dd|D}|j}tjj|d}|jdd||kr4t tj ||j nt t ||j|jdddd t|j|tj|d}|js\|jd}t|||d t|d d S) znDelete tasks by ids :param project: project instance :param queryset: filtered tasks db queryset idcSsg|]}|dqSrr).0taskrrr 'sz delete_tasks..id__inN)rFT)maximum_annotations_changed!overlap_cohort_percentage_changedtasks_number_changedDeleted z tasks)rreloadr)listvalueslentasksrr objectsfilterupdater*delete_tasks_without_signals_from_task_idssummaryreset#async_project_summary_recalculationrupdate_tasks_statesr organizationr TASKS_DELETEDexistsviewsalldeleteafter_bulk_delete_actionsr)rrr tasks_idsrtasks_ids_list project_countr'rrr delete_taskss(      r>c Ks|jddd}tjj|d}|}tt|jddd}t|d}|j |t jj|d}|j || | t |j|tj||d} tjj|d} | jt| jd|j|d ttj} | d ur|tjj|d} | || fi||d t|d d S)zDelete all annotations and drafts by tasks ids :param project: project instance :param queryset: filtered tasks db queryset rTflat task__id__intask__idrequestr!) updated_at updated_by)tasks_querysetNr&z annotationsr) values_listr r,r-rsetr(r)r0%remove_created_annotations_and_labelsr remove_created_drafts_and_labelsr9rr4rANNOTATIONS_DELETEDr r.rnowuser$update_tasks_counters_and_is_labeledrr$DELETE_TASKS_ANNOTATIONS_POSTPROCESSr) rrrtask_ids annotationsr real_task_idsannotations_idsdraftsrDr+ postprocessrrrdelete_tasks_annotationsHs(    rWcKsl|jddd}tjj|d}tt|jddd}|}|tt t jj|d|dt |dd S) z~Delete all predictions by tasks ids :param project: project instance :param queryset: filtered tasks db queryset rTr?rArCr!r&rr) rHr r,r-rIr(rr9rr r r)rrrrQ predictionsrSrrrrdelete_tasks_predictionsosrYcCsLtjj|d}tjj|d}|jtjj|d|j|t |dS)Nr!r)task__in) r r,r-rgetr0rJr remove_data_columnsdelete_tasks_without_signals)r< project_idrrrrrr2~s  r2zRetrieve PredictionsZzSend the selected tasks to all ML backends connected to the project.This operation might be abruptly interrupted due to a timeout. The recommended way to get predictions is to update tasks using the Label Studio API.Please confirm your action.confirm)titletexttype) entry_point permissionraorderdialogz Delete TasksdTzGYou are going to delete the selected tasks. Please confirm your action.)rbrc)rdrerarfr'rgzDelete Annotationsez\You are going to delete all annotations from the selected tasks. Please confirm your action.zDelete Predictionsfz\You are going to delete all predictions from the selected tasks. Please confirm your action.)&__doc__loggingrcore.permissionsr core.redisrcore.utils.commonrdata_manager.functionsr django.confrprojects.modelsrtasks.functionsr tasks.modelsr r r r webhooks.modelsrwebhooks.utilsrall_permissions getLogger__name__loggerrr>rWrYr2predictions_any tasks_deleteactionsrrrrsn            )'