summaryrefslogtreecommitdiffstats
path: root/python/fatcat_tools
diff options
context:
space:
mode:
Diffstat (limited to 'python/fatcat_tools')
-rw-r--r--python/fatcat_tools/importers/grobid_metadata.py24
1 files changed, 10 insertions, 14 deletions
diff --git a/python/fatcat_tools/importers/grobid_metadata.py b/python/fatcat_tools/importers/grobid_metadata.py
index 28358959..7bd7c00e 100644
--- a/python/fatcat_tools/importers/grobid_metadata.py
+++ b/python/fatcat_tools/importers/grobid_metadata.py
@@ -81,9 +81,8 @@ class GrobidMetadataImporter(EntityImporter):
abstract = obj.get('abstract')
if abstract and len(abstract) < MAX_ABSTRACT_BYTES and len(abstract) > 10:
- abobj = dict(
+ abobj = fatcat_client.ReleaseEntityAbstracts(
mimetype="text/plain",
- language=None,
content=clean(obj.get('abstract')))
abstracts = [abobj]
else:
@@ -97,31 +96,28 @@ class GrobidMetadataImporter(EntityImporter):
role="author",
extra=None))
- # XXX: why is this a dict()? not covered by tests?
refs = []
for raw in obj.get('citations', []):
cite_extra = dict()
- ref = dict()
- ref['key'] = clean(raw.get('id'))
- if raw.get('title'):
- ref['title'] = clean(raw['title'])
+ year = None
if raw.get('date'):
try:
year = int(raw['date'].strip()[:4])
- ref['year'] = year
except:
pass
for key in ('volume', 'url', 'issue', 'publisher'):
if raw.get(key):
cite_extra[key] = clean(raw[key])
if raw.get('authors'):
- cite_extra['authors'] = [clean(a['name']) for a in raw['authors']]
- if cite_extra:
- cite_extra = dict(grobid=cite_extra)
- else:
+ cite_extra['author'] = [clean(a['name']) for a in raw['authors']]
+
+ if not cite_extra:
cite_extra = None
- ref['extra'] = cite_extra
- refs.append(ref)
+ refs.append(fatcat_client.ReleaseRef(
+ key=clean(raw.get('id')),
+ year=year,
+ title=clean(raw['title']),
+ extra=cite_extra))
release_date = None
release_year = None