Rooba
/
pydu
mirror of https://github.com/flaggo/pydu.git
1
1
Fork 0
Code Issues Packages Projects Releases Wiki Activity

merge independent implement of inspect for PY2 and PY3 into on one

This commit is contained in:
Prodesire 2017-11-20 08:24:00 +08:00
parent 7484b379f2
commit 4aa8907956
1 changed files with 83 additions and 86 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

169
pydu/inspect.py
View File

@ -5,77 +5,18 @@ import inspect
from pydu.compat import PY2 from pydu.compat import PY2
######################################### def getargspec(func):
# funcs for PY2 """
######################################### Get the names and default values of a function's parameters.
if PY2:
def getargspec(func): A tuple of four things is returned: (args, varargs, keywords, defaults).
'args' is a list of the argument names, including keyword-only argument names.
'varargs' and 'keywords' are the names of the * and ** parameters or None.
'defaults' is an n-tuple of the default values of the last n parameters.
"""
if PY2:
return inspect.getargspec(func) return inspect.getargspec(func)
else:
def get_func_args(func):
argspec = inspect.getargspec(func)
if inspect.ismethod(func):
return argspec.args[1:] # ignore 'self'
return argspec.args
def get_func_full_args(func):
"""
Return a list of (argument name, default value) tuples. If the argument
does not have a default value, omit it in the tuple. Arguments such as
*args and **kwargs are also included.
"""
argspec = inspect.getargspec(func)
if inspect.ismethod(func):
args = argspec.args[1:] # ignore 'self'
else:
args = argspec.args
defaults = argspec.defaults or []
# Split args into two lists depending on whether they have default value
no_default = args[:len(args) - len(defaults)]
with_default = args[len(args) - len(defaults):]
# Join the two lists and combine it with default values
args = [(arg,) for arg in no_default] + zip(with_default, defaults)
# Add possible *args and **kwargs and prepend them with '*' or '**'
varargs = [('*' + argspec.varargs,)] if argspec.varargs else []
kwargs = [('**' + argspec.keywords,)] if argspec.keywords else []
return args + varargs + kwargs
def func_accepts_kwargs(func):
# 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
def func_accepts_var_args(func):
"""
Return True if function 'func' accepts positional arguments *args.
"""
return inspect.getargspec(func)[1] is not None
def func_supports_parameter(func, parameter):
args, varargs, varkw, defaults = inspect.getargspec(func)
if inspect.ismethod(func):
args = args[1:] # ignore 'self'
return parameter in args + [varargs, varkw]
#########################################
# funcs for PY3
#########################################
else:
def getargspec(func):
sig = inspect.signature(func) sig = inspect.signature(func)
args = [ args = [
p.name for p in sig.parameters.values() p.name for p in sig.parameters.values()
@ -99,7 +40,17 @@ else:
return args, varargs, varkw, defaults return args, varargs, varkw, defaults
def get_func_args(func): def get_func_args(func):
"""
Return a list of the argument names. Arguments such as
*args and **kwargs are not included.
"""
if PY2:
argspec = inspect.getargspec(func)
if inspect.ismethod(func):
return argspec.args[1:] # ignore 'self'
return argspec.args
else:
sig = inspect.signature(func) sig = inspect.signature(func)
print(sig) print(sig)
return [ return [
@ -108,12 +59,29 @@ else:
] ]
def get_func_full_args(func): def get_func_full_args(func):
""" """
Return a list of (argument name, default value) tuples. If the argument Return a list of (argument name, default value) tuples. If the argument
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.
""" """
if PY2:
argspec = inspect.getargspec(func)
if inspect.ismethod(func):
args = argspec.args[1:] # ignore 'self'
else:
args = argspec.args
defaults = argspec.defaults or []
# Split args into two lists depending on whether they have default value
no_default = args[:len(args) - len(defaults)]
with_default = args[len(args) - len(defaults):]
# Join the two lists and combine it with default values
args = [(arg,) for arg in no_default] + zip(with_default, defaults)
# Add possible *args and **kwargs and prepend them with '*' or '**'
varargs = [('*' + argspec.varargs,)] if argspec.varargs else []
kwargs = [('**' + argspec.keywords,)] if argspec.keywords else []
return args + varargs + kwargs
else:
sig = inspect.signature(func) sig = inspect.signature(func)
args = [] args = []
for arg_name, param in sig.parameters.items(): for arg_name, param in sig.parameters.items():
@ -132,32 +100,61 @@ else:
return args return args
def func_accepts_kwargs(func): def func_accepts_kwargs(func):
"""
Check whether or not the func accepts kwargs.
"""
# 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.
if PY2:
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
else:
return any( return any(
p for p in inspect.signature(func).parameters.values() p for p in inspect.signature(func).parameters.values()
if p.kind == p.VAR_KEYWORD if p.kind == p.VAR_KEYWORD
) )
def func_accepts_var_args(func): def func_accepts_var_args(func):
""" """
Return True if function 'func' accepts positional arguments *args. Check whether or not the func accepts var args.
""" """
if PY2:
return inspect.getargspec(func)[1] is not None
else:
return any( return any(
p for p in inspect.signature(func).parameters.values() p for p in inspect.signature(func).parameters.values()
if p.kind == p.VAR_POSITIONAL if p.kind == p.VAR_POSITIONAL
) )
def func_supports_parameter(func, parameter): def func_supports_parameter(func, parameter):
"""
Check whether or the func supports the given parameter.
"""
if PY2:
args, varargs, varkw, defaults = inspect.getargspec(func)
if inspect.ismethod(func):
args = args[1:] # ignore 'self'
return parameter in args + [varargs, varkw]
else:
parameters = [name for name in inspect.signature(func).parameters if name != 'self'] parameters = [name for name in inspect.signature(func).parameters if name != 'self']
return parameter in parameters return parameter in parameters
#########################################
# common funcs
#########################################
def func_has_no_args(func): def func_has_no_args(func):
"""
Check whether or not the func has any args.
"""
args = inspect.getargspec(func)[0] if PY2 else [ args = inspect.getargspec(func)[0] if PY2 else [
p for name, p in inspect.signature(func).parameters.items() p for name, p in inspect.signature(func).parameters.items()
if p.kind == p.POSITIONAL_OR_KEYWORD and name != 'self' if p.kind == p.POSITIONAL_OR_KEYWORD and name != 'self'