]j%# UdZddlZddlmZddlmZddlmZddlmZddl m Z ddl m Z dd l mZdd lmZdd lmZdd lmZdd lmZmZmZmZddlmZddlmZddlmZeZ ejBe"Z#dZ$dZ%dZ&dZ'dZ(dZ)e$e jTddddddde%e jVdddd dd!d"e&e jXe jZgd#d$d%de'd&de(e jTd'd(d)dd!dgZ.e/e e0d*<y)+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)flag_set)AllPermissions)start_job_async_or_sync) load_func)DataManagerAction)evaluate_predictions)settings)Project)update_tasks_counters) AnnotationAnnotationDraft PredictionTask)User) WebhookAction)emit_webhooks_for_instancec xt||jdt|jzdzdS)z|Retrieve predictions by tasks ids :param project: project instance :param queryset: filtered tasks db queryset z Retrieved predictionsprocessed_itemsdetail)r countstr)projectquerysetkwargss Q/root/env/lib/python3.12/site-packages/label_studio/data_manager/actions/basic.pyretrieve_tasks_predictionsrs5 "'~~/<#hnnN^J_;_bp;p qqc \t|jd}t|}|Dcgc]}|d }}|jj }t j j|}|jd||k(rZtt j|dtjd|jd|jjntt ||j|j#d d d t%|j&|t(j*|d }|jj-s*|j.j1j3d }t j4||||d t7|zd zdScc}w)znDelete tasks by ids :param project: project instance :param queryset: filtered tasks db queryset idid__inNrlow) queue_namezcalling reset project_id=z delete_tasks()FT)maximum_annotations_changed!overlap_cohort_percentage_changedtasks_number_changedDeleted z tasks)rreloadr)listvalueslentasksrrobjectsfilterupdater*delete_tasks_without_signals_from_task_idsloggerinfor"summaryreset#async_project_summary_recalculationupdate_tasks_statesr organizationr TASKS_DELETEDexistsviewsalldeleteafter_bulk_delete_actionsr) rrr tasks_idsrtasktasks_ids_list project_countr,s r delete_tasksrF"sk X__T*+I  NE-67Td4j7N7MM'')M||""."9H OODO!  O OQ_lqr / |?KL C^U\U_U_` $)Uim w33Wm>Y>Y[deF ==   ! ""$ "">7;$*sSXzBY\dBd ee?8s F)c p|d}|jjd}|jdd}tjj |}|r|j t |}tt|jdd}t|jd}|jj|tjj |} |r| j t | } |jj| |j} |j| jt!|j"|t$j&||d}t(jj | } | j+t-j.|j0 |j3| t5t6j8} | *t(jj | } | || fi|| d t;| zdzdS)zDelete all annotations and drafts by tasks ids :param project: project instance :param queryset: filtered tasks db queryset request annotatorr"Tflat task__id__in) completed_bytask__id)userr#) updated_at updated_by)tasks_querysetr+z annotationsr)dataget values_listr r1r2intsetr-r.r7%remove_created_annotations_and_labelsr remove_created_drafts_and_labelsrr@rr;rANNOTATIONS_DELETEDrr3rnowrP$update_tasks_counters_and_is_labeledrr $DELETE_TASKS_ANNOTATIONS_POSTPROCESSr) rrrrH annotator_idtask_ids annotations real_task_idsannotations_idsdraftsrr0 postprocesss rdelete_tasks_annotationsrfLs YG<<##K0L##Dt#4H$$+++BK!((c,6G(H [44Zd4KLMM;--d34O OO99+F $ $ + + + BFC $56 OO44V<    E MMOw33Wm>_>_apqYG LL  }  5E LLHLLNw||LD 00 0NHIIJK ##8#4GU-f-$ SZ0G.0X YYr c ttjj|j dd}tt jj|j dd}t jj||z}dd d d |Dcgc]D}t|j|jxs|jxs |jd Fc}d ddgdgScc}w)Nr%rNTrJ) task__projectrPr#selectrI Annotator)valuelabelAll)typenamermoptions placeholder searchable) columnCountfields) r-r r1r2rVrrrr" get_full_nameusernameemail)rPr annotator_idsdraft_annotator_idsuserss rdelete_tasks_annotations_formr|{s++2272COOP^eiOjkM66==G=T``agnr`st LL  }7J'J  KE%'(%*  #&dgg,9K9K9M9lQUQ^Q^9lbfblblm $)"&     s A C4 c |jdd}tjj|}t d|j r0|j jddj}n%tt|jdd}|j}|jtttjj||d t|zd zd S) z~Delete all predictions by tasks ids :param project: project instance :param queryset: filtered tasks db queryset r"TrJrL*fflag_root_223_optimize_delete_predictions)r;task_idr#r+rr)rVrr1r2rr;order_bydistinctrXr-rr@rr rr)rrrr` predictionsrbrs rdelete_tasks_predictionsrs ##Dt#4H$$+++BK<7K_K_`#,,.::94:PYY[ D!8!8!8!NOP    E14<<3F3Fm3F3\]$ SZ0G.0X YYr cVtjj|}tjj |}|j j tjj||j j|tj|y)Nr#)r")task__in) rr1r2r rUr7rYr remove_data_columnsdelete_tasks_without_signals)rD project_idrrs rr9r9sw||""."9Hoo!!Z!0G OO99*:L:L:S:S]e:S:fg OO''1%%h/r zRetrieve 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)titletextro) entry_point permissionrorderdialogz Delete TasksdTzGYou are going to delete the selected tasks. Please confirm your action.)rro)rrrrr,rzDelete AnnotationsezYou are going to delete annotations from the selected tasks. You can select specific annotators to delete annotations for. Please confirm your action.)rroformzDelete Predictionsfz\You are going to delete all predictions from the selected tasks. Please confirm your action.actions)1__doc__loggingrcore.feature_flagsrcore.permissionsr core.redisrcore.utils.commonrdata_manager.actionsrdata_manager.functionsr django.confr projects.modelsr tasks.functionsr tasks.modelsr rrr users.modelsrwebhooks.modelsrwebhooks.utilsrall_permissions getLogger__name__r5rrFrfr|rr9predictions_any tasks_delete tasks_changeannotations_deleterr-__annotations__r rrs9'+.'27 #1FF)5 "   8 $r'fT,Z^0Z&02%55'+*  $%22] 0&33_5W5WX%*1  0%55%r O1$ 1r