]j4^dZddlZddlmZddlmZmZmZddlmZddl m Z ddl m Z m Z ddlmZdd lmZmZmZdd lmZmZmZej0eZGd d e j6Zed GddeZedGddeZedGddeZy)a FSM State Models for Label Studio. This module contains the state model definitions (BaseState and concrete state models). These are separated from models.py to avoid registration issues in LSE where extended state models need to be registered instead of the base OSS models. When importing FsmHistoryStateModel, these state models won't be automatically imported and registered, allowing LSE to register its own extended versions. N)datetime)AnyDictOptional)settings)models)QuerySet UUIDField)register_state_model)AnnotationStateChoicesProjectStateChoicesTaskStateChoices) UUID7Fieldgenerate_uuid7timestamp_from_uuid7ceZdZdZededdZejddddZ ejddd Z ejdddd Z ejd ddd Z ejej ej"ddZej&edZej,ddZej0dddZGddZdZedZedefdZde fdZ!e"de#dfdZ$e"de#e fdZ%e"de&dfd Z'e"d!ed"ede&dfd#Z(e"d$efd%Z)e"de*e e+ffd&Z,e"dejZfd'Z.e"de fd(Z/y))* BaseStatea Abstract base class for all state models using UUID7 for optimal time-series performance. This is the core of the FSM system, providing: - UUID7 primary key with natural time ordering - Standard state transition metadata - Audit trail information - Context data storage - Performance-optimized helper methods Benefits of this architecture: - INSERT-only operations for maximum concurrency - Natural time ordering eliminates need for created_at indexes - Global uniqueness enables distributed system support - Time-based partitioning for large amounts of state records with consistent performance - Complete audit trail by design TFz:UUID7 provides natural time ordering and global uniqueness) primary_keydefaulteditable help_textzKOrganization ID that owns this state record (for multi-tenant applications))nullblankdb_indexr2zCurrent state of the entity) max_lengthrrz%Previous state before this transition)rrrrdz>Name of the transition method that triggered this state changez(User who triggered this state transition) on_deleterrzTAdditional context data for this transition (e.g., validation results, external IDs))rrz/Human-readable reason for this state transition)rrzLHuman-readable timestamp for debugging (UUID7 id contains precise timestamp)) auto_now_addrrceZdZdZdgZdZy)BaseState.MetaT-ididN)__name__ __module__ __qualname__abstractordering get_latest_byG/root/env/lib/python3.12/site-packages/label_studio/fsm/state_models.pyMetar!cs7 r+r-ct||jdd}|jjd|d|jd|jS)N_idunknown z: u → )getattr_get_entity_nametitleprevious_statestate)self entity_ids r,__str__zBaseState.__str__isaDT%:%:%<$=S"A9M '')//12!I;bATAT@UUZ[_[e[eZfggr+c:|j}t||S)zGet the related entity object)r3r2)r7 entity_names r,entityzBaseState.entityms++- t[))r+returnc,t|jS)zExtract timestamp from UUID7 ID)rr#)r7s r,timestamp_from_uuidzBaseState.timestamp_from_uuidss$DGG,,r+cx|jj}|jdr|ddjSy)u>Extract entity name from model name (e.g., TaskState → task)StateNr<) __class__r$endswithlower)r7 model_names r,r3zBaseState._get_entity_namexs7^^,,   w 'cr?((* *r+c|j}|jjdi||ijdj S)z Get current state using UUID7 natural ordering. Uses UUID7's natural time ordering to efficiently find the latest state without requiring created_at indexes or complex queries. r"r*)_get_entity_field_nameobjectsfilterorder_byfirstclsr< entity_fields r,get_current_statezBaseState.get_current_statesH4467 !s{{!!;\6$:;DDUKQQSSr+c|j}|jjdi||ijdj }|r |j SdS)z Get current state value as string using UUID7 natural ordering. Uses UUID7's natural time ordering to efficiently find the latest state without requiring created_at indexes or complex queries. r"Nr*)rHrIrJrKrLr6)rNr<rO current_states r,get_current_state_valuez!BaseState.get_current_state_valuesZ4467 * **DlF-CDMMeTZZ\ &3}""==r+c~|j}|jjdi||ijdS)z(Get complete state history for an entityr"r*)rHrIrJrKrMs r,get_state_historyzBaseState.get_state_historys?4467 !s{{!!;\6$:;DDUKKr+ start_timeend_timec|j}|jjdi||i}tj|||j dS)z Efficient time-range queries using UUID7. Uses UUID7's embedded timestamp for direct time-based filtering without requiring timestamp indexes. r#r*)rHrIrJrfilter_by_time_rangerK)rNr<rVrWrOquerysets r,get_states_in_rangezBaseState.get_states_in_rangesT4467 %3;;%%?v(>?..xXNWWX\]]r+sincec|j}|jjdi||i}tj||j dS)z)Get all states since a specific timestampr#r*)rHrIrJrfilter_since_timerK)rNr<r\rOrZs r,get_states_sincezBaseState.get_states_sincesQ4467 %3;;%%?v(>?++He<EEdKKr+ciS)a4 Get denormalized fields to include in the state record. Override this method in subclasses to provide denormalized data that should be stored with each state transition for performance optimization and auditing purposes. Args: entity: The entity instance being transitioned Returns: Dictionary of field names to values that should be stored in the state record Example: @classmethod def get_denormalized_fields(cls, entity): return { 'project_id': entity.project_id, 'organization_id': entity.project.organization_id, 'task_type': entity.task_type, 'priority': entity.priority } r*rNr<s r,get_denormalized_fieldsz!BaseState.get_denormalized_fieldss 4 r+cl|j}|jj|jS)z(Get the entity model for the state model)rH_meta get_field related_model)rN field_names r,get_entity_modelzBaseState.get_entity_models-//1 yy"":.<< > >L(;*?LL  ^X ^ ^V^_jVk ^ ^LXLL S#X6=== sr+rtaskceZdZdZej ddej Zejde jdZ ejdd Z Gd d Zed Zy ) TaskStateu Core task state tracking for Label Studio. Provides basic task state management with: - Simple 3-state workflow (CREATED → IN_PROGRESS → COMPLETED) - High-performance queries with UUID7 ordering z tasks.Task fsm_states) related_namerrTrchoicesrz3From task.project_id - denormalized for performancerrceZdZdZej ddgdej gddej gdd ej dd gd gZdgZy ) TaskState.Metafsmtask_idr"task_current_state_idxfieldsname project_idr6r"task_project_state_idxrlr6r"task_org_reporting_idxr#task_history_idxNr$r%r& app_labelrIndexindexesr(r*r+r,r-rsi  FLLE 29Q R FLL >E] ^ FLL CJb c FLLD 18J K  7r+r-cd|jiS)z.Get denormalized fields for TaskState creationr)rras r,rbz!TaskState.get_denormalized_fieldss &++  r+N)r$r%r&rjrroCASCADEr~rmrrr6rkrr-r|rbr*r+r,rrs| 6  \ PVP^P^ _D F  4D4L4LW[ \E,,,!VJ    r+r annotationc*eZdZdZej dej dZejde jdZ ejdd Z ejdd Zejddd ZGd dZedZy)AnnotationStateu Core annotation state tracking for Label Studio. Provides basic annotation state management with: - Simple 2-state workflow (CREATED → COMPLETED) ztasks.AnnotationrrrrTrz6From annotation.task_id - denormalized for performancerz>From annotation.task.project_id - denormalized for performancez>From annotation.completed_by_id - denormalized for performancerrrceZdZdZej ddgdej gddej gdd ej gd d gZdgZy ) AnnotationState.Metar annotation_idr"anno_current_state_idxr)rr6r"anno_task_state_idx)completed_by_idr6r"anno_user_report_idxranno_project_report_idxNrr*r+r,r-r$se  FLL% 8?W X FLL ;BW X FLL CJ` a FLL >E^ _  7r+r-c|jj|jj|jr|jdSddS)z4Get denormalized fields for AnnotationState creationN)rrr)r~r#rrras r,rbz'AnnotationState.get_denormalized_fields0sH{{~~ ++009?9O9Ov55  VZ  r+N)r$r%r&rjrrorrrmr rr6rkrrrr-r|rbr*r+r,rr s#""#5^jkJ F  4J4R4R]a bE*f))!YG-,,!aJ2f11 D,lO    r+rprojectceZdZdZej dej dZejde jdZ ejddd Z Gd d Zed Zy ) ProjectStateu Core project state tracking for Label Studio. Provides basic project state management with: - Simple 3-state workflow (CREATED → IN_PROGRESS → COMPLETED) - Project lifecycle tracking zprojects.ProjectrrrTrz9From project.created_by_id - denormalized for performancerceZdZdZej ddgdej gddej ddgd gZdgZy ) ProjectState.Metarrr"project_current_state_idxrrproject_org_state_idxrlproject_org_reporting_idxNrr*r+r,r-rMsX  FLLu 5rs && 0- GF   8 $y yFf& & & Rl#+ i+ $+ \i " 9" !" r+