]jQ dZddlmZmZmZmZmZmZmZddl m cm Z ddl mZddlmZddlmcmZddlmcmZddZdZdZd Zdd Zd e j:d eeeeeefefd dfdZdedeee e!efd e!fdZ"dZ#ddeeeeeefefd e j:fdZ$dZ%ddZ& ddeeefdeeeefd eeeeeefeffdZ'y) )DictOptionalListTupleAnyCallableUnionN)minidom)LabelStudioTagc "||t|dfS)z Converts labels to a tuple structure. Args: labels: The labels to be converted. tag_type: Tag type to assign to each label. Returns: A tuple containing labels as per the tag type. Labeltag_type_convert_to_tuple)labelsrkwargss Q/root/env/lib/python3.12/site-packages/label_studio_sdk/label_interface/create.pyrrs f/I JJc g}|D][}t|tr3t|^}}|d|dt|f}|j|F|jdd|iif]d|t|fS)z Constructs a taxonomy. Args: labels: The labels to construct the taxonomy. Returns: A tuple containing the taxonomy. rChoicevalueTaxonomy) isinstancetuplerappend)rrresultargparentchildrenparent_childrens rtaxonomyr$sF: c5 ! 1# 6 FX%ay&)U8_EO MM/ * MM8gs^R8 9 : f ..rc d|t|fS)z Constructs a choices tuple. Args: labels: The labels to be converted. Returns: A tuple containing the choices. Choicesr)rrs rchoicesr'2s v08 99rc|tj|d}tj|}|j dS)z> Returns a pretty-printed XML string for the Element. zutf-8z )indent)ETtostringr parseString toprettyxml)element rough_stringreparseds r _prettifyr1?s8;;w0L""<0H   t  ,,rc2|sytfd|DS)z Converts arguments to a tuple structure of choices. Args: args: The labels to be converted. tag_type: Tag type to assign to each label. Returns: A tuple containing all labels in specified format. Nc3,K|] }d|iif yw)rN).0r rs r z$_convert_to_tuple..VsBc8gs^R0Bs)r)argsrs `rrrHs  BTB BBrr!elementsreturnc2|D]}t|trtj||*t|tr+|j D]\}}|j ||et|tst|tst||y)z Adds nested elements to the parent element in the Element Tree. Args: parent: The parent ET.Element object. elements: A list containing element definitions. )r!N) rstrr* SubElementdictitemssetlistrcreate_element)r!r8r.kvs r_create_nested_elementsrDYsy3 gs # MM&' *  &  !1 1a  !  &*We*D 76 23rnametagc (t|tr[|jxs||_|jdvrdnd}|j|j t t |dd|f}nOt|ttfrt|dkrg|dn|}n!t|tr|idf}n td |d jd ||d jtjvr|d j!d s d|z|d d <|S)aS Converts tags from str, list, tuple, or class instance to a tuple format. Args: name: Name of the tag. tag: Tag in str, list, tuple or class instance format. Returns: A tuple version of the input tag. Raises: TypeError: If input tag is not a str, list, tuple, or LabelStudioTag. )r&rrrrNrr4z9Input tag must be one of str, list, tuple, LabelStudioTagrrErr$)rr rErF collect_attrsrgetattrr@rlenr; TypeError setdefaultlowerOT _TAG_TO_CLASSget)rErFchild_tag_typeels r_convertrUks#~&88#t%(WW0G%GW WWc'')+rUrOrPrQ)tagsrErF tag_tupletag_names r_find_first_object_tagrZsMZZ\ cT3' Q< >> r// /  rr.c|\}}}|tj||ntj|}|jD]\}}|j |||r t |||S)z Creates an XML element. Args: element: List of tag, attributes and children to feed into the element. parent: Parent element to append the new element to (Optional). Returns: An ElementTree element. )r*r<Elementr>r?rD)r.r!rFattrsr"rTrBrCs rrArAsi#C'-'9vs #rzz#B 1 q! H- Ircvtjd}|D]}t|}|j| |S)z Creates an ElementTree from the input element definitions, and return a string for it. Args: elements: Variable length argument list of elements to add to the ElementTree. Returns: XML configuration View)r*r\rAr)r8viewr.rTs rtree_from_tuplesras= ::f D G $ B Krcn|rt|}|jddStj|dS)z z unicode)encoding)r1replacer*r+)treeprettypps rtree_to_stringrjs3 t_zz4b99{{4)44rrWmappingcvg}t|}|jD]\}}t||}|djtj vrSd|dvs|ddD|r4|dj d|vr|j |dd|dd<n|dd|dd<|j||S)z Convert tags into a structured format. Args: tags: A dictionary of tags. mapping: Optional mapping of tag name transformations. Returns: A list of the structured tags. rtoNamerrE)rZr>rUrOCTrQrRr)rWrkr8first_object_tagrErFrTs rconvert_tags_descriptionrpsH-d3ZZ\ c dC  a5;;=B,, ,("Q%2G2a5QY?Kb2a599V,7")++beFm"<1h"21"5f"=1h Or)Labels)r)N)T)(__doc__typingrrrrrr r xml.etree.ElementTreeetree ElementTreer*xml.domr %label_studio_sdk.label_interface.baser ,label_studio_sdk.label_interface.object_tagslabel_interface object_tagsrP-label_studio_sdk.label_interface.control_tags control_tagsrnrr$r'r1rr\r;rDr@rrUrZrArarjrpr4rrr~sCDDD""@99:: K/, :-C"3BJJ3$uS$sTWx.Z^E^?_:`3ei3$3U3e^#CDD$DsDcND'@!ABTVT^T^.$5@D4S>&.tCH~&>HLUSVX\]`be]eXfhlSlMmHnr