]j UdZddlZddlmZddlmZddlmZddlm Z ddl m Z m Z m Z ejeZeZdZdd Zd Zd Zeej,d d dddeddgZeeed<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)AllPermissions)start_job_async_or_sync)DataManagerAction)LabelInterface) Annotation PredictionTaskc >|d}|jddj}|dvsJd|dk(rtnt}|jdxs|jd}|jdd jd k(}t |j }|j d D cic]} | j| } } |dk(rd } nd } |dk(s|d}| d} n| d|} t|jd} tjdt| d|| D]} g}|jj| jd}|D] }t||| }|j!|"|rSdj#t%t'|Dcgc]}|d|j)|c}| j*| <dj#t%tt'|| j*| <t,jj/| dgdt,jj|j1j2}|j4j7|gddt| ddScc} wcc}w)N request_datasource annotations)r predictionsz)Source must be annotations or predictionscustom_control_tag control_tag with_countersYesyescontrolcachecache_predictionsALL_all_datazCache labels for z tasks and control tag )taskresultz, z: i)fields batch_size)idzUpdated z tasks) response_codedetail)getlowerrrr label_config find_tagsnamelistonlyloggerinfolenobjectsfilterextract_labelsextendjoinsortedsetcountrr bulk_updatefirstrsummaryupdate_data_columns)projectquerysetkwargsr r source_classrrlabel_interfacetaglabel_interface_tags column_nametasksr task_labelsr annotationlabelslabel first_tasks X/root/env/lib/python3.12/site-packages/label_studio/data_manager/actions/cache_labels.pycache_labels_jobrHs.)L   h 6 < < >F 3 3`5`` 3!'=!8:jL""#78[L]^&DIIk " &*YYvd3{;K6L/M%NDIIk "O" LLUF8E!!X^^%5%8%8!9J OO'' 5 hs5zl&,I JJK[8^s 1JJcVg}|jD]}| |d|k(sd|vs|dD]}|d|s t|d|ts#|dk(rd}d}|G|d|vr@||d}|jj dd}|jj dd}|dk(r"|d|D]} |j | d n^|d|D]"} |j |j | $n0t|d|d tr|j|d||S) N from_namevaluetaxonomytrue/ showFullPathfalse pathSeparatorr) r isinstancer(attrr#appendr1strr0) rCrr?rDregionkeyrOrQlabel_interface_tagelemss rGr/r/@sl F##  6+#6+#E7V\K\g '?3'Jvgs7KT,Rj('- (+ /;{@SWk@k2FvkGZ2[/+>+C+C+G+GX_+`L,?,D,D,H,HZ],^M'72)/)=9 & eBi 89*0)=I & m.@.@.G HI$F7OC$8$;SA fWoc&:;/  : Mc \tt|||j|jdddiS)z6Cache labels from annotations to a new column in tasksiPF)organization_idr job_timeoutr!r )rrHr]r)r9r:requestr;s rG cache_labelsr`bs3//\\  S !!r[c |j}dg}|jD]\}}|j|dddd|dddd d dd d d dgddddddgdgdgS)NrselectrzChoose a control tag)typer'rEoptionsinputrz.Custom control tag if it's not in label config)rdr'rErz With countersrNor Source Annotations Predictions) columnCountr)get_parsed_configitemsrU)userr9rD control_tagsrXrs rGcache_labels_formrpos  & & (F7L,,.!QC ! %)3+ $0M %+, %t} %$% -}= %  r[z Cache LabelsrbTzConfirm that you want to add a new task.data field with cached labels from annotations. This field will help you to quickly filter or order tasks by labels. After this operation you must refresh the Data Manager page fully to see the new column!confirm)textrdform) entry_point permissiontitleorder experimentaldialogactions)N)__doc__loggingcore.permissionsr core.redisrdata_manager.actionsr label_studio_sdk.label_interfacer tasks.modelsrrr getLogger__name__r*all_permissionsrHr/r`rpprojects_changerzr(__annotations__r[rGrs+.2;55   8 $ "-K`D "#P$%55g%  $ r[