summaryrefslogtreecommitdiff
path: root/astroid
diff options
context:
space:
mode:
authorJacob Walls <jacobtylerwalls@gmail.com>2023-04-15 15:26:51 -0400
committerJacob Walls <jacobtylerwalls@gmail.com>2023-04-16 09:40:26 -0400
commit309ca8dd7b4d0c95cb3cbb25180857f44050b672 (patch)
tree7311b8e1f75345afdc264c6fa217ce03be79c0ed /astroid
parent85b698466afe12c750842dc2522f2823365d9fbc (diff)
downloadastroid-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.py10
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 = [