从traceback获取最后一个函数的调用参数?

时间:2009-10-30 16:02:46

标签: python exception-handling traceback

我可以获取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

3 个答案:

答案 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()的参数。为了获得我认为你想要的行为,你必须装饰你正在调用的所有库函数。