Профилирование выбранных функций в Python ⇐ Python

[Anonymous]

Я могу профилировать и визуализировать все свое приложение с помощью:
python -m cProfile -o program.prof my_program.py
snakeviz program.prof

Можно ли написать декоратор, который будет профилировать только некоторые функции?
В конечном итоге это будет выглядеть так:# This is the profiling decorator function.
def profile_function(func):
def wrapper(*args, **kwargs):

# TODO write profiling information to program.prof

return func(*args, **kwargs)
return wrapper

@profile_function
def function1():
for _ in range(0, 1000):
_**2

@profile_function
def function2():
for _ in range(0, 100000):
_**4

# Do not profile function3
def function3():
return

function1()
function2()
function3()


Подробнее здесь: https://stackoverflow.com/questions/793 ... -in-python