Make compatible with pylint 2.x as well as 1.x. Fixes #9

verboselogs/pylint.py

@@ -10,7 +10,20 @@
 .. _Pylint: https://pypi.python.org/pypi/pylint
 """
 
-from astroid import MANAGER, scoped_nodes, nodes
+from astroid import MANAGER
+
+try:
+    # astroid 2.x
+    from astroid import FunctionDef as FunctionDef
+    from astroid import Const as Const
+    from astroid import ClassDef as ClassDef
+    from astroid import Module as Module
+except ImportError:
+    # astroid 1.x
+    from astroid.scoped_nodes import Function as FunctionDef
+    from astroid.nodes import Const as Const
+    from astroid.scoped_nodes import Class as ClassDef
+    from astroid.scoped_nodes import Module as Module
 
 
 def register(linter):
@@ -21,16 +34,16 @@ def verboselogs_class_transform(cls):
     """Make Pylint aware of our custom logger methods."""
     if cls.name == 'RootLogger':
         for meth in ['notice', 'spam', 'success', 'verbose']:
-            cls.locals[meth] = [scoped_nodes.Function(meth, None)]
+            cls.locals[meth] = [FunctionDef(meth, None)]
 
 
 def verboselogs_module_transform(mod):
     """Make Pylint aware of our custom log levels."""
     if mod.name == 'logging':
         for const in ['NOTICE', 'SPAM', 'SUCCESS', 'VERBOSE']:
-            mod.locals[const] = [nodes.Const(const)]
+            mod.locals[const] = [Const(const)]
 
 
 # Register the above methods with Pylint.
-MANAGER.register_transform(scoped_nodes.Class, verboselogs_class_transform)
-MANAGER.register_transform(scoped_nodes.Module, verboselogs_module_transform)
+MANAGER.register_transform(ClassDef, verboselogs_class_transform)
+MANAGER.register_transform(Module, verboselogs_module_transform)