add annotations to ScannerWorkflowEvent

diff --git a/scanner_workflow.py b/scanner_workflow.py
index 453f087afa9a44f9be2478b7f2eb45afbf1d1853..ab7b907c877007c63fde67af1e261708aef053c5 100755 (executable)
--- a/scanner_workflow.py
+++ b/scanner_workflow.py
@@ -1,5 +1,7 @@
 #!/usr/bin/env python3
 
+from __future__ import annotations
+
 import click
 from watchdog.observers import Observer
 from watchdog.events import FileSystemEventHandler, FileSystemEvent
@@ -15,13 +17,13 @@ from time import sleep
 class ScannerWorkflowEvent(FileSystemEventHandler):
     """Subclass of FileSystemEventHandler to handle OCRing PDFs"""
 
-    scanner_workflow = None
+    scanner_workflow: ScannerWorkflow
 
-    def __init__(self, scanner_workflow=None):
+    def __init__(self, scanner_workflow: ScannerWorkflow):
         super().__init__()
         self.scanner_workflow = scanner_workflow
         if not self.scanner_workflow:
-            raise Error("No scanner_workflow passed to ScannerWorkflowEvent")
+            raise Exception("No scanner_workflow passed to ScannerWorkflowEvent")
 
     def on_any_event(self, event: FileSystemEvent):
         if event.is_directory: