fairnb/api/invenio.py

@@ -98,6 +98,14 @@ class InvenioRDMConnector:
 
         return response
 
+    def create_doi(self, url: str):
+        response = rq.post(
+            url,
+            headers=self.headers,
+        ).json()
+
+        return response
+
     def download_file(
         self, record_id: str, filename: str, out_path: pathlib.Path, expand: bool = True
     ):

fairnb/entity/invenio_entity.py

@@ -78,15 +78,18 @@ class InvenioRDMEntity(Entity):
             self.record_metadata
         ).json()
         record_id = create_response["id"]
-        self.invenio_manager.upload_record(dir_path,
-                                           regex,
-                                           only_filename=False if self.location.is_dir() else True)
+
+        create_doi_response = self.invenio_manager.connector.create_doi(create_response["links"]["reserve_doi"])
+        uri = create_doi_response["links"]["doi"]
+
+        self.invenio_manager.upload_record(
+            dir_path,
+            regex,
+            only_filename=False if self.location.is_dir() else True)
 
         if self.publish_record:
             publish_response = self.invenio_manager.publish_record()
             uri = publish_response["links"]["self_doi"]
-        else:
-            uri = create_response["links"]["record"]  # Invalid until published
 
         metadata = EntityProvenance.new(
             name=self.name,