diff options
Diffstat (limited to 'astroid/brain/brain_argparse.py')
| -rw-r--r-- | astroid/brain/brain_argparse.py | 19 |
1 files changed, 6 insertions, 13 deletions
diff --git a/astroid/brain/brain_argparse.py b/astroid/brain/brain_argparse.py index 1b120b4e..d4899117 100644 --- a/astroid/brain/brain_argparse.py +++ b/astroid/brain/brain_argparse.py @@ -1,10 +1,4 @@ -from astroid import ( - MANAGER, - arguments, - nodes, - inference_tip, - UseInferenceDefault, -) +from astroid import MANAGER, arguments, nodes, inference_tip, UseInferenceDefault def infer_namespace(node, context=None): @@ -13,28 +7,27 @@ def infer_namespace(node, context=None): # Cannot make sense of it. raise UseInferenceDefault() - class_node = nodes.ClassDef('Namespace', 'docstring') + class_node = nodes.ClassDef("Namespace", "docstring") class_node.parent = node.parent for attr in set(callsite.keyword_arguments): fake_node = nodes.EmptyNode() fake_node.parent = class_node fake_node.attrname = attr class_node.instance_attrs[attr] = [fake_node] - return iter((class_node.instantiate_class(), )) + return iter((class_node.instantiate_class(),)) def _looks_like_namespace(node): func = node.func if isinstance(func, nodes.Attribute): return ( - func.attrname == 'Namespace' + func.attrname == "Namespace" and isinstance(func.expr, nodes.Name) - and func.expr.name == 'argparse' + and func.expr.name == "argparse" ) return False MANAGER.register_transform( - nodes.Call, inference_tip(infer_namespace), - _looks_like_namespace, + nodes.Call, inference_tip(infer_namespace), _looks_like_namespace ) |