]j+ dZddlZddlZddlmZmZmZddlmZmZejr ddl m Z ddl m Z ejeZGddZeZd efd Zd efd ZGd d ZeZd efdZd eddfdZd ededfdZdededfdZGddZeZ dd edededededef dZ y)z FSM Model Registry for Model State Management. This module provides a registry system for state models and state choices, allowing the FSM to be decoupled from concrete implementations. N)DictOptionalType)Model TextChoices) BaseState)BaseTransitionc`eZdZdZdZdedeefdZdede eefdZ de efdZ d Z y ) StateChoicesRegistryz Registry for managing state choices for different entity types. Provides a centralized way to register, discover, and manage state choices for different entity types in the FSM system. ci|_yN)_choicesselfs C/root/env/lib/python3.12/site-packages/label_studio/fsm/registry.py__init__zStateChoicesRegistry.__init__s 68 entity_name choices_classc>||j|j<y)z Register state choices for an entity type. Args: entity_name: Name of the entity (e.g., 'task', 'annotation') choices_class: Django TextChoices class defining valid states N)rlower)rrrs rregisterzStateChoicesRegistry.register s.; k'')*rreturncT|jj|jS)z Get state choices for an entity type. Args: entity_name: Name of the entity Returns: Django TextChoices class or None if not found )rgetrrrs r get_choicesz StateChoicesRegistry.get_choices*s"}}  !2!2!455rcHt|jjSz*Get a list of all registered entity types.)listrkeysrs r list_entitiesz"StateChoicesRegistry.list_entities6sDMM&&())rc8|jjy)zp Clear all registered choices. Useful for testing to ensure clean state between tests. N)rclearrs rr$zStateChoicesRegistry.clear:s rN)__name__ __module__ __qualname____doc__rstrrrrrrr r"r$rrr r sV9;C;[8I; 6s 6x[8I/J 6*tCy*rr rc,tj|S)z Get state choices for an entity type. Args: entity_name: Name of the entity Returns: Django TextChoices class or None if not found )state_choices_registryrrs rget_state_choicesr.Gs " - -k ::rcHdttdttffd }|S)a Decorator to register state choices for an entity type. Args: entity_name: Name of the entity type Example: @register_state_choices('task') class TaskStateChoices(models.TextChoices): CREATED = 'CREATED', _('Created') IN_PROGRESS = 'IN_PROGRESS', _('In Progress') COMPLETED = 'COMPLETED', _('Completed') rrc4tj||Sr )r,r)rrs r decoratorz)register_state_choices..decoratorcs'' ]Cr)rrrr1s` rregister_state_choicesr3Ts(k!2tK7H rcheZdZdZdZdeddfdZdededfdZdede fd Z d Z de edffd Z y ) StateModelRegistryz Registry for state models and their configurations. This allows projects to register their state models dynamically without hardcoding them in the FSM framework. ci|_yr )_modelsrs rrzStateModelRegistry.__init__rs /1 rr state_modelrc"|j}||jvr=tjdd||j|j|jd||j|<tjdd||jdy) z Register a state model for an entity type. Args: entity_name: Name of the entity (e.g., 'task', 'annotation') state_model: The state model class for this entity z Overwriting existing state modelzfsm.registry_overwrite)event entity_typeprevious_model new_modelextrazRegistered state modelzfsm.model_registered)r:r; model_nameN)rr7loggerdebugr%)rrr8 entity_keys rregister_modelz!StateModelRegistry.register_modelus!&&(  % LL25#-&*ll:&>&G&G!,!5!5   $/ Z  $/))22  rrcT|jj|jS)z Get the state model for an entity type. Args: entity_name: Name of the entity Returns: State model class or None if not registered )r7rrrs r get_modelzStateModelRegistry.get_models"|| 1 1 344rc:|j|jvS)z2Check if a model is registered for an entity type.)rr7rs r is_registeredz StateModelRegistry.is_registereds  "dll22rcj|jjtjdddiy)z1Clear all registered models (useful for testing).zState model registry clearedr:zfsm.registry_clearedr>N)r7r$rArBrs rr$zStateModelRegistry.clears/  *23  rc6|jjS)zGet all registered models.)r7copyrs rget_all_modelsz!StateModelRegistry.get_all_modelss||  ""rN)r%r&r'r(rr)rDrrFboolrHr$rrLr*rrr5r5jse2 # K > 5S 5Xk-B 5333 #S+%5 6#rr5cdfd }|S)a Decorator to register a state model. Args: entity_name: Name of the entity (e.g., 'task', 'annotation') Example: @register_state_model('task') class TaskState(BaseState): @classmethod def get_denormalized_fields(cls, entity): return { 'project_id': entity.project_id, 'priority': entity.priority } c4tj||Sr state_model_registryrD)r8rs rr1z'register_state_model..decorators++KEr)r8rrrr*r2s` rregister_state_modelrRs$ rr8rc0tj||y)z Convenience function to register a state model programmatically. Args: entity_name: Name of the entity (e.g., 'task', 'annotation') state_model: The state model class for this entity NrP)rr8s rregister_state_model_classrTs'' [Arrc,tj|S)z Convenience function to get a state model. Args: entity_name: Name of the entity Returns: State model class or None if not registered )rQrFr-s rget_state_modelrVs ) )+ 66rentityc`|jjj}t|S)z"Get the state model for an entity.)_metar@rrV)rWrs rget_state_model_for_entityrZs%,,))//1K ; ''rcveZdZdZdZdededdfdZdedededfd Zdede edffd Z de efd Z d Z y )TransitionRegistryz Registry for managing declarative transitions. Provides a centralized way to register, discover, and execute transitions for different entity types and state models. ci|_yr  _transitionsrs rrzTransitionRegistry.__init__s DFrrtransition_nametransition_classr cb||jvri|j|<||j||<y)a6 Register a transition class for an entity. Args: entity_name: Name of the entity type (e.g., 'task', 'annotation') transition_name: Name of the transition (e.g., 'start_task', 'submit_annotation') transition_class: The transition class to register Nr^)rrr`ras rrzTransitionRegistry.registers7 d// /-/D  k *:J+&7rrcX|jj|ij|S)z Get a registered transition class. Args: entity_name: Name of the entity type transition_name: Name of the transition Returns: The transition class if found, None otherwise )r_r)rrr`s rget_transitionz!TransitionRegistry.get_transitions(  $$["599/JJrcV|jj|ijS)z Get all registered transitions for an entity type. Args: entity_name: Name of the entity type Returns: Dictionary mapping transition names to transition classes )r_rrKrs rget_transitions_for_entityz-TransitionRegistry.get_transitions_for_entitys&  $$["5::<Y =.tCy."rr\r`triggers_on_createtriggers_on_update triggers_onforce_state_recordc(dfd }|S)a Decorator to register a state transition class with trigger metadata. This decorator not only registers the transition but also configures when it should be triggered based on model changes. Args: entity_name: Name of the entity type (e.g., 'task', 'project') transition_name: Name of the transition (e.g., 'task_created') triggers_on_create: If True, triggers when entity is created triggers_on_update: If True, can trigger on updates (default: True) triggers_on: List of field names that trigger this transition force_state_record: If True, creates state record even if state doesn't change (for audit trails) Example: # Trigger only on creation @register_state_transition('task', 'task_created', triggers_on_create=True) class TaskCreatedTransition(ModelChangeTransition): pass # Trigger when specific fields change @register_state_transition('project', 'project_published', triggers_on=['is_published']) class ProjectPublishedTransition(ModelChangeTransition): pass # Trigger when any of several fields change @register_state_transition('project', 'settings_changed', triggers_on=['maximum_annotations', 'overlap_cohort_percentage']) class ProjectSettingsChangedTransition(ModelChangeTransition): pass c|_|_xsg|_|_|_t j ||Sr )_triggers_on_create_triggers_on_update_trigger_fields_transition_name_force_state_recordtransition_registryr)rarrlr`rkrirjs rr1z,register_state_transition..decoratorWsN/A,/A,+6+<"(,;)/A,$$[/CSTr)rar rr r*)rr`rirjrkrlr1s`````` rregister_state_transitionru/sP    r)FTNF)!r(loggingtypingrrrdjango.db.modelsrr TYPE_CHECKINGfsm.state_modelsrfsm.transitionsr getLoggerr%rAr r,r)r.r3r5rQrRrTrVrZr\rtrMr rur*rrr}s5 ''/ *.   8 $++^./ ;3 ;,D#D#P*+c2BCBkB 7 7+)> 7(u(+1F( <"<"@)* %#$ 3333 3  3  3r