Factor record_issue into its own method

This commit is contained in:
Gilles Peskine 2018-11-23 21:11:30 +01:00
parent 556d7d9e3b
commit 043980585c

View File

@ -43,11 +43,14 @@ class IssueTracker(object):
for i, line in enumerate(iter(f.readline, b"")): for i, line in enumerate(iter(f.readline, b"")):
self.check_file_line(filepath, line, i + 1) self.check_file_line(filepath, line, i + 1)
def record_issue(self, filepath, line_number):
if filepath not in self.files_with_issues.keys():
self.files_with_issues[filepath] = []
self.files_with_issues[filepath].append(line_number)
def check_file_line(self, filepath, line, line_number): def check_file_line(self, filepath, line, line_number):
if self.issue_with_line(line): if self.issue_with_line(line):
if filepath not in self.files_with_issues.keys(): self.record_issue(filepath, line_number)
self.files_with_issues[filepath] = []
self.files_with_issues[filepath].append(line_number)
def output_file_issues(self, logger): def output_file_issues(self, logger):
if self.files_with_issues.values(): if self.files_with_issues.values():