Add type annotation

ehdgua01 · Apr 22, 2021 · 1f72d82 · 1f72d82
1 parent 5556149
commit 1f72d82
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 8 deletions.
diff --git a/blocksync/hooks.py b/blocksync/hooks.py
@@ -1,4 +1,4 @@
-from typing import Callable, Optional
+from typing import Any, Callable, Optional
 
 from blocksync.status import Status
 
@@ -7,12 +7,12 @@
 
 class Hooks:
     def __init__(self):
-        self.root_before: Optional[Callable] = None
-        self.before: Optional[Callable] = None
-        self.root_after: Optional[Callable] = None
-        self.after: Optional[Callable] = None
-        self.monitor: Optional[Callable] = None
-        self.on_error: Optional[Callable] = None
+        self.root_before: Optional[Callable[[None], Any]] = None
+        self.before: Optional[Callable[[None], Any]] = None
+        self.root_after: Optional[Callable[[Status], Any]] = None
+        self.after: Optional[Callable[[Status], Any]] = None
+        self.monitor: Optional[Callable[[Status], Any]] = None
+        self.on_error: Optional[Callable[[Status], Any]] = None
 
     def _run(self, hook: Optional[Callable], *args, **kwargs):
         if hook:

diff --git a/blocksync/worker.py b/blocksync/worker.py
@@ -87,7 +87,6 @@ def _sync(self):
         except Exception as e:
             self._log(str(e), level=logging.ERROR, exc_info=True)
             self.syncer.hooks.run_on_error(e, self.syncer.status)
-        return
 
     def _log(self, msg: str, level: int = logging.INFO, *args, **kwargs):
         self.logger.log(level, f"[Worker {self.worker_id}]: {msg}", *args, **kwargs)