Open eltoder opened 1 year ago
Thanks for the report! All proposed solutions have in common that they need to find out which variables are defined in the scope of the class. Since Vulture currently has no notion of scope, they are tricky to implement. If you see a solution, I'm all ears :-)
@jendrikseipp That sounds like an even bigger problem :-) Adding scope information should improve the accuracy quite a bit. Python scoping rules are not very complicated.
There are some libraries that enrich the abstract syntax tree (on which Vulture operates) with scoping information, but I haven't looked deeper into them. I think that could be a good starting point.
Can I work on this?
Sure, happy to see progress here! It might be a good idea to discuss the high level approach though, before you spend time on implementing the details. If you have a high-level plan, I'm happy to discuss it.
@jendrikseipp looked into this a little more, and wanted to run the current implementation idea by you...
Progress thus far:
scan
function in core.py
visit
functionvisit
function decides what helper to use based on the type of nodevisit_Call
and visit_ClassDef
functions because these are invoked when the list()
function is called, and when a class is defined with the Enum
subclass.Based on these findings, here is a high level approach of the solution:
visit_Call
, check if the node.func.id=='list'
- are we calling the list function?node.args[0].id
) and use the scoping information provided by asttokens to check if the argument is a classEnum
is a superclass.defined_variables
Yes, your explanation makes sense and the approach is correct, I'd say. (In the last step, you probably mean "used_variables".)
The only thing I'm unsure about is the AST library. Have you compared asttokens with https://github.com/kavigupta/ast_scope and https://github.com/pylint-dev/astroid ? We need to pick one that does the job, is well-maintained and doesn't have too many dependencies.
Enum classes allow iteration over the class, which returns all enum members. This is often convenient particularly when writing tests. Unfortunately, vulture does not understand this and reports enum members as unused. For example:
produces these errors:
This can be whitelisted, but it's a bit tedious when there are many enum members.
Some ideas on how this can be improved:
list(E)
,tuple(E)
,a, b, c = E
,for e in E
and mark all members as used.__all__
, all members can be considered used?