From c0005e38292da32ec3697456289e60f3dde39411 Mon Sep 17 00:00:00 2001 From: Mathieu Loiseau <mathieu.loiseau@liris.cnrs.fr> Date: Tue, 23 Jan 2024 17:10:21 +0100 Subject: [PATCH] fast timing --- src/wiktextract_wrapper.py | 15 +++++++++++++-- 1 file changed, 13 insertions(+), 2 deletions(-) diff --git a/src/wiktextract_wrapper.py b/src/wiktextract_wrapper.py index 42532d4..c4e33f1 100644 --- a/src/wiktextract_wrapper.py +++ b/src/wiktextract_wrapper.py @@ -25,6 +25,11 @@ from wiktextract_context import get_wiktextract_context import json from importlib import metadata import git +import time +wiktextractime = -1 +convertime = -1 +startime = -1 + class Wiktextract: def __init__(self, wiktlang: str, wordlang: str): @@ -34,14 +39,18 @@ class Wiktextract: self.wxr = get_wiktextract_context(wiktlang, wordlang) def parse_page(self, title: str): + global wiktextractime + global convertime + global startime page = self.wxr.wtp.get_page(title) if not page: return None - + startime = time.time() result = parse_page(self.wxr, title, page.body) - + wiktextractime = time.time() converted_result = self.convert(result) + convertime = time.time() return converted_result #return result @@ -114,6 +123,8 @@ if __name__ == "__main__": wkstrkt = Wiktextract(args.wikt_language, args.word_language) result = wkstrkt.parse_page(args.entry) print(json.dumps(result, ensure_ascii=args.force_ascii)) + endtime = time.time() + print(f"Execution ({endtime-startime}â€):\n\twiktextract → {wiktextractime-startime}\n\tconversion → {convertime-wiktextractime}\n\tprint result → {endtime - convertime}") else: print("{'err':'You need to specify a word'}") if formerdir != None: -- GitLab