我怎样才能读取一个函数的签名,包括默认的参数值?
给定一个函数对象,我怎样才能得到它的签名? 例如,对于:
def myMethod(firt, second, third='something'): pass
我想得到"myMethod(firt, second, third='something')"
。
import inspect def foo(a, b, x='blah'): pass print(inspect.getargspec(foo)) # ArgSpec(args=['a', 'b', 'x'], varargs=None, keywords=None, defaults=('blah',))
但是,请注意,自Python 3.0以来, inspect.getargspec()
已被弃用。
3.0 – 3.4build议inspect.getfullargspec()
。 请参阅: https : //docs.python.org/3.4/library/inspect.html#inspect.getargspec
3.5+推荐inspect.signature()
。 请参阅: https : //docs.python.org/3.5/library/inspect.html#inspect.getargspec
可以find函数签名的最简单的方法是help(function)
:
>>> def function(arg1, arg2="foo", *args, **kwargs): pass >>> help(function) Help on function function in module __main__: function(arg1, arg2='foo', *args, **kwargs)
另外,在Python 3中,一个名为signature
的方法被添加到了inspect
模块中,该方法被devise用来表示可调用对象的签名及其返回注释 :
>>> from inspect import signature >>> def foo(a, *, b:int, **kwargs): ... pass >>> sig = signature(foo) >>> str(sig) '(a, *, b:int, **kwargs)' >>> str(sig.parameters['b']) 'b:int' >>> sig.parameters['b'].annotation <class 'int'>
#! /usr/bin/env python import inspect from collections import namedtuple DefaultArgSpec = namedtuple('DefaultArgSpec', 'has_default default_value') def _get_default_arg(args, defaults, arg_index): """ Method that determines if an argument has default value or not, and if yes what is the default value for the argument :param args: array of arguments, eg: ['first_arg', 'second_arg', 'third_arg'] :param defaults: array of default values, eg: (42, 'something') :param arg_index: index of the argument in the argument array for which, this function checks if a default value exists or not. And if default value exists it would return the default value. Example argument: 1 :return: Tuple of whether there is a default or not, and if yes the default value, eg: for index 2 ie for "second_arg" this function returns (True, 42) """ if not defaults: return DefaultArgSpec(False, None) args_with_no_defaults = len(args) - len(defaults) if arg_index < args_with_no_defaults: return DefaultArgSpec(False, None) else: value = defaults[arg_index - args_with_no_defaults] if (type(value) is str): value = '"%s"' % value return DefaultArgSpec(True, value) def get_method_sig(method): """ Given a function, it returns a string that pretty much looks how the function signature would be written in python. :param method: a python method :return: A string similar describing the pythong method signature. eg: "my_method(first_argArg, second_arg=42, third_arg='something')" """ # The return value of ArgSpec is a bit weird, as the list of arguments and # list of defaults are returned in separate array. # eg: ArgSpec(args=['first_arg', 'second_arg', 'third_arg'], # varargs=None, keywords=None, defaults=(42, 'something')) argspec = inspect.getargspec(method) arg_index=0 args = [] # Use the args and defaults array returned by argspec and find out # which arguments has default for arg in argspec.args: default_arg = _get_default_arg(argspec.args, argspec.defaults, arg_index) if default_arg.has_default: args.append("%s=%s" % (arg, default_arg.default_value)) else: args.append(arg) arg_index += 1 return "%s(%s)" % (method.__name__, ", ".join(args)) if __name__ == '__main__': def my_method(first_arg, second_arg=42, third_arg='something'): pass print get_method_sig(my_method) # my_method(first_argArg, second_arg=42, third_arg="something")
尝试调用对象的help
来了解它。
>>> foo = [1, 2, 3] >>> help(foo.append) Help on built-in function append: append(...) L.append(object) -- append object to end
也许晚了一点,但如果你还想保持参数的顺序和默认值 ,那么你可以使用抽象语法树模块(ast) 。
这是一个概念certificate(注意将代码sorting并将它们与缺省值匹配的代码绝对可以被改进/变得更加清晰):
import ast for class_ in [c for c in module.body if isinstance(c, ast.ClassDef)]: for method in [m for m in class_.body if isinstance(m, ast.FunctionDef)]: args = [] if method.args.args: [args.append([a.col_offset, a.id]) for a in method.args.args] if method.args.defaults: [args.append([a.col_offset, '=' + a.id]) for a in method.args.defaults] sorted_args = sorted(args) for i, p in enumerate(sorted_args): if p[1].startswith('='): sorted_args[i-1][1] += p[1] sorted_args = [k[1] for k in sorted_args if not k[1].startswith('=')] if method.args.vararg: sorted_args.append('*' + method.args.vararg) if method.args.kwarg: sorted_args.append('**' + method.args.kwarg) signature = '(' + ', '.join(sorted_args) + ')' print method.name + signature
如果你正在试图做的是打印function,然后使用pydoc。
import pydoc def foo(arg1, arg2, *args, **kwargs): '''Some foo fn''' pass >>> print pydoc.render_doc(foo).splitlines()[2] foo(arg1, arg2, *args, **kwargs)
如果您尝试实际分析function签名,则使用检查模块的argspec。 在将用户的钩子脚本函数validation为一个通用框架时,我必须这样做。