python - 确定函数的参数和关键字参数
问题描述
您如何确定对函数的有效调用的形式?
例如,假设我们有一个info
完成此任务的函数;info
可能会像这样工作(我愿意接受关于表示返回信息的更完整和更一致的方式的建议):
def foo():
pass
info(foo)
# { 'args': (), 'kwargs': {} }
def bar(a):
pass
info(bar)
# { 'args': ('a',), 'kwargs': {} }
def baz(a, b=42):
pass
info(baz)
# { 'args': ('a',), 'kwargs': { 'b': 42 } }
def qux(a, *args, b=42, **kwargs):
pass
info(qux)
# { 'args': ('a',), 'kwargs': { 'b': 42 }, 'optional': {'*args', '**kwargs'} }
该info
功能应该适用于任何功能。我不确定如何为每个模式编写示例返回:例如,help(range.__init__)
显示
# __init__(self, /, *args`, **kwargs)
我不确定这/
意味着什么。
from 的返回info
需要是可以计算的(通过合理的努力),以产生对' 参数的任意、正确的调用,例如,用于随机测试。info
解决方案
已经有一个用于此目的的函数,inspect.getfullargspec
它返回namedtuples
:
>>> import inspect
>>> inspect.getfullargspec(foo)
FullArgSpec(args=[], varargs=None, varkw=None, defaults=None, kwonlyargs=[], kwonlydefaults=None, annotations={})
>>> inspect.getfullargspec(qux)
FullArgSpec(args=['a'], varargs='args', varkw='kwargs', defaults=None, kwonlyargs=['b'], kwonlydefaults={'b': 42}, annotations={})
>>> inspect.getfullargspec(bar)
FullArgSpec(args=['a'], varargs=None, varkw=None, defaults=None, kwonlyargs=[], kwonlydefaults=None, annotations={})
>>> inspect.getfullargspec(baz)
FullArgSpec(args=['a', 'b'], varargs=None, varkw=None, defaults=(42,), kwonlyargs=[], kwonlydefaults=None, annotations={})
但是,如果你愿意,你可以从这里构建一个函数:
def info(func):
"""returns function argument info"""
specs = inspect.getfullargspec(func)
dict_ = {}
dict_['args'] = tuple(specs.args)
dict_['kwargs'] = {} if specs.kwonlydefaults is None else specs.kwonlydefaults
dict_['optional'] = set()
dict_['defaults'] = {} if specs.defaults is None else specs.defaults
if specs.varargs is not None:
dict_['optional'].add(f"*{specs.varargs}")
if specs.varkw is not None:
dict_['optional'].add(f"*{specs.varkw}")
if not dict_['optional']:
dict_['optional'] = {}
return dict_
>>> info(foo)
{'args': (), 'kwargs': {}, 'optional': {}, 'defaults': {}}
>>> info(qux)
{'args': ('a',), 'kwargs': {'b': 42}, 'optional': {'*args', '*kwargs'}, 'defaults': {}}
>>> info(bar)
{'args': ('a',), 'kwargs': {}, 'optional': {}, 'defaults': {}}
>> info(baz)
{'args': ('a', 'b'), 'kwargs': {}, 'optional': {}, 'defaults': (42,)}
42
inbaz
不是关键字参数,它是默认参数。因为调用时不需要提供关键字b
。
*
引用关键字中的只help(__init__)
需要跟随的参数,即它告诉后面的参数必须是keyword-only
参数,类似地,前面的任何参数/
都必须是positional argument
,更多信息请参见PEP457
, PEP570
, PEP3102
。
其中许多信息可以从code
函数的固有对象中获得,它具有以下属性:
for attr in dir(qux.__code__):
if not attr.startswith('_'):
print(attr,':',getattr(qux.__code__, attr))
co_argcount : 1
co_cellvars : ()
co_code : b'd\x00S\x00'
co_consts : (None,)
co_filename : <ipython-input-43-6608913c4d65>
co_firstlineno : 1
co_flags : 79
co_freevars : ()
co_kwonlyargcount : 1
co_lnotab : b'\x00\x01'
co_name : qux
co_names : ()
co_nlocals : 4
co_stacksize : 1
co_varnames : ('a', 'b', 'args', 'kwargs')
但是,这些描述性不够,也不容易访问,并且仅供 python 内部使用。因此,除非您绝对需要自定义功能,否则inspect.getfullargspec
可能是最好的选择。
fullargspec
作为一个输出,namedtuple
您可以轻松访问不同的字段:
>>> argspecs = inspect.getfullargspec(qux)
>>> argspecs.args
['a']
>>> argspecs.kwonlydefaults
{'b': 42}
如果你想要一个 dict 你可以调用_asdict
结果的方法namedtuple
:
>>> inspect.getfullargspec(qux)._asdict() #gives OrderedDict
OrderedDict([('args', ['a']),
('varargs', 'args'),
('varkw', 'kwargs'),
('defaults', None),
('kwonlyargs', ['b']),
('kwonlydefaults', {'b': 42}),
('annotations', {})])
>>> dict(inspect.getfullargspec(qux)._asdict()) #call dict to get regular dict
{'args': ['a'],
'varargs': 'args',
'varkw': 'kwargs',
'defaults': None,
'kwonlyargs': ['b'],
'kwonlydefaults': {'b': 42},
'annotations': {}}
推荐阅读
- c - 如何让 CMake 目标打印失败的命令?
- python - Python:一行中的双重for循环
- kotlin - 如何使用 slack api 发布项目符号列表
- canvas - 为什么在 Canvas 的绘制应用程序中使用 beginPath 方法?
- .net - 如何创建数据库不匹配的多租户应用程序
- java - CosmosDatabase.createContainerIfNotExists() -> “具有指定 ID、名称或唯一索引的资源已存在。”
- visual-studio-code - 如何在 Vscode 启动时摆脱终端中的 WSL 信息显示
- django - 相对于后端改变前端的东西?
- database - CockroachDB如何从本地或附近节点而不是更远的节点进行读取
- pyspark - 在不使用熊猫的情况下转换pyspark数据框中的dbf文件的任何方法