Implement support for vulture 0.9 and up.

prospector-dev · Oct 20, 2016 · 3f4d873 · 3f4d873
1 parent a5606a9
commit 3f4d873
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/prospector/tools/vulture/__init__.py b/prospector/tools/vulture/__init__.py
@@ -27,7 +27,11 @@ def scavenge(self, _=None):
                 ))
                 continue
             self.file = module
-            self.scan(module_string)
+            self.filename = module
+            try:
+                self.scan(module_string, filename=module)
+            except TypeError:
+                self.scan(module_string)
 
     def get_messages(self):
         all_items = (
@@ -40,7 +44,11 @@ def get_messages(self):
         vulture_messages = []
         for code, template, items in all_items:
             for item in items:
-                loc = Location(item.file, None, None, item.lineno, -1)
+                try:
+                    filename = item.file
+                except AttributeError:
+                    filename = item.filename
+                loc = Location(filename, None, None, item.lineno, -1)
                 message_text = template % item
                 message = Message('vulture', code, loc, message_text)
                 vulture_messages.append(message)