From 61409a412373db179297329d805d290b8ac4396d Mon Sep 17 00:00:00 2001 From: Daniel Hahler Date: Sun, 31 Mar 2019 19:18:24 +0200 Subject: [PATCH] Fix GetCoveragePyLines for coverage.py 5 --- python/vim_coverage.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/python/vim_coverage.py b/python/vim_coverage.py index 5121594..9c105af 100644 --- a/python/vim_coverage.py +++ b/python/vim_coverage.py @@ -33,10 +33,15 @@ def GetCoveragePyLines(path, source_file): cov.load() finally: os.chdir(prev_cwd) + try: + data = cov.get_data() + except AttributeError: + # Coverage.py before 5. + data = cov.data try: # Coverage.py 4.0 and higher. - covered_lines = cov.data.lines(source_file) + covered_lines = data.lines(source_file) except TypeError: - covered_lines = cov.data.line_data()[source_file] + covered_lines = data.line_data()[source_file] uncovered_lines = cov.analysis(source_file)[2] return (covered_lines or [], uncovered_lines)