From: Don Armstrong <don@donarmstrong.com>
Date: Thu, 31 Aug 2023 00:52:34 +0000 (-0700)
Subject: add annotations to ScannerWorkflowEvent
X-Git-Url: https://git.donarmstrong.com/?a=commitdiff_plain;h=138c37bb1f4085f66b9d0b047e9a26e703b39780;p=scanner_workflow.git

add annotations to ScannerWorkflowEvent
---

diff --git a/scanner_workflow.py b/scanner_workflow.py
index 453f087..ab7b907 100755
--- 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: