From 586726322069bab62f33dccab70101ca76f595c2 Mon Sep 17 00:00:00 2001 From: Ned Batchelder Date: Sat, 31 Dec 2022 19:36:23 -0500 Subject: mypy: add annotate.py and jsonreport.py --- coverage/annotate.py | 18 ++++++++++++++---- 1 file changed, 14 insertions(+), 4 deletions(-) (limited to 'coverage/annotate.py') 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. -- cgit v1.2.1