Add typing to ContextManager.__init__

gorakhargosh · Oct 6, 2023 · e486f42 · e486f42
1 parent 715b6e3
commit e486f42
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/src/watchdog/utils/dirsnapshot.py b/src/watchdog/utils/dirsnapshot.py
@@ -51,6 +51,7 @@
 import errno
 import os
 from stat import S_ISDIR
+from typing import Callable, Iterator, Optional
 
 
 class DirectorySnapshotDiff:
@@ -250,11 +251,11 @@ class ContextManager:
 
         def __init__(
             self,
-            path,
-            recursive=True,
-            stat=os.stat,
-            listdir=os.scandir,
-            ignore_device=False,
+            path: str,
+            recursive: bool = True,
+            stat: Callable[[str], os.stat_result] = os.stat,
+            listdir: Callable[[Optional[str]], Iterator[os.DirEntry]] = os.scandir,
+            ignore_device: bool = False,
         ):
             self.path = path
             self.recursive = recursive