From cc0625ffcffea20ccd032ca0aefa2ba42b63aa18 Mon Sep 17 00:00:00 2001
From: appleb_m <martin.appleby@psi.ch>
Date: Fri, 27 Mar 2026 16:59:44 +0100
Subject: [PATCH] Models: added additional Session States for Pending. Also
 Added from_dict function for SampleShortInfo Model

---
 src/aare/common/models.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/src/aare/common/models.py b/src/aare/common/models.py
index 1db6f879..3129e686 100644
--- a/src/aare/common/models.py
+++ b/src/aare/common/models.py
@@ -403,6 +403,9 @@ class SampleShortInfo(BaseModel):
         else:
             return f"{self.location.segment}{self.location.pos}-{self.pin:02d}"
 
+    @classmethod
+    def from_dict(cls, data: dict):
+        return cls(**data)
 
 class SampleShortInfoList(BaseModel):
     s: List[SampleShortInfo]
@@ -549,9 +552,11 @@ class BeamlineStateEnum(Enum):
 
 
 class SessionsStateEnum(Enum):
+    Vacant = 0
     OwnedByYou = 1
     OwnedByElse = 2
-    Vacant = 3
+    PendingYouToElse = 3
+    PendingElseToYou = 4
 
 class SampleCameraSettings(BaseModel):
     gain: float