railway_manager: adapt models.py

Author: younesschrifi

railway_manager_interface/osrd_railway_manager_interface/models.py

@@ -80,15 +80,24 @@ class Margins(BaseModel):
     """
 
 
-class OperationalPointIdentifierId(BaseModel):
+class OperationalPointPartReference(BaseModel):
+    operational_point: (
+        OperationalPointReferenceId
+        | OperationalPointReferenceTrigram
+        | OperationalPointReferenceUic
+    )
+    track_reference: TrackReferenceId | TrackReferenceName | None = None
+
+
+class OperationalPointReferenceId(BaseModel):
     operational_point: str
     """
     The object id of an operational point
     """
     type: Literal["id"]
 
 
-class OperationalPointIdentifierTrigram(BaseModel):
+class OperationalPointReferenceTrigram(BaseModel):
     secondary_code: str | None = None
     """
     An optional secondary code to identify a more specific location
@@ -100,7 +109,7 @@ class OperationalPointIdentifierTrigram(BaseModel):
     type: Literal["trigram"]
 
 
-class OperationalPointIdentifierUic(BaseModel):
+class OperationalPointReferenceUic(BaseModel):
     secondary_code: str | None = None
     """
     An optional secondary code to identify a more specific location
@@ -112,15 +121,6 @@ class OperationalPointIdentifierUic(BaseModel):
     """
 
 
-class OperationalPointReference(BaseModel):
-    operational_point: (
-        OperationalPointIdentifierId
-        | OperationalPointIdentifierTrigram
-        | OperationalPointIdentifierUic
-    )
-    track_reference: TrackReferenceId | TrackReferenceName | None = None
-
-
 class OptionsChangeGroup(BaseModel):
     value: TrainScheduleOptions
 
@@ -183,7 +183,7 @@ class PathItem(BaseModel):
     The unique identifier of the path item.
     This is used to reference path items in the train schedule.
     """
-    location: TrackOffset | OperationalPointReference
+    location: TrackOffset | OperationalPointPartReference
     """
     The location of a path waypoint
     """