功能：新增托盘状态机

2026-07-24 19:37:46 +08:00
parent 5209911dd8
commit bab09f4eb3
3 changed files with 144 additions and 0 deletions
@@ -0,0 +1,30 @@
+from enum import Enum
+
+
+class PalletFlowState(str, Enum):
+    UNSTOCKED = 'UNSTOCKED'
+    EMPTY_IN_WAREHOUSE = 'EMPTY_IN_WAREHOUSE'
+    FILLED_IN_WAREHOUSE = 'FILLED_IN_WAREHOUSE'
+    AT_SITE_UNUSED = 'AT_SITE_UNUSED'
+    AT_SITE_PARTIAL = 'AT_SITE_PARTIAL'
+    AT_SITE_EMPTY = 'AT_SITE_EMPTY'
+    RETURNING = 'RETURNING'
+
+
+class DocumentStatus(str, Enum):
+    DRAFT = 'DRAFT'
+    CONFIRMED = 'CONFIRMED'
+
+
+class ReturnStatus(str, Enum):
+    REQUESTED = 'REQUESTED'
+    CONFIRMED = 'CONFIRMED'
+
+
+class PalletEventType(str, Enum):
+    INBOUND_CONFIRMED = 'INBOUND_CONFIRMED'
+    FILLING_CONFIRMED = 'FILLING_CONFIRMED'
+    SHIPMENT_CONFIRMED = 'SHIPMENT_CONFIRMED'
+    USAGE_RECORDED = 'USAGE_RECORDED'
+    RETURN_REQUESTED = 'RETURN_REQUESTED'
+    RETURN_CONFIRMED = 'RETURN_CONFIRMED'
@@ -0,0 +1,38 @@
+from module_business.domain.enums import PalletFlowState
+
+
+class InvalidPalletTransition(ValueError):
+    """托盘状态迁移不符合业务规则。"""
+
+
+class PalletStateMachine:
+    _ALLOWED_TRANSITIONS = frozenset(
+        {
+            (PalletFlowState.UNSTOCKED, PalletFlowState.EMPTY_IN_WAREHOUSE),
+            (
+                PalletFlowState.EMPTY_IN_WAREHOUSE,
+                PalletFlowState.FILLED_IN_WAREHOUSE,
+            ),
+            (PalletFlowState.FILLED_IN_WAREHOUSE, PalletFlowState.AT_SITE_UNUSED),
+            (PalletFlowState.AT_SITE_UNUSED, PalletFlowState.AT_SITE_PARTIAL),
+            (PalletFlowState.AT_SITE_UNUSED, PalletFlowState.AT_SITE_EMPTY),
+            (PalletFlowState.AT_SITE_UNUSED, PalletFlowState.RETURNING),
+            (PalletFlowState.AT_SITE_PARTIAL, PalletFlowState.AT_SITE_PARTIAL),
+            (PalletFlowState.AT_SITE_PARTIAL, PalletFlowState.AT_SITE_EMPTY),
+            (PalletFlowState.AT_SITE_PARTIAL, PalletFlowState.RETURNING),
+            (PalletFlowState.AT_SITE_EMPTY, PalletFlowState.RETURNING),
+            (PalletFlowState.RETURNING, PalletFlowState.FILLED_IN_WAREHOUSE),
+            (PalletFlowState.RETURNING, PalletFlowState.EMPTY_IN_WAREHOUSE),
+        }
+    )
+
+    @classmethod
+    def ensure_transition(
+        cls,
+        current: PalletFlowState,
+        target: PalletFlowState,
+    ) -> None:
+        if (current, target) not in cls._ALLOWED_TRANSITIONS:
+            raise InvalidPalletTransition(
+                f'不允许的托盘状态迁移：{current.value} -> {target.value}'
+            )