fix(cff): error handling on invalid yaml

sdsc-ordes · Feb 1, 2024 · ff6bd5d · ff6bd5d
1 parent db81df7
commit ff6bd5d
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/gimie/parsers/cff.py b/gimie/parsers/cff.py
@@ -21,6 +21,7 @@
 
 from rdflib.term import URIRef
 
+from gimie import logger
 from gimie.graph.namespaces import SDO
 from gimie.parsers.abstract import Parser, Property
 
@@ -100,11 +101,15 @@ def get_cff_doi(data: bytes) -> Optional[str]:
 
     """
 
-    cff = yaml.safe_load(data.decode())
+    try:
+        cff = yaml.safe_load(data.decode())
+    except yaml.scanner.ScannerError:
+        logger.warn("cannot read CITATION.cff, skipped.")
+        return None
 
     try:
         doi = cff["doi"]
-    except KeyError:
+    except (KeyError, TypeError):
         return None
 
     return doi_to_url(doi)