o Mf#. @s dZddlZddlZddlZddlmZddlmZddl m Z ddl m Z m Z ddlmZeeZeZdd Zd d Zd d ZddZddZddZdZddZddZeejddddededdeejd ddd!deddeejd"ddd#deddgZdS)$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)DataManagerException)settings) AnnotationTask)TaskSerializerBulkc Ks|d}|j}|jd}tjj||d}|std|d|}t|j ddd}z | |j j Wn t y>Ynwg} |D] } | |j |j|j|j |d } t| |d } | tdi| qCtjj| tjd } t|| d |jtjj|d d ddt| ddS)Nrequestsource_annotation_id)projectidzSource annotation z! not found in the current projectr T)flat)task_idcompleted_by_idresult result_countparent_annotation_idr propagated_annotation) batch_size)id__in)tasks_querysetzCreated z annotations response_codedetail)userdatagetrobjectsfilterrfirstset values_listremovetaskr KeyErrorrrradd_annotation_fieldsappend bulk_creater BATCH_SIZEpost_process_annotations$update_tasks_counters_and_is_labeledrlen) r querysetkwargsrrr annotationssource_annotationtasksdb_annotationsibodyrrp/var/www/html/analyze/labelStudio/lib/python3.10/site-packages/label_studio/data_manager/actions/experimental.pypropagate_annotationss<  r6cCsFtjj|d}ddd|rdt|jdndd}d |gd gS) Nr numberr zEnter source annotation IDz [first ID: ]typenamelabel columnCountfields)rrrr strr )rr first_annotationfieldrrr5propagate_annotations_form9srFcKs|d}|jd}|jd}|jd}|}||vr$td|||d}tjj|d} tj tj krD| j|dj|d} n| jd |igd jd ||giigd } d } d } | D]Q} d } | j D]<}|d d|kr||d i|gvrg}|d |D]}||kr| || d7} d} q| |q||d |<qd| r| j dgd| d7} q]|j|j|jtjj|d} |j| dd| d| dS)Nrold_label_namenew_label_name control_tagz6Wrong old label name, it is not from labeling config: r<r7)result__icontains from_name)result__containsvaluerFr?Tr) update_fieldsrUpdated z labels in r)rrget_parsed_config Exceptionlowerrrrr DJANGO_DBDJANGO_DB_SQLITErr'savesummaryresetupdate_data_columnsr1all%update_created_annotations_and_labels)r r-r.rrGrHrIlabels label_typer/ label_countannotation_count annotationchangedsub new_labelsr>rrr5 rename_labelsDsT           rccCsr|}g}g}|D]\}}||dg7}||q dddd|ddddtt|dd d d d gd gS)Nr[r?selectrIzChoose a label control tagr<r=r>optionsrGzOld label nameinputrHzNew label namer;r@)rPitemsrr'listr!)rr r[ old_names control_tagskeyr>rrr5rename_labels_form|s*   rmcKsddlm}m}m}m}|d}|jd}|jd} |jd} |} tt td} | | vs4J| | | } | dkrFt || | |n7t j t jkrht|d } | D]}| |j|<qUtjj| d gd d n|j||d ||g|| |d d d|j|gdd| ddS)Nr)FFunc JSONFieldValuer value_name value_typerMStringNumber ExpressionrwrrBr jsonb_set)function)rrrOz tasksr)django.db.modelsrnrorprqrrcountrCfloatkeysadd_expressionrrSrTrionlyrr bulk_updateupdaterVrXr )r r-r.rnrorprqrrrrsrMsizecastr1r$rrr5add_data_fields4         rcCs|dd}}||kr@|||ddd}|d||||dd||d}|||ddd}||ks |S)N[r9r?r,;)findreplace)paramsstartendrrr5process_arrayss,rzrange(2) or sample() or random(, ) or choices(["", "", ...], [, , ...]) or replace("old-string", "new-string")cCs|d\}}t|}|ddd}t|dkr"|ddkr"gn|}t|D]\}}|dddd ||<q(t|d }|d krlt|dksNJd t|d} t| | |} t| D] \}} | ||j |<q_n|d krt|dkszJdt td||} t| D] \}} | ||j |<qn|dkrt|dksJdt|dt|d} } t|D] }t | | ||j |<qn|dkrdt|krdksJdJdt|dkrt |dnd}t jt |d||d} t| D] \}} | ||j |<qn>|dkr@t|dksJdt |dt |d}}|D]}||j vr=|j ||||j |<q)ntdttjj|d gdddS)N()r:rr?rr'"rrangez,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: rxry)splitrrr, enumerateririntrrrrrandintjsonloadsrrQadd_data_field_examplesrrr)r-rrMrrcommandargsr3argr1rvaluesvminimummaximumr old_value new_valuer$rrr5rs^       rcCs0ddddddddgd d dd d dgd gS)Nr?rgrrNamer;rdrsTypertrerMrqr@r)rr rrr5add_data_field_forms  rzAdd Or Modify Data Fieldr?TzConfirm 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)textr<form) 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.) __doc__loggingrujsonrcore.permissionsrdata_manager.functionsr django.confr tasks.modelsrrtasks.serializersr getLogger__name__loggerall_permissionsr6rFrcrmrrrrrprojects_change tasks_changeactionsrrrr5sl     ' 8(  :