Mark imports in __all__ as used

CHANGELOG.md

@@ -1,4 +1,5 @@
 * Add whitelist for `pint.UnitRegistry.default_formatter` (Ben Elliston, #258).
+* Mark imports in `__all__` as used (kreathon, #172).
 
 # 2.4 (2022-05-19)
 

tests/test_imports.py

@@ -209,6 +209,27 @@ def test_import_alias_use_alias(v):
     check(v.unused_vars, [])
 
 
+def test_import_with__all__(v):
+    v.scan(
+        """\
+# define.py
+class Foo:
+    pass
+
+class Bar:
+    pass
+
+# main.py
+from define import Foo, Bar
+
+__all__ = ["Foo"]
+
+"""
+    )
+    check(v.defined_imports, ["Foo", "Bar"])
+    check(v.unused_imports, ["Bar"])
+
+
 def test_ignore_init_py_files(v):
     v.scan(
         """\

vulture/core.py

@@ -53,6 +53,16 @@ def _is_test_file(filename):
     )
 
 
+def _assigns_special_variable__all__(node):
+    assert isinstance(node, ast.Assign)
+    name_targets = {
+        target.id for target in node.targets if isinstance(target, ast.Name)
+    }
+    return "__all__" in name_targets and isinstance(
+        node.value, (ast.List, ast.Tuple)
+    )
+
+
 def _ignore_class(filename, class_name):
     return _is_test_file(filename) and "Test" in class_name
 
@@ -616,6 +626,13 @@ def visit_Name(self, node):
         elif isinstance(node.ctx, (ast.Param, ast.Store)):
             self._define_variable(node.id, node)
 
+    def visit_Assign(self, node):
+        if _assigns_special_variable__all__(node):
+            assert isinstance(node.value, (ast.List, ast.Tuple))
+            for elt in node.value.elts:
+                if isinstance(elt, ast.Str):
+                    self.used_names.add(elt.s)
+
     def visit_While(self, node):
         self._handle_conditional_node(node, "while")