1 files changed, 103 insertions, 0 deletions
diff --git a/fatcat/api_client.py b/fatcat/api_client.py
new file mode 100644
index 00000000..12c70407
--- /dev/null
+++ b/fatcat/api_client.py
@@ -0,0 +1,103 @@
+
+import json
+import requests
+
+
+class FatCatApiClient:
+
+    def __init__(self, host_url):
+        self.host_url = host_url
+        self.session = requests.Session()
+
+
+    def get(self, path):
+        return self.session.get(self.host_url + path)
+
+
+    def post(self, path, data=None, headers=None):
+        hdrs = {"content-type": "application/json"}
+        if headers:
+            hdrs.update(headers)
+        #if type(data) == dict:
+        #    data = json.dumps(data, indent=None).encode('utf-8')
+        return self.session.post(self.host_url + path, json=data, headers=hdrs)
+
+
+    def import_crossref_file(self, json_file):
+        with open(json_file, 'r') as file:
+            for line in file:
+                obj = json.loads(line)
+                self.import_crossref_dict(obj)
+
+
+    def new_edit_group(self):
+        rv = self.post('/v0/editgroup')
+        assert rv.status_code == 200
+        editgroup_id = rv.json()['id']
+        return editgroup_id
+
+
+    def import_crossref_dict(self, meta):
+
+        # creators
+        creators = []
+        for am in meta['author']:
+            c = dict(name="{} {}".format(am['given'], am['family']),
+                     sortname="{}, {}".format(am['family'], am['given']),
+                     orcid=None)
+            creators.append(c)
+
+        # container
+        container = dict(
+            issn=meta['ISSN'][0],
+            name=meta['container-title'][0],
+            #container_id=None,
+            #sortname=meta['short-container-title'][0])
+            publisher=meta['publisher'])
+        #rv = self.post('/v0/container', data=container)
+        #assert rv.status_code == 200
+        #container_id = rv.json()['id']
+
+        # references
+        refs = []
+        for i, rm in enumerate(meta.get('reference', [])):
+            ref = dict(
+                doi=rm.get("DOI", None),
+                index=i+1,
+                # TODO: how to generate a proper stub here from k/v metadata?
+                stub="| ".join(rm.values()))
+            refs.append(ref)
+
+        # work and release
+        title = meta['title'][0]
+        rv = self.post('/v0/work',
+            data=dict(title=title)) #work_type="book"
+        assert rv.status_code == 200
+        work_id = rv.json()['id']
+
+        rv = self.post('/v0/release', data=dict(
+            title=title,
+            work=work_id,
+            # XXX: creators=creators,
+            # XXX: refs=refs,
+            #container=container_id,
+            release_type=meta['type'],
+            doi=meta['DOI'],
+            date=meta['created']['date-time'],
+            license=meta.get('license', [dict(URL=None)])[0]['URL'] or None,
+            issue=meta.get('issue', None),
+            volume=meta.get('volume', None),
+            pages=meta.get('page', None),
+            extra=dict(crossref={
+                'links': meta.get('link', []),
+                'subject': meta['subject'],
+                'type': meta['type'],
+                'alternative-id': meta.get('alternative-id', [])})))
+        assert rv.status_code == 200
+        release_id = rv.json()['id']
+
+
+    def health(self):
+        rv = self.get("/health")
+        assert rv.status_code == 200
+        return rv.json()