diff options
Diffstat (limited to 'fatcat_covid19')
| -rw-r--r-- | fatcat_covid19/derivatives.py | 2 | ||||
| -rw-r--r-- | fatcat_covid19/enrich.py | 3 | ||||
| -rw-r--r-- | fatcat_covid19/transform.py | 5 |
3 files changed, 6 insertions, 4 deletions
diff --git a/fatcat_covid19/derivatives.py b/fatcat_covid19/derivatives.py index 5ade0ef..c9339e8 100644 --- a/fatcat_covid19/derivatives.py +++ b/fatcat_covid19/derivatives.py @@ -126,7 +126,7 @@ def enrich_derivatives_file(json_input, json_output, base_dir): """ for l in json_input: l = json.loads(l) - result = do_line(l, base_dir) + result = enrich_derivatives_row(l, base_dir) if result: print(json.dumps(result, sort_keys=True), file=json_output) diff --git a/fatcat_covid19/enrich.py b/fatcat_covid19/enrich.py index 458c83d..e7d6da2 100644 --- a/fatcat_covid19/enrich.py +++ b/fatcat_covid19/enrich.py @@ -51,7 +51,7 @@ def enrich_fatcat_row(row, api_session): if fatcat_release: row['fatcat_release'] = fatcat_release row['release_id'] = fatcat_release['ident'] - print(json.dumps(row, sort_keys=True)) + return row def enrich_fatcat_file(json_input, json_output): @@ -65,3 +65,4 @@ def enrich_fatcat_file(json_input, json_output): result = enrich_fatcat_row(l, api_session) if result: print(json.dumps(result, sort_keys=True), file=json_output) + diff --git a/fatcat_covid19/transform.py b/fatcat_covid19/transform.py index c31c9f4..16774ab 100644 --- a/fatcat_covid19/transform.py +++ b/fatcat_covid19/transform.py @@ -68,7 +68,8 @@ def fulltext_to_elasticsearch(row, force_bool=True): if release.get('abstracts'): for a in release['abstracts']: abstracts.append(a['content']) - abstract_langs.append(a['lang']) + if a.get('lang'): + abstract_langs.append(a['lang']) contrib_names = [] contrib_affiliations = [] @@ -199,6 +200,6 @@ def transform_es_file(json_input, json_output): """ for l in json_input: l = json.loads(l) - result = fulltext_to_elasticsearch(l, args) + result = fulltext_to_elasticsearch(l) if result: print(json.dumps(result, sort_keys=True), file=json_output) |