fix: use alternative ICAT mapping for Document pid when doi is None #314

ral-facilities · Feb 10, 2022 · 736c6bd · 736c6bd
1 parent 341ee51
commit 736c6bd
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
@@ -236,6 +236,10 @@ class Document(PaNOSCAttribute):
     members: Optional[List["Member"]] = []
     parameters: Optional[List["Parameter"]] = []
 
+    @validator("pid", pre=True, always=True)
+    def set_pid(cls, value):  # noqa: B902, N805
+        return f"pid:{value}" if isinstance(value, int) else value
+
     @validator("is_public", pre=True, always=True)
     def set_is_public(cls, value):  # noqa: B902, N805
         if not value: