diff --git a/src/wiktextract_wrapper.py b/src/wiktextract_wrapper.py index 42532d4d73383dc0a8ae38c8b12e1a7e110a765b..c4e33f1cc0f7abf9f82d1b885193e5c6d6a69733 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: