Refs #31732 -- Fixed django.utils.inspect caching for bound methods.

Thanks Alexandr Artemyev for the report, and Simon Charette for the
original patch.
This commit is contained in:
Adam Johnson 2021-03-19 11:23:28 +00:00 committed by Mariusz Felisiak
parent ac72a216a7
commit 562898034f
2 changed files with 31 additions and 12 deletions

View File

@ -3,14 +3,23 @@ import inspect
@functools.lru_cache(maxsize=512) @functools.lru_cache(maxsize=512)
def _get_signature(func): def _get_func_parameters(func, remove_first):
return inspect.signature(func) parameters = tuple(inspect.signature(func).parameters.values())
if remove_first:
parameters = parameters[1:]
return parameters
def _get_callable_parameters(meth_or_func):
is_method = inspect.ismethod(meth_or_func)
func = meth_or_func.__func__ if is_method else meth_or_func
return _get_func_parameters(func, remove_first=is_method)
def get_func_args(func): def get_func_args(func):
sig = _get_signature(func) params = _get_callable_parameters(func)
return [ return [
arg_name for arg_name, param in sig.parameters.items() param.name for param in params
if param.kind == inspect.Parameter.POSITIONAL_OR_KEYWORD if param.kind == inspect.Parameter.POSITIONAL_OR_KEYWORD
] ]
@ -21,10 +30,10 @@ def get_func_full_args(func):
does not have a default value, omit it in the tuple. Arguments such as does not have a default value, omit it in the tuple. Arguments such as
*args and **kwargs are also included. *args and **kwargs are also included.
""" """
sig = _get_signature(func) params = _get_callable_parameters(func)
args = [] args = []
for arg_name, param in sig.parameters.items(): for param in params:
name = arg_name name = param.name
# Ignore 'self' # Ignore 'self'
if name == 'self': if name == 'self':
continue continue
@ -42,7 +51,7 @@ def get_func_full_args(func):
def func_accepts_kwargs(func): def func_accepts_kwargs(func):
"""Return True if function 'func' accepts keyword arguments **kwargs.""" """Return True if function 'func' accepts keyword arguments **kwargs."""
return any( return any(
p for p in _get_signature(func).parameters.values() p for p in _get_callable_parameters(func)
if p.kind == p.VAR_KEYWORD if p.kind == p.VAR_KEYWORD
) )
@ -52,7 +61,7 @@ def func_accepts_var_args(func):
Return True if function 'func' accepts positional arguments *args. Return True if function 'func' accepts positional arguments *args.
""" """
return any( return any(
p for p in _get_signature(func).parameters.values() p for p in _get_callable_parameters(func)
if p.kind == p.VAR_POSITIONAL if p.kind == p.VAR_POSITIONAL
) )
@ -60,11 +69,11 @@ def func_accepts_var_args(func):
def method_has_no_args(meth): def method_has_no_args(meth):
"""Return True if a method only accepts 'self'.""" """Return True if a method only accepts 'self'."""
count = len([ count = len([
p for p in _get_signature(meth).parameters.values() p for p in _get_callable_parameters(meth)
if p.kind == p.POSITIONAL_OR_KEYWORD if p.kind == p.POSITIONAL_OR_KEYWORD
]) ])
return count == 0 if inspect.ismethod(meth) else count == 1 return count == 0 if inspect.ismethod(meth) else count == 1
def func_supports_parameter(func, parameter): def func_supports_parameter(func, name):
return parameter in _get_signature(func).parameters return any(param.name == name for param in _get_callable_parameters(func))

View File

@ -22,6 +22,16 @@ class Person:
class TestInspectMethods(unittest.TestCase): class TestInspectMethods(unittest.TestCase):
def test_get_callable_parameters(self):
self.assertIs(
inspect._get_callable_parameters(Person.no_arguments),
inspect._get_callable_parameters(Person.no_arguments),
)
self.assertIs(
inspect._get_callable_parameters(Person().no_arguments),
inspect._get_callable_parameters(Person().no_arguments),
)
def test_get_func_full_args_no_arguments(self): def test_get_func_full_args_no_arguments(self):
self.assertEqual(inspect.get_func_full_args(Person.no_arguments), []) self.assertEqual(inspect.get_func_full_args(Person.no_arguments), [])
self.assertEqual(inspect.get_func_full_args(Person().no_arguments), []) self.assertEqual(inspect.get_func_full_args(Person().no_arguments), [])