]j7 LdZddlZddlmZmZmZmZddlmZddl m Z ddl m Z ddl mZmZej eZe Zdded ed eeeeffd Zd eed eeefd efdZd eed eeeffdZd eed eeefd eeeeffdZded eeeeffdZy)z Utility functions for working with the declarative transition system. This module provides helper functions to make it easier to integrate the new Pydantic-based transition system with existing Label Studio code. N)AnyDictListType)Model)transition_registry)get_state_manager)BaseTransitionTransitionValidationErrorentityvalidatereturnc |jjj}tj|}|s|Si}|j D]\}} t j|}|r |jnd} ddl m }  |} | |||| dt|dd} | j| } | |||| | t|dd}|j|r|||<|S#ttf$r|||<YwxYw#t $rYt"$rE}t$j'dd||jjt)|dd Yd}~d}~wwxYw) a Get available transitions for an entity. Args: entity: The entity to get transitions for user: User context for validation (only used when validate=True) validate: Whether to validate each transition against current state. When False, returns all registered transitions for the entity type. When True, filters to only transitions valid from current state (may be expensive). Returns: Dictionary mapping transition names to transition classes. When validate=False: All registered transitions for the entity type. When validate=True: Only transitions valid for the current state. NTransitionContextorganization_idr current_usercurrent_state_object current_state target_staterz&Unexpected error validating transitionzfsm.transition_validation_error)eventtransition_name entity_typeerrorT)extraexc_info)_meta model_namelowerrget_transitions_for_entityitems StateManagerget_current_state_objectstate transitionsrgetattrget_target_state TypeError ValueErrorcan_transition_from_stater Exceptionloggerwarningstr)r userr entity_name availablevalid_transitionsnametransition_classrrr temp_instanceminimal_contextrcontextes K/root/env/lib/python3.12/site-packages/label_studio/fsm/transition_utils.pyget_available_transitionsr<s ,,))//1K#>>{KI "+//"398 #/#H#H#P :N066TXM 7  0 2 "3!!%)="/!%$+F4Et$L # -==oN (!%9+) '0A4 H G 99'B*:!$'Q9v Iz* +;!$'  &)    NN8>'+#)<<#:#: V      sB+D1C'4/D'C>;D=C>>D E E:EEr6datacl |di|S#t$r }td|jd|d}~wwxYw)am Create a transition instance from a dictionary of data. This handles Pydantic validation and provides clear error messages. Args: transition_class: The transition class to instantiate data: Dictionary of transition data Returns: Validated transition instance Raises: ValueError: If data validation fails zFailed to create z: N)r-r+__name__)r6r=r:s r;create_transition_from_dictrAksI O'$'' O,-=-F-F,Gr!MNNOs 3.3c"|jS)z Get the JSON schema for a transition class. Useful for generating API documentation or frontend forms. Args: transition_class: The transition class Returns: JSON schema dictionary )model_json_schema)r6s r;get_transition_schemarDs  - - //c$ |di|iS#t$rz}i}t|drQ|jD]=}djd|dD}||vrg||<||j |d?nt |g|d<|cYd}~Sd}~wwxYw) z Validate transition data without creating an instance. Args: transition_class: The transition class data: Data to validate Returns: Dictionary of field names to error messages (empty if valid) errors.c32K|]}t|yw)N)r0).0locs r; z+validate_transition_data..s BcS BsrKmsg__root__Nr?)r-hasattrrGjoinappendr0)r6r=r:rGrfields r;validate_transition_datarSs 4    1h  3 BU5\ BB&$&F5Mu $$U5\2  3 #&a&F:   s BA/B B Bc ,|jjj}tj|}t }g}|j D]\}} ddlm}|}t|dr||ddddt|ddnd} |j| } |j| |j||j| |jxsdt!|j"j%d|S#t&$rYwxYw) z Get a summary of the state flow for an entity type. This analyzes all registered transitions and builds a flow diagram. Args: entity: Example entity instance Returns: List of state flow information rrpkNrr)rr6r descriptionfields)rr r!rr"setr#r'rrOr(r)addrQr@__doc__list model_fieldskeysr-) r r2r'statesflowsrr6r transitionr8targets r;get_entity_state_flowrcs$,,))//1K%@@MKUF E-8->->-@ ))  6)+J64("!!%)-"&!%$+F4Et$L   00AF JJv  LL'6(8(A(A$*#3#;#;#Ar"#3#@#@#E#E#GH  - D L   sB&D DD)NF)r[loggingtypingrrrrdjango.db.modelsr fsm.registryrfsm.state_managerr fsm.transitionsr r getLoggerr@r.r$boolr0r<rArDrSrcr?rEr;rls((",/E   8 $ " SeS$SSWX[]abp]qXqSrSlO$~2FOdSVX[S[nOaoO, 0D,@ 0T#s(^ 0tN/C4PSUXPX>^bcfhlmphqcq^r85%5Dc3h,@5rE