From 955d4942d5111cbbd8f4b99d4e3c6d79262dab6e Mon Sep 17 00:00:00 2001
From: Bryan Newbold <bnewbold@robocracy.org>
Date: Mon, 13 May 2019 14:23:02 -0700
Subject: partial python impl of ext_id and release_stage refactors

---
 python/fatcat_tools/importers/cdl_dash_dat.py    |  8 +++++---
 python/fatcat_tools/importers/crossref.py        | 26 +++++++++++++-----------
 python/fatcat_tools/importers/grobid_metadata.py |  1 +
 3 files changed, 20 insertions(+), 15 deletions(-)

(limited to 'python/fatcat_tools/importers')

diff --git a/python/fatcat_tools/importers/cdl_dash_dat.py b/python/fatcat_tools/importers/cdl_dash_dat.py
index 1a21194d..e1252b6d 100755
--- a/python/fatcat_tools/importers/cdl_dash_dat.py
+++ b/python/fatcat_tools/importers/cdl_dash_dat.py
@@ -96,7 +96,9 @@ def cdl_dash_release(meta, extra=None):
         ))
 
     r = ReleaseEntity(
-        doi=doi,
+        ext_ids=ReleaseEntityExtIds(
+            doi=doi,
+        ),
         title=clean(meta['title'], force_xml=True),
         publisher=clean(meta['publisher']),
         release_year=int(meta['publicationYear']),
@@ -170,9 +172,9 @@ def auto_cdl_dash_dat(api, dat_path, release_id=None, editgroup_id=None):
                 agent="fatcat_tools.auto_cdl_dash_dat")))
         editgroup_id = eg.editgroup_id
 
-    if not release_id and release.doi:
+    if not release_id and release.ext_ids.doi:
         try:
-            r = api.lookup_release(doi=release.doi)
+            r = api.lookup_release(doi=release.ext_ids.doi)
             release_id = r.ident
         except fatcat_client.rest.ApiException:
             pass
diff --git a/python/fatcat_tools/importers/crossref.py b/python/fatcat_tools/importers/crossref.py
index 722f93aa..999ce13f 100644
--- a/python/fatcat_tools/importers/crossref.py
+++ b/python/fatcat_tools/importers/crossref.py
@@ -355,10 +355,10 @@ class CrossrefImporter(EntityImporter):
         # release status
         if obj['type'] in ('journal-article', 'conference-proceeding', 'book',
                 'dissertation', 'book-chapter'):
-            release_status = "published"
+            release_stage = "published"
         else:
             # unknown
-            release_status = None
+            release_stage = None
 
         # external identifiers
         extids = self.lookup_ext_ids(doi=obj['DOI'].lower())
@@ -408,18 +408,20 @@ class CrossrefImporter(EntityImporter):
             title=title,
             original_title=original_title,
             release_type=release_type,
-            release_status=release_status,
+            release_stage=release_stage,
             release_date=release_date,
             release_year=release_year,
             publisher=publisher,
-            doi=obj['DOI'].lower(),
-            pmid=extids['pmid'],
-            pmcid=extids['pmcid'],
-            wikidata_qid=extids['wikidata_qid'],
-            isbn13=isbn13,
-            core_id=extids['core_id'],
-            arxiv_id=extids['arxiv_id'],
-            jstor_id=extids['jstor_id'],
+            ext_ids=fatcat_client.ReleaseEntityExtIds(
+                doi=obj['DOI'].lower(),
+                pmid=extids['pmid'],
+                pmcid=extids['pmcid'],
+                wikidata_qid=extids['wikidata_qid'],
+                isbn13=isbn13,
+                core=extids['core_id'],
+                arxiv=extids['arxiv_id'],
+                jstor=extids['jstor_id'],
+            ),
             volume=clean(obj.get('volume')),
             issue=clean(obj.get('issue')),
             pages=clean(obj.get('page')),
@@ -437,7 +439,7 @@ class CrossrefImporter(EntityImporter):
         # lookup existing DOI (don't need to try other ext idents for crossref)
         existing = None
         try:
-            existing = self.api.lookup_release(doi=re.doi)
+            existing = self.api.lookup_release(doi=re.ext_ids.doi)
         except fatcat_client.rest.ApiException as err:
             if err.status != 404:
                 raise err
diff --git a/python/fatcat_tools/importers/grobid_metadata.py b/python/fatcat_tools/importers/grobid_metadata.py
index 7281a7a1..ba91d183 100644
--- a/python/fatcat_tools/importers/grobid_metadata.py
+++ b/python/fatcat_tools/importers/grobid_metadata.py
@@ -156,6 +156,7 @@ class GrobidMetadataImporter(EntityImporter):
             volume=clean(obj['journal'].get('volume')),
             issue=clean(obj['journal'].get('issue')),
             abstracts=abstracts,
+            ext_ids=fatcat_client.ReleaseEntityExtIds(),
             extra=extra)
         return re
 
-- 
cgit v1.2.3