From dedf6cd92fc2b50afd620e8ae7ebb2e2f237ea0f Mon Sep 17 00:00:00 2001 From: shortcutme Date: Fri, 19 Jan 2018 02:32:44 +0100 Subject: [PATCH] Support initial calls for console bench function --- src/Ui/UiRequest.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/Ui/UiRequest.py b/src/Ui/UiRequest.py index 04c77599..1fffa890 100644 --- a/src/Ui/UiRequest.py +++ b/src/Ui/UiRequest.py @@ -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]