Support initial calls for console bench function

This commit is contained in:
shortcutme 2018-01-19 02:32:44 +01:00
parent 99fdbcd1fe
commit dedf6cd92f
No known key found for this signature in database
GPG key ID: 5B63BAE6CB9613AE

View file

@ -637,10 +637,12 @@ class UiRequest(object):
sites = self.server.sites
main = sys.modules["main"]
def bench(code, times=100):
def bench(code, times=100, init=None):
sites = self.server.sites
main = sys.modules["main"]
s = time.time()
if init:
exec(init, globals(), locals())
for _ in range(times):
back = eval(code, globals(), locals())
return ["%s run: %.3fs" % (times, time.time() - s), back]