Support pylint 2.0 UnknownMessage rename to UnknownMessageError

prospector-dev · Mar 10, 2017 · aee69f4 · aee69f4
1 parent c893531
commit aee69f4
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/prospector/tools/pylint/__init__.py b/prospector/tools/pylint/__init__.py
@@ -4,7 +4,10 @@
 import sys
 import os
 from pylint.config import find_pylintrc
-from pylint.utils import UnknownMessage
+try:
+    from pylint.utils import UnknownMessage
+except ImportError:
+    from pylint.utils import UnknownMessageError as UnknownMessage
 from prospector.message import Message, Location
 from prospector.tools.base import ToolBase
 from prospector.tools.pylint.collector import Collector

diff --git a/prospector/tools/pylint/collector.py b/prospector/tools/pylint/collector.py
@@ -1,6 +1,9 @@
 from __future__ import absolute_import
 from pylint.reporters import BaseReporter
-from pylint.utils import UnknownMessage
+try:
+    from pylint.utils import UnknownMessage
+except ImportError:
+    from pylint.utils import UnknownMessageError as UnknownMessage
 from prospector.message import Location, Message