似乎函数调用计数不同的原因是因为第二次运行代码时myscript.py
导入的模块不会再次导入。
获得一致结果的第一种方法是在执行分析之前导入myscript.py
。 但是,这意味着如果我导入的模块在导入时执行某些任务,则不会对其进行分析。
prof = Profile()
progname = 'myscript.py'
sys.path.insert(0, os.path.dirname(progname))
modname, _ = os.path.splitext(os.path.basename(progname))
__import__(modname, globals(), locals(), [], 0)
with open(progname, 'rb') as fp:
code = compile(fp.read(), progname, 'exec')
globs = {
'__file__': progname,
'__name__': '__main__',
'__package__': None,
'__cached__': None,
}
prof.runctx(code, globs, None)
prof.create_stats()
print(len(prof.stats))
我找到的第二种方法是删除执行脚本时注册的所有模块。 优点是,如果我在GUI运行时修改源,它将重新加载更改。 我现在的缺点是,一些atexit注册处理程序现在崩溃,因为之前删除了所需的模块:
prof = Profile()
progname = 'myscript.py'
sys.path.insert(0, os.path.dirname(progname))
with open(progname, 'rb') as fp:
code = compile(fp.read(), progname, 'exec')
globs = {
'__file__': progname,
'__name__': '__main__',
'__package__': None,
'__cached__': None,
}
modules = sys.modules.copy()
prof.runctx(code, globs, None)
newmodes = [modname for modname in sys.modules if modname not in modules]
for modname in newmodes:
del sys.modules[modname]
prof.create_stats()
print(len(prof.stats))
最后,我找到的最好方法是在一个单独的进程中执行分析:
import concurrent.futures
import marshal
from cProfile import Profile
from pstats import Stats
import sys
progname = 'myscript.py'
with concurrent.futures.ProcessPoolExecutor() as executor:
future = executor.submit(_run, progname)
stats = Stats()
stats.stats = marshal.loads(future.result())
stats.get_top_level_stats()
def _run(progname):
sys.path.insert(0, os.path.dirname(progname))
with open(progname, 'rb') as fp:
code = compile(fp.read(), progname, 'exec')
globs = {
'__file__': progname,
'__name__': '__main__',
'__package__': None,
}
prof = Profile()
prof.runctx(code, globs, None)
prof.create_stats()
return marshal.dumps(prof.stats)