如何从pycallgraph中删除库模块或特定函数

Question

我使用愈伤图来分析我的代码性能。但是，调用图非常混乱，有很多对系统函数的调用，以及一些我不想文档化的函数。我怎样才能阻止pycallgraph报告这些电话呢？

Answer 1

Pycallgraph提供了过滤功能，可以过滤出您希望从调用图中排除的任何模块、类或函数。在启动跟踪并传递给pycallgraph之前，应该定义以下函数

示例

def filtercalls(call_stack, modul, clas, func, full):
    mod_ignore = ['shutil','scipy.optimize','re','os','sys','json']
    func_ignore = ['CustomFunctionName','pdbcall']
    clas_ignore = ['pdb']
    return modul not in mod_ignore and func not in func_ignore and clas not in clas_ignore

pycallgraph跟踪开始是

pycallgraph.start_trace(filter_func=filtercalls)

这样，您在过滤器调用中提供的任何模块、类或函数都将被删除。请注意，在标准库中只提供模块名是不够的。因此，在mod_ignore中包含numpy仍然会导致numpy.core被包括在内。