diff options
author | Jacob Walls <jacobtylerwalls@gmail.com> | 2023-04-15 15:26:51 -0400 |
---|---|---|
committer | Jacob Walls <jacobtylerwalls@gmail.com> | 2023-04-16 09:40:26 -0400 |
commit | 309ca8dd7b4d0c95cb3cbb25180857f44050b672 (patch) | |
tree | 7311b8e1f75345afdc264c6fa217ce03be79c0ed /astroid | |
parent | 85b698466afe12c750842dc2522f2823365d9fbc (diff) | |
download | astroid-git-309ca8dd7b4d0c95cb3cbb25180857f44050b672.tar.gz |
Fix infer_call_result() crash on methods called with_metaclass() (#2118)
Diffstat (limited to 'astroid')
-rw-r--r-- | astroid/nodes/scoped_nodes/scoped_nodes.py | 10 |
1 files changed, 9 insertions, 1 deletions
diff --git a/astroid/nodes/scoped_nodes/scoped_nodes.py b/astroid/nodes/scoped_nodes/scoped_nodes.py index 530d9e6d..5945f05d 100644 --- a/astroid/nodes/scoped_nodes/scoped_nodes.py +++ b/astroid/nodes/scoped_nodes/scoped_nodes.py @@ -1697,10 +1697,18 @@ class FunctionDef(_base_nodes.MultiLineBlockNode, _base_nodes.Statement, Lambda) # generators, and filter it out later. if ( self.name == "with_metaclass" + and caller is not None and len(self.args.args) == 1 and self.args.vararg is not None ): - metaclass = next(caller.args[0].infer(context), None) + if isinstance(caller.args, Arguments): + metaclass = next(caller.args.args[0].infer(context), None) + elif isinstance(caller.args, list): + metaclass = next(caller.args[0].infer(context), None) + else: + raise TypeError( # pragma: no cover + f"caller.args was neither Arguments nor list; got {type(caller.args)}" + ) if isinstance(metaclass, ClassDef): try: class_bases = [ |