This checks the short-container-title filed before making the replace…

…ment
bibcure · Jul 16, 2020 · e8ab193 · e8ab193
1 parent 9d2ab39
commit e8ab193
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 5 deletions.
diff --git a/.vscode/settings.json b/.vscode/settings.json
@@ -1,5 +1,6 @@
 {
     "python.linting.pylintEnabled": false,
-    "python.linting.pep8Enabled": true,
-    "python.linting.enabled": true
+    "python.linting.pycodestyleEnabled": false,
+    "python.linting.enabled": true,
+    "python.linting.flake8Enabled": true
 }
diff --git a/doi2bib/crossref.py b/doi2bib/crossref.py
@@ -23,7 +23,7 @@ def get_bib(doi):
     bib = r.content
     bib = str(bib, "utf-8")
 
-    return found, bib 
+    return found, bib
 
 
 def get_json(doi):
@@ -65,8 +65,9 @@ def get_bib_from_doi(doi, abbrev_journal=True):
 
         found, item = get_json(doi)
         if found:
-            abbreviated_journal = item["message"]["short-container-title"][0].strip()
-            if abbreviated_journal:
+            abbreviated_journal = item["message"]["short-container-title"]
+            if len(abbreviated_journal) > 0:
+                abbreviated_journal = abbreviated_journal[0].strip()
                 bib = re.sub(
                     r"journal = \{[^>]*?\}",
                     "journal = {" + abbreviated_journal + "}",