diff options
Diffstat (limited to 'python/fatcat_tools/normal.py')
| -rw-r--r-- | python/fatcat_tools/normal.py | 19 |
1 files changed, 10 insertions, 9 deletions
diff --git a/python/fatcat_tools/normal.py b/python/fatcat_tools/normal.py index e37cace8..e27754f5 100644 --- a/python/fatcat_tools/normal.py +++ b/python/fatcat_tools/normal.py @@ -15,7 +15,8 @@ import pycountry DOI_REGEX = re.compile(r"^10.\d{3,6}/\S+$") -def clean_doi(raw): + +def clean_doi(raw: str) -> Optional[str]: """ Removes any: - padding whitespace @@ -91,7 +92,7 @@ def test_clean_doi(): ARXIV_ID_REGEX = re.compile(r"^(\d{4}.\d{4,5}|[a-z\-]+(\.[A-Z]{2})?/\d{7})(v\d+)?$") -def clean_arxiv_id(raw): +def clean_arxiv_id(raw: str) -> Optional[str]: """ Removes any: - 'arxiv:' prefix @@ -162,7 +163,7 @@ def test_clean_wikidata_qid(): assert clean_wikidata_qid("qfba3") == None assert clean_wikidata_qid("") == None -def clean_pmid(raw): +def clean_pmid(raw: str) -> Optional[str]: if not raw: return None raw = raw.strip() @@ -179,7 +180,7 @@ def test_clean_pmid(): assert clean_pmid("qfba3") == None assert clean_pmid("") == None -def clean_pmcid(raw): +def clean_pmcid(raw: str) -> Optional[str]: if not raw: return None raw = raw.strip() @@ -189,7 +190,7 @@ def clean_pmcid(raw): return raw return None -def clean_sha1(raw): +def clean_sha1(raw: str) -> Optional[str]: if not raw: return None raw = raw.strip().lower() @@ -209,7 +210,7 @@ def test_clean_sha1(): assert clean_sha1("qfba3fba0e1937aa0297de3836b768b5dfb23d7b") == None assert clean_sha1("0fba3fb a0e1937aa0297de3836b768b5dfb23d7b") == None -def clean_sha256(raw): +def clean_sha256(raw: str) -> Optional[str]: raw = raw.strip().lower() if len(raw.split()) != 1: return None @@ -226,7 +227,7 @@ def test_clean_sha256(): ISSN_REGEX = re.compile(r"^\d{4}-\d{3}[0-9X]$") -def clean_issn(raw): +def clean_issn(raw: str) -> Optional[str]: if not raw: return None raw = raw.strip().upper() @@ -244,7 +245,7 @@ def test_clean_issn(): ISBN13_REGEX = re.compile(r"^97(?:8|9)-\d{1,5}-\d{1,7}-\d{1,6}-\d$") -def clean_isbn13(raw): +def clean_isbn13(raw: str) -> Optional[str]: if not raw: return None raw = raw.strip() @@ -260,7 +261,7 @@ def test_clean_isbn13(): ORCID_REGEX = re.compile(r"^\d{4}-\d{4}-\d{4}-\d{3}[\dX]$") -def clean_orcid(raw): +def clean_orcid(raw: str) -> Optional[str]: if not raw: return None raw = raw.strip() |