Implement create funding mutation

thoth-pub · Feb 1, 2022 · 8735bd6 · 8735bd6
1 parent 44e41b2
commit 8735bd6
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/thothlibrary/client.py b/thothlibrary/client.py
@@ -115,6 +115,10 @@ def create_location(self, location):
         """Construct and trigger a mutation to add a new location object"""
         return self.mutation("createLocation", location)
 
+    def create_funding(self, funding):
+        """Construct and trigger a mutation to add a new funding object"""
+        return self.mutation("createFunding", funding)
+
     @staticmethod
     def supported_versions():
         """

diff --git a/thothlibrary/mutation.py b/thothlibrary/mutation.py
@@ -185,6 +185,18 @@ class ThothMutation():
             ],
             "return_value": "locationId"
         },
+        "createFunding": {
+            "fields": [
+                ("workId", True),
+                ("institutionId", True),
+                ("program", True),
+                ("projectName", False),
+                ("projectShortname", False),
+                ("grantNumber", False),
+                ("jurisdiction", False)
+            ],
+            "return_value": "fundingId"
+        },
         "updateWork": {
             "fields": [
                 ("workId", True),