From c13887b253120b3efe7866e4fa5d3cda4f400c33 Mon Sep 17 00:00:00 2001 From: Mathieu Loiseau <mathieu.loiseau@liris.cnrs.fr> Date: Tue, 23 Jan 2024 16:51:38 +0100 Subject: [PATCH] clean (a little bit) 6c829c435fff9f4fc470d4510bf96dd702487903 --- .gitignore | 1 + src/wiktextract_wrapper.py | 43 ++++++++++++++++++++------------------ 2 files changed, 24 insertions(+), 20 deletions(-) diff --git a/.gitignore b/.gitignore index 314250c..66a38c4 100644 --- a/.gitignore +++ b/.gitignore @@ -136,6 +136,7 @@ celerybeat.pid env/ venv/ ENV/ +liveq/ env.bak/ venv.bak/ diff --git a/src/wiktextract_wrapper.py b/src/wiktextract_wrapper.py index 3703823..f6ac13f 100644 --- a/src/wiktextract_wrapper.py +++ b/src/wiktextract_wrapper.py @@ -1,11 +1,23 @@ -import sys -import os - -# Specify the path to your project directory -project_directory = '/home/enzosim/live-query-wiktextract' +if __name__ == "__main__": + import os + import argparse + from argparse import RawTextHelpFormatter #pour le formattage de l'aide + parser = argparse.ArgumentParser(formatter_class=RawTextHelpFormatter, description="""Wiktextract wrapper + \033[1m\033[32mex :\033[0m + ‣\033[0m\033[32m./wiktextract_wrapper.py -l en -w fr -e yellow\033[0m""") + parser.add_argument("-l", "--word_language", help="language of the sought entry", type=str, default = "en") + parser.add_argument("-w", "--wikt_language", help="language of the wiktionary", type=str, default = "en") + parser.add_argument("-e", "--entry", help="the entry", type=str, default=None) + parser.add_argument("-z", "--zero_config", help="Don't use if you know how to configure a server (this changes the working directory)", action="store_true") + args = parser.parse_args() + formerdir = None + if args.zero_config: + import sys + import pathlib + #print(f"zero config {pathlib.Path(__file__).parent.parent.resolve()}") + formerdir = os.getcwd() + os.chdir(pathlib.Path(__file__).parent.parent.resolve()) -# Change the current working directory -os.chdir(project_directory) from wiktextract.page import parse_page from wiktextract_context import get_wiktextract_context @@ -24,8 +36,8 @@ class Wiktextract: page = self.wxr.wtp.get_page(title) if not page: return None - - + + result = parse_page(self.wxr, title, page.body) #print(json.dumps(result), "\n\n\n") @@ -99,20 +111,11 @@ class Wiktextract: if __name__ == "__main__": - import argparse - from argparse import RawTextHelpFormatter #pour le formattage de l'aide - parser = argparse.ArgumentParser(formatter_class=RawTextHelpFormatter, description="""Wiktextract wrapper - \033[1m\033[32mex :\033[0m - ‣\033[0m\033[32m./wiktextract_wrapper.py -l en -w fr -e yellow\033[0m""") - parser.add_argument("-l", "--word_language", help="language of the sought entry", type=str, default = "en") - parser.add_argument("-w", "--wikt_language", help="language of the wiktionary", type=str, default = "en") - parser.add_argument("-e", "--entry", help="the entry", type=str, default=None) - args = parser.parse_args() if args.entry != None: wkstrkt = Wiktextract(args.wikt_language, args.word_language) result = wkstrkt.parse_page(args.entry) print(json.dumps(result, ensure_ascii=False)) else: print("{'err':'You need to specify a word'}") - - + if formerdir != None: + os.chdir(formerdir) -- GitLab