]j1 UdZddlZddlZddlZddlmZddlmZddl m Z ddl m Z ddl mZddlmZdd lmZmZdd lmZej,eZeZd Zd Zd ZdZdZdZdZ dZ!dZ"eejFdddde zde"ddeejHdddddeddeejHdddddeddgZ%e&e e'd<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) fast_first)DataManagerAction)DataManagerException)settings)ValidationError) AnnotationTask)TaskSerializerBulkc f|d}|j}|jjd}tjj ||}|st d|d|j}t|jdd} |j|jjg} |D]d} | |j|j|j|j|d } t!j"| |d } | j%tdi| ftjj'| t(j* } t!j,|| d |j/t0jj |  ddt3| ddS#t$rYwxYw)Nrequestsource_annotation_id)projectidzSource annotation z! not found in the current projectrTflat)task_idcompleted_by_idresult result_countparent_annotation_idrpropagated_annotation) batch_size)id__in)tasks_querysetzCreated z annotations response_codedetail)userdatagetr objectsfilterrfirstset values_listremovetaskrKeyErrorrrr add_annotation_fieldsappend bulk_creater BATCH_SIZEpost_process_annotations$update_tasks_counters_and_is_labeledr len) rquerysetkwargsr r!r annotationssource_annotationtasksdb_annotationsibodys X/root/env/lib/python3.12/site-packages/label_studio/data_manager/actions/experimental.pypropagate_annotationsr<sYG < +    s %F$$ F0/F0cttjj|}ddd|rdt |j dndzd}d |gd gS) NrnumberrzEnter source annotation IDz [first ID: ]typenamelabel columnCountfields)rr r$r%strr)r!rfirst_annotationfields r;propagate_annotations_formrM<sf!*"4"4";";G";"LM&-9I\#.11231 5rS E % 1 22c |d}|jjd}|jjd}|jjd}|j}||vrtd|z||dj }t j j|} tjtjk(r#| j|j|} n+| jd |ig jd ||giig } d } d } | D]} d } | jD]}|jd d|k(s||jd ij|gvs>g}|d |D]1}||k(r|j|| dz } d} !|j|3||d |<| s| jdg| dz } tjd|j d|j"j%|j"j'|j(j+t j j|} |j"j-| dd| d| dS)Nr old_label_namenew_label_name control_tagz6Wrong old label name, it is not from labeling config: rCr>)result__icontains from_name)result__containsvaluerFrFTr) update_fieldszcalling reset project_id=z rename_labels()rUpdated z labels in r)r"r#get_parsed_configrlowerr r$r%r DJANGO_DBDJANGO_DB_SQLITErr-saveloggerinforsummaryresetupdate_data_columnsr7all%update_created_annotations_and_labels)rr3r4r rPrQrRlabels label_typer5 label_countannotation_count annotationchangedsub new_labelsrEs r; rename_labelsrmGsYG\\%%&67N\\%%&67N,,""=1K  & & (F& VYgghh $V,224J$$++G+  \*51E.")).9#q( "&"))%0 1,6G Z( 6  OO8*O 5  ! )". KK+GJJ<7GHI OO OO'' (9(9(;<$$++G+> D((D(9??AMU# >D  $$ $ D U #E\!xuj9   !:!: :v./E .(- *% . LL $ $UF8 $ M OOfI:,'%-(    \\((M(:@@B  ! OO / / = htfF,C DDrNc|jdd}}||k7r[|||djdzdz}|d||||jddz||dz}|||djdzdz}||k7r[|S)N[r@rFr,;)findreplace)paramsstartends r;process_arraysrsS!23E 3,fUVn))#..2%6%#4#<#, ) or choices(["", "", ...], [, , ...]) or replace("old-string", "new-string")c|jd\}}t|}|jddjd}t|dk(r |ddk(rgn|}t |D]*\}}|jddjdd ||<,t |j d }|d k(rYt|dk(sJd t|d} t| | |z} t | D]\}} | ||j|<n|d k(r\t|dk(sJdtjtd||} t | D]\}} | ||j|<n|dk(rit|dk(sJdt|dt|d} } t|D](}tj| | ||j|<*nC|dk(rdt|cxkr dksJdJdt|dk(rtj|dnd}tjtj|d||} t | D]\}} | ||j|<n|dk(rt|dk(sJdtj|dtj|d}}|D]=}||jvs|j|j|||j|<?nt!dt"zt$j&j)|d gdy)N()rArrFrr'"r"rangez,range(start:int) should have start argument samplezsample() doesn't have argumentsrandomz.random(min, max) should have 2 args: min & maxchoicesz[choices(values:list, weights:list) should have 1 or 2 args: values & weights (default=None)) populationweightskrzOreplace(old_value:str, new_value:str) should have 2 args: old value & new valuez#Undefined expression, you can use: rr)splitrrr2 enumeratertrintrr"rrrandintjsonloadsrradd_data_field_examplesr r$r)r3rrVr~commandargsr9argr7rvaluesvminimummaximumr old_value new_valuer*s r;rrs,KK$MGT $ D <<R & &s +DTaDGrM2tDD/:3++c3'//S9Q: v& 'E'4yA~MMM~DG uedl+f% *DAq()E!HMM* % * H 4yA~@@@~uQ~t4f% *DAq()E!HMM* % * H 4yA~OOO~tAw<T!Wt IA(.w(HE!HMM* % I I 3t9 q  l   l  *-Ta$**T!W%T4::d1g+>SWXf% *DAq()E!HMM* % * I 4yA~ppp~#zz$q'2DJJtAw4G9  \DTYY&(, *(=(E(EiQZ([ *% \ CF]]^^LLUF8ErNc2ddddddddgd d dd d dgd gS)NrFrrr~NamerBrorTyperrprVr}rGr )r!rs r;add_data_field_formrsB ,H$(#A  !'GD   rNzAdd Or Modify Data FieldrFTzConfirm that you want to add a new field in tasks. After this operation you must refresh the Data Manager page fully to see the new column! You can use the following expressions: confirm)textrCform) entry_point permissiontitleorder experimentaldialogzPropagate AnnotationszConfirm that you want to copy the source annotation to all selected tasks. Note: this action can be applied only for similar source objects: images with the same width and height, texts with the same length, audios with the same durations.z Rename LabelszConfirm that you want to rename a label in all annotations. Also you have to change label names in the labeling config manually.actions)(__doc__loggingrujsonrcore.permissionsr core.utils.dbrdata_manager.actionsrdata_manager.functionsr django.confrrest_framework.exceptionsr tasks.modelsr r tasks.serializersr getLogger__name__r^all_permissionsr<rMrmrxrrrrrprojects_change tasks_changerrt__annotations__r rNr;rs% +$27 5)0   8 $ "$N36r>/Ed*7Ft(&%55+68OP'  -%22(. . "%%22 S&  ?,$ ,rN