]j@ dZddlZddlZddlmZmZddlmZmZddlZddl m Z ddl m Z ddl mZejeZdej$fd Zd ej$defd Zd d ed eedeej$ej$ffdZdedej$fdZdej$defdZGddZGddZd!dZd defdZd deefdZ d"dedeefdZdededeefdZ dededeefdZ!y)#aP FSM utility functions. This module provides: 1. UUID7 utilities for time-series optimization (uses uuid-utils library) 2. FSM-specific helper functions for organization resolution and state management UUID7 provides natural time ordering and global uniqueness, making it ideal for INSERT-only architectures with millions of records. N)datetimetimezone)OptionalTuple)CurrentContext) load_func)settingsreturncftj}tjt |S)ai Generate a UUID7 with embedded timestamp for natural time ordering. UUID7 embeds the timestamp in the first 48 bits, providing: - Natural chronological ordering without additional indexes - Global uniqueness across distributed systems - Time-based partitioning capabilities Returns: UUID7 instance with embedded timestamp ) uuid_utilsuuid7uuidUUIDstr) uuid7_objs @/root/env/lib/python3.12/site-packages/label_studio/fsm/utils.pygenerate_uuid7rs%  "I 99S^ $$uuid7_idcv|jdz dz}tj|dz tjS)aI Extract timestamp from UUID7 ID. Args: uuid7_id: UUID7 instance to extract timestamp from Returns: datetime: Timestamp embedded in the UUID7 Example: uuid7_id = generate_uuid7() timestamp = timestamp_from_uuid7(uuid7_id) # timestamp is when the UUID7 was generated Pl)tz)intr fromtimestamprutc)r timestamp_mss rtimestamp_from_uuid7r0s3 LLB&=9L  ! !,"5(,, GGr start_timeend_timecR|#tjtj}t |j dzdz }t |j dzdz}t j|dzdzdz}t j|dzdzdzdz}||fS)a Generate UUID7 range for time-based queries. Creates UUID7 boundaries for efficient time-range filtering without requiring timestamp indexes. Args: start_time: Start of time range end_time: End of time range (defaults to now) Returns: Tuple of (start_uuid, end_uuid) for range queries Example: start_uuid, end_uuid = uuid7_time_range( datetime(2024, 1, 1), datetime(2024, 1, 2) ) # Query: WHERE id >= start_uuid AND id <= end_uuid rrrl)rnowrrr timestamprr)rr start_timestamp_msend_timestamp_ms start_uuidend_uuids ruuid7_time_ranger,Es*<< -Z113d:;a?8--/$67!; 2b 8YG:VWJyy."4CzRVcdeH x rr'cxt|jdz}tj|dzdzdzS)a Generate UUID7 from specific timestamp for range queries. Args: timestamp: Timestamp to embed in UUID7 Returns: UUID7 with embedded timestamp Example: # Get all states from the last hour start_time = timezone.now() - timedelta(hours=1) start_uuid = uuid7_from_timestamp(start_time) states = StateModel.objects.filter(id__gte=start_uuid) rrr#r$r%)rr'rr)r'rs ruuid7_from_timestampr.is<"y**,t34L 99,",;zJ KKr uuid_valuec |jdk(S)z Validate that a UUID is a valid UUID7. Args: uuid_value: UUID to validate Returns: True if valid UUID7, False otherwise )version)r/s rvalidate_uuid7r3s    ""rcXeZdZdZedZed dedeefdZedefdZ y) UUID7Fieldz Custom field utilities for UUID7 handling in Django models. Provides helper methods for UUID7-specific operations that can be used by models inheriting from BaseState. c@|jdjS)z.Get latest record using UUID7 natural orderingz-id)order_byfirst)querysets rget_latest_by_uuid7zUUID7Field.get_latest_by_uuid7s  '--//rNrr cFt||\}}|j||S)z=Filter queryset by time range using UUID7 embedded timestamps)id__gteid__lte)r,filter)r9rr r*r+s rfilter_by_time_rangezUUID7Field.filter_by_time_ranges( 0 HE Hz8DDrsincec<t|}|j|S)z1Filter queryset for records since a specific time)r<)r.r>)r9r@r*s rfilter_since_timezUUID7Field.filter_since_times*%0 z22rN) __name__ __module__ __qualname____doc__ staticmethodr:rrr?rBrrr5r5sb00E8ExPXGYEE 3833rr5cJeZdZdZddeefdZd dedejfdZ y) UUID7Generatorz UUID7 generator with optional custom timestamp. Useful for testing or when you need to generate UUIDs with specific timestamps. Nbase_timestampcj|xs#tjtj|_d|_y)z Initialize generator with optional base timestamp. Args: base_timestamp: Base timestamp to use (defaults to current time) rN)rr&rrrL_counter)selfrLs r__init__zUUID7Generator.__init__s%-J X\\0J r offset_msr ct|jjdz|z}|xjdz c_|dzdz|jdzdzzdz}t j |S) z Generate UUID7 with timestamp offset. Args: offset_ms: Millisecond offset from base timestamp Returns: UUID7 with adjusted timestamp rr"rr#i@r$r%)rrLr'rNrr)rOrQruuid_ints rgeneratezUUID7Generator.generateso4..88:TABYN   !B&95$--%:OTV9VW[efyyX&&rrC)r) rDrErFrGrrrPrrrrUrIrrrKrKs/ x'9'#'dii'rrKctj}|r|S|yt|dd}|st|dr$|jrt|jdd}net|drY|j rMt|j dr7|j jr!t|j jdd}|s0|r.t|dr"|j r|j j}|tj||S)a Resolve organization_id using consistent logic without additional queries. This provides organization_id resolution for logging and state tracking without duplicating database queries. Args: entity: The entity to resolve organization_id for user: Optional user for fallback organization resolution Returns: organization_id or None Norganization_idprojecttaskactive_organization) rget_organization_idgetattrhasattrrXrYrZidset_organization_id)entityuserrWs rresolve_organization_idrbs%88:O~f&7>O  69 %&..%fnn6GNO VV $i9X]c]h]h]p]p%fkk&9&9;LdSO t6K(LQUQiQi2255"**?; rc*tjS)a Check if FSM is enabled via feature flags and thread-local override. PERFORMANCE: This function now checks the cached FSM state that was set when the user was first initialized in CurrentContext. This avoids repeated feature flag lookups throughout the request. The check order is: 1. Check thread-local override (for test cleanup, bulk operations) 2. Check cached feature flag state (set once per request) 3. Fallback to direct feature flag check (for edge cases without context) Args: user: User for feature flag evaluation (optional, used as fallback only) Returns: True if FSM should be active )ris_fsm_enabled)ras rrdrds*  ( ( **rc t|sy ddlm}|}|j|S#t$r}t j d|jjd|jdt|d|jj|jt||t|d Yd}~yd}~wwxYw) z Safely get current state with error handling. Args: entity: The entity to get state for user: The user making the request (for feature flag checking) Returns: Current state string or None if failed Nrget_state_managerz Failed to get current state for  : zfsm.get_state_error)event entity_type entity_idrWerrorextra) rdfsm.state_managerrgget_current_state_value Exceptionloggerwarning_meta label_lowerpkrrb)r`rarg StateManageres rget_current_state_saferzs $ 7(* 33F;;  .v||/G/G.H&))TVWZ[\W]V^ _.%||77#YY#:64#HQ    s, B=A>B88B=inferred_statec t|sytjr|S ddlm}|}|j |}| ||k(s|s|S|_t jd|jjd|jdd|jj|jd y|jjj} t| |} | rSt jd | d|jd | |j|| d  |j|| |||xsi |St jd| d|d| |j|d y#t$r~} t j!d|jjd|jdt#| d|jj|jt#| ddYd} ~ yd} ~ wwxYw)a^ Get current state, or initialize it if it doesn't exist. This function handles "cold start" scenarios where pre-existing entities don't have FSM state records. It will: 1. If the state already exists, use that 2. If the state doesn't exist, infer the state from the entity and initialize it with an appropriate transition 2. Return the state value (never returns None if initialization succeeds) Args: entity: The entity to get or initialize state for user: User for FSM context inferred_state: Pre-computed inferred state reason: Custom reason for the state initialization (optional, overrides default reason) context_data: Additional context data to store with state record (optional) overwrite_state: Overwrite the state if it already exists (optional) Returns: Current or newly initialized state value, or None if FSM disabled or failed Examples: >>> task = Task.objects.get(id=123) # Pre-existing task without state >>> from fsm.state_inference import get_or_infer_state >>> inferred_state = get_or_infer_state(task) >>> state = get_or_initialize_state(task, user=request.user, inferred_state=inferred_state) >>> # state is now 'COMPLETED' or 'CREATED' based on task.is_labeled >>> # and a state record has been created NrrfzCannot initialize state for rhz - inference failedzfsm.initialize_state_failed)rjrkrlrnz(Initializing FSM state for pre-existing zfsm.cold_start_initialization)rjrkrlr{transition_name)r`r}rareason context_dataz'No initialization transition found for z -> z fsm.no_initialization_transition)rjrkrlr{z&Failed to get or initialize state for rizfsm.get_or_initialize_error)rjrkrlrmT)roexc_info)rdris_fsm_disabledrprgrqrsrtru model_namerwlower"get_initialization_transition_nameinfoexecute_transitionrrrmr) r`rar{r~roverwrite_statergrx current_staterkr}rys rget_or_initialize_stater8s< $ %%'F7(* %<*IQ`  ! NN.v||/F/F.Gq Sfg:#)<<#:#:!'  ll--335 <[.Y  KK:;-q T<#.!'&4'6     + + /)/R , " ! NN9+d>JZ[?#.!'&4      4V\\5L5L4MQvyykY[\_`a\b[c d6%||66#YYQ     s,(E A E .BE 3,E G')A4G""G'rk target_statecddlm}m}m}|dk(r1||jk(ry||j k(ry||j k(ryy |dk(r1||jk(ry||j k(ry ||j k(ry y |d k(r||jk(ry y )a7 Get the appropriate transition name for initializing an entity to a target state. Args: entity_type: Type of entity ('task', 'project', 'annotation') target_state: The target state to initialize to Returns: Transition name, or None if no appropriate transition exists r)AnnotationStateChoicesProjectStateChoicesTaskStateChoicesrY task_createdtask_completedtask_in_progressrXproject_createdproject_in_progressproject_completed annotationannotation_createdN)fsm.state_choicesrrrCREATED COMPLETED IN_PROGRESS)rkrrrrs r#_get_initialization_transition_namers`_f +33 3! -77 7# -99 9%   ! .66 6$ 0<< <( 0:: :&   $ 199 9' rcBttj||SrC)rr "FSM_INITIALIZATION_TRANSITION_NAME)rkrs rrrs A9X@@ A+| \\rrC)NN)NNF)"rGloggingrrrtypingrrr core.current_requestrcore.utils.commonr django.confr getLoggerrDrsrrrr,r.boolr3r5rKrbrdrrzrrrrIrrrsk  '"/'   8 $% %$H499HH*! ! Xh5G! SXY]YbYbdhdmdmYmSn! HLHLL2 #tyy #T #334"'"'T+\++0#@X]k"%k c]k\SPXY\P]D]C]s]xX[}]r