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