flame-analyzer 0.1.5

例子

可以通过以下基准测试将火焰图保存到 SVG

from flame_analyzer import FileFlame


with FileFlame('./file_flame_test.svg'):
    # Some expensive piece of code.
    [len(u.email) for u in  User.objects.all()]

或者直接到 IPython notebook

from flame_analyzer import InlineFlame


with InlineFlame():
    # Some expensive piece of code.
    [len(u.email) for u in  User.objects.all()]

您还可以通过添加 width kwarg 来选择配置宽度

with FileFlame(
    './file_flame_test.svg', flame_width=1200,
    options={'title': 'This is my test title'}
):
    # some expensive piece of code
    [len(u.email) for u in  User.objects.all()]

扩展

默认情况下，IPython 和 Django 都是可选导入，这意味着您可以安装此库并在终端中使用它来调试您的应用程序代码，而无需安装它们。可以为其他数据库框架添加支持，或者如果您想通过创建自己的钩子并添加到您想要的输出火焰类型来挂钩上下文进入/退出方法

from flame_analyzer import InlineFlame

class CustomHook:
    """
    Append the time taken to execute to the flame graphs title.
    """
    def before(self):
        self.called_before = '< Called before code execution >'

    def after(self):
        self.called_after = '< Called after code execution >'

    def modify_flame_options(self, flame_options):
        title = flame_options.get('title', '')
        flame_options['title'] = self.called_before + ' --- ' + self.called_after
        return flame_options


class CustomInlineFlame(InlineFlame):
    hook_classes = (CustomHook,)

with CustomInlineFlame(flame_width=500):
    total_email_length = 0
    for u in User.objects.all():
        total_email_length += len(u.email)
    print(total_email_length)

输出 IPython 查看的 Graph

归功于以下项目：

• https://github.com/brendangregg/FlameGraph

• https://github.com/23andMe/djdt-flamegraph