Add OBO Foundry check function (#1356)

src/bioregistry/__init__.py

@@ -78,6 +78,7 @@
     has_no_terms,
     is_deprecated,
     is_novel,
+    is_obo_foundry,
     is_proprietary,
     normalize_curie,
     normalize_parsed_curie,

src/bioregistry/resolve.py

@@ -40,6 +40,7 @@
     "get_obo_download",
     "get_json_download",
     "get_owl_download",
+    "is_obo_foundry",
     "get_rdf_download",
     "get_version",
     "get_banana",
@@ -775,6 +776,21 @@ def is_proprietary(prefix: str) -> Optional[bool]:
     return entry.proprietary
 
 
+def is_obo_foundry(prefix: str) -> Optional[bool]:
+    """Get if the prefix has an OBO Foundry link.
+
+    :param prefix: The prefix to look up
+    :returns: If the prefix corresponds to an OBO Foundry resource
+
+    >>> assert is_obo_foundry('chebi')
+    >>> assert not is_proprietary('pdb')
+    """
+    entry = get_resource(prefix)
+    if entry is None:
+        return None
+    return entry.get_obofoundry_prefix() is not None
+
+
 def parse_curie(
     curie: str,
     *,

tests/test_paper_ranking.py

@@ -1,7 +1,7 @@
 """Test for checking the paper ranking model."""
 
-import json
 import datetime
+import json
 import unittest
 from pathlib import Path
 from unittest.mock import patch