diff options
author | Bryan Newbold <bnewbold@robocracy.org> | 2021-11-03 14:22:56 -0700 |
---|---|---|
committer | Bryan Newbold <bnewbold@robocracy.org> | 2021-11-03 16:46:07 -0700 |
commit | 7cd0d97849f37d366a33c98204321c471fb6011c (patch) | |
tree | 034bd0a1d86e4bf32c4e9caa3826f0b7ebb6a95f | |
parent | 45612465cc08114fa19588ea1b0aba0ace1e7b03 (diff) | |
download | fatcat-7cd0d97849f37d366a33c98204321c471fb6011c.tar.gz fatcat-7cd0d97849f37d366a33c98204321c471fb6011c.zip |
typing: add assertions to fatcat_tool code to make type assumptions explicit
-rw-r--r-- | python/fatcat_tools/cleanups/common.py | 1 | ||||
-rw-r--r-- | python/fatcat_tools/harvest/pubmed.py | 1 | ||||
-rw-r--r-- | python/fatcat_tools/workers/changelog.py | 1 |
3 files changed, 3 insertions, 0 deletions
diff --git a/python/fatcat_tools/cleanups/common.py b/python/fatcat_tools/cleanups/common.py index 7ebfc8a0..4e8e49fd 100644 --- a/python/fatcat_tools/cleanups/common.py +++ b/python/fatcat_tools/cleanups/common.py @@ -141,4 +141,5 @@ class EntityCleaner: ) self._editgroup_id = eg.editgroup_id + assert self._editgroup_id return self._editgroup_id diff --git a/python/fatcat_tools/harvest/pubmed.py b/python/fatcat_tools/harvest/pubmed.py index a1b4da0e..118580db 100644 --- a/python/fatcat_tools/harvest/pubmed.py +++ b/python/fatcat_tools/harvest/pubmed.py @@ -259,6 +259,7 @@ def generate_date_file_map(host: str = "ftp.ncbi.nlm.nih.gov") -> Dict[str, Any] filedate, ) = match.groups() # ('pubmed20n1017.xml', 'Tue Dec 17 15:23:32 EST 2019') date = dateparser.parse(filedate) + assert date is not None fullpath = "/pubmed/updatefiles/{}.gz".format(filename) date_str = date.strftime("%Y-%m-%d") mapping[date_str].add(fullpath) diff --git a/python/fatcat_tools/workers/changelog.py b/python/fatcat_tools/workers/changelog.py index 8f8efdda..ff358c66 100644 --- a/python/fatcat_tools/workers/changelog.py +++ b/python/fatcat_tools/workers/changelog.py @@ -35,6 +35,7 @@ class ChangelogWorker(FatcatWorker): # topic if self.offset is None: print("Checking for most recent changelog offset...") + assert self.produce_topic msg = most_recent_message(self.produce_topic, self.kafka_config) if msg: self.offset = json.loads(msg.decode("utf-8"))["index"] |