refactor DOI domain lookup into python code; expand table

1 files changed, 14 insertions, 0 deletions

diff --git a/fatcat_scholar/schema.py b/fatcat_scholar/schema.py
index d3a91a7..480b8fa 100644
--- a/fatcat_scholar/schema.py
+++ b/fatcat_scholar/schema.py
@@ -19,6 +19,7 @@ from pydantic import BaseModel
 
 from fatcat_openapi_client import ReleaseEntity, ReleaseContrib
 from fatcat_scholar.api_entities import entity_to_dict
+from fatcat_scholar.biblio_hacks import doi_link_domain
 
 
 class DocType(str, Enum):
@@ -104,6 +105,19 @@ class ScholarBiblio(BaseModel):
     contrib_names: List[str]
     affiliations: List[str]
 
+    def doi_link_domain(self, default: str = "doi.org") -> str:
+        if not self.doi_prefix:
+            return default
+        domain = doi_link_domain(
+            self.doi_prefix,
+            container_name=self.container_name,
+            publisher=self.publisher,
+        )
+        if domain:
+            return domain
+        else:
+            return default
+
     def citation_str(self, style: str) -> Optional[str]:
         """
         Tries to format this biblio metadata as a citation string. If it fails,