diff options
| author | Tim Graham <timograham@gmail.com> | 2015-06-10 17:24:04 -0400 |
|---|---|---|
| committer | Tim Graham <timograham@gmail.com> | 2015-06-15 13:43:22 -0400 |
| commit | 3872a33132a4bb6aa22b237927597bbfdf6f21d7 (patch) | |
| tree | fc3f7de9a6eb0169751b2493b11153697a8bb0dd /django/utils/inspect.py | |
| parent | 4b600ed24488b2d1fc0fd59fc883a84ee745fcf8 (diff) | |
Fixed #24979 -- Removed usage of inspect.getargspec().
Diffstat (limited to 'django/utils/inspect.py')
| -rw-r--r-- | django/utils/inspect.py | 80 |
1 files changed, 80 insertions, 0 deletions
diff --git a/django/utils/inspect.py b/django/utils/inspect.py new file mode 100644 index 0000000000..3e3ad0ac23 --- /dev/null +++ b/django/utils/inspect.py @@ -0,0 +1,80 @@ +from __future__ import absolute_import + +import inspect + +from django.utils import six + + +def getargspec(func): + if six.PY2: + return inspect.getargspec(func) + + sig = inspect.signature(func) + args = [ + p.name for p in sig.parameters.values() + if p.kind == inspect.Parameter.POSITIONAL_OR_KEYWORD + ] + varargs = [ + p.name for p in sig.parameters.values() + if p.kind == inspect.Parameter.VAR_POSITIONAL + ] + varargs = varargs[0] if varargs else None + varkw = [ + p.name for p in sig.parameters.values() + if p.kind == inspect.Parameter.VAR_KEYWORD + ] + varkw = varkw[0] if varkw else None + defaults = [ + p.default for p in sig.parameters.values() + if p.kind == inspect.Parameter.POSITIONAL_OR_KEYWORD and p.default is not p.empty + ] or None + return args, varargs, varkw, defaults + + +def get_func_args(func): + if six.PY2: + argspec = inspect.getargspec(func) + return argspec.args[1:] # ignore 'self' + + sig = inspect.signature(func) + return [ + arg_name for arg_name, param in sig.parameters.items() + if param.kind == inspect.Parameter.POSITIONAL_OR_KEYWORD + ] + + +def func_accepts_kwargs(func): + if six.PY2: + # Not all callables are inspectable with getargspec, so we'll + # try a couple different ways but in the end fall back on assuming + # it is -- we don't want to prevent registration of valid but weird + # callables. + try: + argspec = inspect.getargspec(func) + except TypeError: + try: + argspec = inspect.getargspec(func.__call__) + except (TypeError, AttributeError): + argspec = None + return not argspec or argspec[2] is not None + + return any( + p for p in inspect.signature(func).parameters.values() + if p.kind == p.VAR_KEYWORD + ) + + +def func_has_no_args(func): + args = inspect.getargspec(func)[0] if six.PY2 else [ + p for p in inspect.signature(func).parameters.values() + if p.kind == p.POSITIONAL_OR_KEYWORD and p.default is p.empty + ] + return len(args) == 1 + + +def func_supports_parameter(func, parameter): + if six.PY3: + return parameter in inspect.signature(func).parameters + else: + args, varargs, varkw, defaults = inspect.getargspec(func) + return parameter in args |
