fix: use alternative ICAT mapping for Instrument pid when pid is None #…

…314
ral-facilities · Feb 10, 2022 · ae7e57a · ae7e57a
1 parent 3385282
commit ae7e57a
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/datagateway_api/src/search_api/models.py b/datagateway_api/src/search_api/models.py
@@ -287,6 +287,10 @@ class Instrument(PaNOSCAttribute):
 
     datasets: Optional[List[Dataset]] = []
 
+    @validator("pid", pre=True, always=True)
+    def set_pid(cls, value):  # noqa: B902, N805
+        return f"pid:{value}" if isinstance(value, int) else value
+
     @classmethod
     def from_icat(cls, icat_data, required_related_fields):
         return super(Instrument, cls).from_icat(icat_data, required_related_fields)