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

…314
ral-facilities · Feb 10, 2022 · bf1c830 · bf1c830
1 parent 7640e8d
commit bf1c830
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
@@ -410,6 +410,10 @@ class Technique(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(Technique, cls).from_icat(icat_data, required_related_fields)