Make scripts run without install (#239)

google · Aug 15, 2023 · f80bfb6 · f80bfb6
1 parent ce2eda8
commit f80bfb6
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 4 deletions.
diff --git a/scripts/encode_data.py b/scripts/encode_data.py
@@ -17,10 +17,15 @@
 import functools
 import itertools
 import multiprocessing
+import os
 import sys
 import typing
 
-from budoux import utils
+# module hack
+LIB_PATH = os.path.join(os.path.dirname(__file__), '..')
+sys.path.insert(0, os.path.abspath(LIB_PATH))
+
+from budoux import utils  # noqa (module hack)
 
 ArgList = typing.Optional[typing.List[str]]
 DEFAULT_OUTPUT_FILENAME = 'encoded_data.txt'

diff --git a/scripts/prepare_knbc.py b/scripts/prepare_knbc.py
@@ -26,10 +26,15 @@
 
 import argparse
 import os
+import sys
 import typing
 from html.parser import HTMLParser
 
-from budoux import utils
+# module hack
+LIB_PATH = os.path.join(os.path.dirname(__file__), '..')
+sys.path.insert(0, os.path.abspath(LIB_PATH))
+
+from budoux import utils  # noqa (module hack)
 
 
 class KNBCHTMLParser(HTMLParser):

diff --git a/scripts/tests/test_encode_data.py b/scripts/tests/test_encode_data.py
@@ -18,12 +18,11 @@
 import typing
 import unittest
 
-from budoux import utils
-
 # module hack
 LIB_PATH = os.path.join(os.path.dirname(__file__), '..', '..')
 sys.path.insert(0, os.path.abspath(LIB_PATH))
 
+from budoux import utils  # noqa (module hack)
 from scripts import encode_data  # noqa (module hack)