Как использовать line_profiler для обернутой функции? ⇐ Python

[Anonymous]

Я пытался использовать line_profiler для обернутой функции. Однако вместо обернутой функции профилируется функция-оболочка. В приведенном ниже фрагменте исполняемого файла показано, что я вижу, что моя функция-оболочка @decorate профилируется вместо моей функции d().
Фрагмент: testy_profiler.py< /p>

Код: Выделить всё

from line_profiler import LineProfiler

def a():
print("a")

from functools import wraps
def decorate(func):
@wraps(func)
def wrapper(*args, **kwargs):
r = func(*args, **kwargs)
print("decorating")
return r
return wrapper

@decorate
def d():
print("d")

def go():
a()
d()

lprofiler = LineProfiler()
lprofiler.add_function(a)
lprofiler.add_function(d)
lp_wrapper = lprofiler(go)
lp_wrapper()
lprofiler.print_stats()

Выход:

Код: Выделить всё

$ python test_profiler.py
a
d
decorating
Timer unit: 1e-07 s

Total time: 1e-05 s
File: C:\Users\RENARDCH\Workspace\testy_profiler\test_profiler.py
Function: a at line 3

Line #      Hits         Time  Per Hit   % Time  Line Contents
==============================================================
3                                           def a():
4         1        100.0    100.0    100.0      print("a")

Total time: 4.4e-06 s
File: C:\Users\RENARDCH\Workspace\testy_profiler\test_profiler.py
Function: wrapper at line 8

Line #      Hits         Time  Per Hit   % Time  Line Contents
==============================================================
8                                                  @wraps(func)
9                                                  def wrapper(*args, **kwargs):
10         1         24.0     24.0     54.5                 r = func(*args, **kwargs)
11         1         18.0     18.0     40.9                 print("decorating")
12         1          2.0      2.0      4.5                 return r

Total time: 1.87e-05 s
File: C:\Users\RENARDCH\Workspace\testy_profiler\test_profiler.py
Function: go at line 19

Line #      Hits         Time  Per Hit   % Time  Line Contents
==============================================================
19                                           def go():
20         1        126.0    126.0     67.4      a()
21         1         61.0     61.0     32.6      d()

Это особенно создает проблему, когда две функции обернуты одной и той же оболочкой, поскольку для оболочки будет создано глобальное профилирование, а не профилирование для каждой функции индивидуально.

Код: Выделить всё

line_profiler

версия – 3.4.0

Подробнее здесь: https://stackoverflow.com/questions/741 ... d-function