mypy: add annotate.py and jsonreport.py

1 files changed, 14 insertions, 4 deletions

diff --git a/coverage/annotate.py b/coverage/annotate.py
index 07ff644d..c92c29b7 100644
--- a/coverage/annotate.py
+++ b/coverage/annotate.py
@@ -3,12 +3,22 @@
 
 """Source file annotation for coverage.py."""
 
+from __future__ import annotations
+
 import os
 import re
 
+from typing import Iterable, Optional, TYPE_CHECKING
+
 from coverage.files import flat_rootname
 from coverage.misc import ensure_dir, isolate_module
+from coverage.plugin import FileReporter
 from coverage.report import get_analysis_to_report
+from coverage.results import Analysis
+from coverage.types import TMorf
+
+if TYPE_CHECKING:
+    from coverage import Coverage
 
 os = isolate_module(os)
 
@@ -35,15 +45,15 @@ class AnnotateReporter:
 
     """
 
-    def __init__(self, coverage):
+    def __init__(self, coverage: Coverage) -> None:
         self.coverage = coverage
         self.config = self.coverage.config
-        self.directory = None
+        self.directory: Optional[str] = None
 
     blank_re = re.compile(r"\s*(#|$)")
     else_re = re.compile(r"\s*else\s*:\s*(#|$)")
 
-    def report(self, morfs, directory=None):
+    def report(self, morfs: Optional[Iterable[TMorf]], directory: Optional[str]=None) -> None:
         """Run the report.
 
         See `coverage.report()` for arguments.
@@ -54,7 +64,7 @@ class AnnotateReporter:
         for fr, analysis in get_analysis_to_report(self.coverage, morfs):
             self.annotate_file(fr, analysis)
 
-    def annotate_file(self, fr, analysis):
+    def annotate_file(self, fr: FileReporter, analysis: Analysis) -> None:
         """Annotate a single file.
 
         `fr` is the FileReporter for the file to annotate.