diff options
| -rw-r--r-- | fuzzycat/entities.py | 3 | ||||
| -rw-r--r-- | fuzzycat/verify.py | 7 |
2 files changed, 9 insertions, 1 deletions
diff --git a/fuzzycat/entities.py b/fuzzycat/entities.py index 0b9ebc3..4849d31 100644 --- a/fuzzycat/entities.py +++ b/fuzzycat/entities.py @@ -1,3 +1,6 @@ +""" +Taken from https://git.io/JLRbX, transforms/entities.py +""" import collections import json diff --git a/fuzzycat/verify.py b/fuzzycat/verify.py index c3c7179..f6c96f3 100644 --- a/fuzzycat/verify.py +++ b/fuzzycat/verify.py @@ -81,7 +81,7 @@ import json import operator import re import sys -from typing import Counter, Dict, Tuple +from typing import Counter, Dict, Tuple, Type from glom import PathAccessError, glom @@ -90,6 +90,8 @@ from fuzzycat.data import (CONTAINER_NAME_BLACKLIST, PUBLISHER_BLACKLIST, TITLE_ TITLE_FRAGMENT_BLACKLIST) from fuzzycat.utils import (author_similarity_score, contains_chemical_formula, dict_key_exists, has_doi_prefix, jaccard, num_project, parse_page_string, slugify_string) +from fuzzycat.entities import entity_to_dict +from fatcat_openapi_client import ReleaseEntity Verify = collections.namedtuple("Verify", "status reason") @@ -144,6 +146,9 @@ class GroupVerifier: self.counter["total"] = sum(v for _, v in self.counter.items()) +def verify_release_entities(a: ReleaseEntity, b: ReleaseEntity, min_title_length=5) -> Type[Verify]: + return verify(entity_to_dict(a), entity_to_dict(b), min_title_length=min_title_length) + def verify(a: Dict, b: Dict, min_title_length=5) -> Tuple[str, str]: """ Compare two entities (dicts), return tuple of match status and reason. |