Profiling: fix do_profile decorator

[MustafaJafar]

Changelog Description

• Fix do_profile decorator

Testing notes:

In AYON console, running these cases should work:

import os
from ayon_core.lib.profiling import do_profile

@do_profile
def get_bundle_settings(bundle_name):
    import ayon_api 

    return ayon_api.get_addons_settings(bundle_name)


bundle_name = os.getenv("AYON_BUNDLE_NAME")
settings = get_bundle_settings(bundle_name)

import os
from ayon_core.lib.profiling import do_profile

@do_profile()
def get_bundle_settings(bundle_name):
    import ayon_api 

    return ayon_api.get_addons_settings(bundle_name)


bundle_name = os.getenv("AYON_BUNDLE_NAME")
settings = get_bundle_settings(bundle_name)

import os
from ayon_core.lib.profiling import do_profile

@do_profile(r"c:\dev\data.prof")
def get_bundle_settings(bundle_name):
    import ayon_api 

    return ayon_api.get_addons_settings(bundle_name)


bundle_name = os.getenv("AYON_BUNDLE_NAME")
settings = get_bundle_settings(bundle_name)

[iLLiCiTiT]

Usually this is done using required kwargs, or it is one of the options.

Suggested change

	def do_profile(to_file=None):
	"""Wraps function in profiler run and print stat after it is done.
	Args:
	to_file (str, optional): If specified, dumps stats into the file
	instead of printing.
	"""
	def _do_profile(fn):
	@functools.wraps(fn)
	def profiled(*args, **kwargs):
	profiler = cProfile.Profile()
	try:
	profiler.enable()
	res = fn(*args, **kwargs)
	profiler.disable()
	return res
	finally:
	if to_file:
	profiler.dump_stats(to_file)
	else:
	profiler.print_stats()
	return profiled
	# If used as @do_profile, to_file is the function
	if callable(to_file):
	fn = to_file
	to_file = None
	return _do_profile(fn)
	if to_file:
	to_file = to_file.format(pid=os.getpid())
	def profiled(*args, **kwargs):
	profiler = cProfile.Profile()
	try:
	profiler.enable()
	res = fn(*args, **kwargs)
	profiler.disable()
	return res
	finally:
	if to_file:
	profiler.dump_stats(to_file)
	else:
	profiler.print_stats()
	return _do_profile
	def do_profile(*args, to_file=None):
	"""Wraps function in profiler run and print stat after it is done.
	Args:
	to_file (str, optional): If specified, dumps stats into the file
	instead of printing.
	"""
	def _do_profile(fn):
	@functools.wraps(fn)
	def profiled(*args, **kwargs):
	profiler = cProfile.Profile()
	try:
	profiler.enable()
	res = fn(*args, **kwargs)
	profiler.disable()
	return res
	finally:
	if to_file:
	profiler.dump_stats(to_file)
	else:
	profiler.print_stats()
	return profiled
	if to_file:
	to_file = to_file.format(pid=os.getpid())
	if args:
	return _do_profile(args[0])
	return _do_profile