我可以获取traceback中调用的最后一个函数的参数吗?怎么样?
我想制作标准错误的捕手,以制作可读代码,同时向用户提供详细信息。
在下面的例子中,我希望GET_PARAMS返回一个提供给os.chown的参数元组。检查Alex Martelli建议的inspect
模块,我找不到。
def catch_errors(fn):
def decorator(*args, **kwargs):
try:
return fn(*args, **kwargs)
except (IOError, OSError):
msg = sys.exc_info()[2].tb_frame.f_locals['error_message']
quit(msg.format(SEQUENCE_OF_PARAMETERS_OF_THE_LAST_FUNCTION_CALLED)\
+ '\nError #{0[0]}: {0[1]}'.format(sys.exc_info()[1].args), 1)
return decorator
@catch_errors
def do_your_job():
error_message = 'Can\'t change folder ownership \'{0}\' (uid:{1}, gid:{2})'
os.chown('/root', 1000, 1000) # note that params aren't named vars.
if __name == '__main__' and os.getenv('USERNAME') != 'root':
do_your_job()
(感谢装饰师的Jim Robert)
答案 0 :(得分:5)
对于此类检查任务,请始终首先考虑标准库中的模块inspect
。在这里,inspect.getargvalues为您提供给定框架的参数值,inspect.getinnerframes为您提供回溯对象中感兴趣的帧。
答案 1 :(得分:3)
以下是此类功能的示例以及您无法解决的一些问题:
import sys
def get_params(tb):
while tb.tb_next:
tb = tb.tb_next
frame = tb.tb_frame
code = frame.f_code
argcount = code.co_argcount
if code.co_flags & 4: # *args
argcount += 1
if code.co_flags & 8: # **kwargs
argcount += 1
names = code.co_varnames[:argcount]
params = {}
for name in names:
params[name] = frame.f_locals.get(name, '<deleted>')
return params
def f(a, b=2, c=3, *d, **e):
del c
c = 4
e['g'] = 6
assert False
try:
f(1, f=5)
except:
print get_params(sys.exc_info()[2])
输出结果为:
{'a': 1, 'c': 4, 'b': 2, 'e': {'g': 6, 'f': 5}, 'd': ()}
我没有使用inspect.getinnerframes()
来显示另一种获取所需帧的方法。虽然它简化了一点,但它也做了一些额外的工作,而你需要的是相对较慢(inspect.getinnerframes()
读取回溯中每个模块的源文件;这对于一个调试调用并不重要,但可能是一个在其他情况下的问题)。
答案 2 :(得分:0)
使用装饰器来解决您尝试实现的问题是异常处理程序获取的帧是do_your_job()
s,而不是os.listdir()
s,os.makedirs()
s或{{ 1}}秒。因此,您要打印的信息是os.chown()
的参数。为了获得我认为你想要的行为,你必须装饰你正在调用的所有库函数。