From 950803839c7e0037b5359c8633281364130dba39 Mon Sep 17 00:00:00 2001
From: Mathieu Loiseau <mathieu.loiseau@liris.cnrs.fr>
Date: Sat, 22 Jun 2024 13:51:00 +0200
Subject: [PATCH] structure wikstraktor

---
 src/wiktextract_wrapper.py | 17 +++++++++--------
 1 file changed, 9 insertions(+), 8 deletions(-)

diff --git a/src/wiktextract_wrapper.py b/src/wiktextract_wrapper.py
index 2d6bc48..47e6841 100755
--- a/src/wiktextract_wrapper.py
+++ b/src/wiktextract_wrapper.py
@@ -14,17 +14,18 @@ if __name__ == "__main__":
 
     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())
+
+if ('args' in dir() and args.zero_config) or 'zero_config' in dir():
+    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())
 
 import time
 t = time.time()
 from wiktextract.page import parse_page
-if args.show_timings:
+if 'args' in dir() and args.show_timings:
     print(time.time()-t)
 from wiktextract_context import get_wiktextract_context
 import json
@@ -69,7 +70,7 @@ class Wiktextract:
             transformed_pos = {
                 "sources": [
                     {
-                        "wiktionary_language": pos.get("lang_code", ""),  # Assuming this key is present in your data
+                        "wiktionary_language": self.wiktlang,  # Assuming this key is present in your data
                         "wiktextract_version": metadata.version("wiktextract"),
                         "wrapper_version": git.Repo(search_parent_directories=True).head.object.hexsha
                     }
-- 
GitLab