omdev 0.0.0.dev57__py3-none-any.whl → 0.0.0.dev59__py3-none-any.whl

omdev/antlr/consts.py

@@ -0,0 +1,7 @@
+ANTLR_VERSION = '4.13.2'
+ANTLR_JAR_NAME = f'antlr-{ANTLR_VERSION}-complete.jar'
+ANTLR_JAR_URL = f'https://www.antlr.org/download/{ANTLR_JAR_NAME}'
+ANTLR_RUNTIME_PACKAGE = 'antlr4-python3-runtime'
+ANTLR_GITHUB_REPO = 'antlr/antlr4'
+
+ANTLR_RUNTIME_VENDOR = 'omlish.antlr._runtime'

omdev/antlr/gen.py

@@ -0,0 +1,110 @@
+"""
+TODO:
+ - fix relpath in omlish/
+ - sem-bounded parallelism
+"""
+import argparse
+import os.path
+import re
+import shutil
+import subprocess
+
+from omlish import check
+from omlish import lang
+
+from ..cache import data as dcache
+from .consts import ANTLR_JAR_URL
+from .consts import ANTLR_RUNTIME_VENDOR
+
+
+ANTLR_JAR_CACHE = dcache.UrlSpec(ANTLR_JAR_URL)
+
+
+class GenPy:
+    def __init__(
+            self,
+            dir: str,  # noqa
+            *,
+            out_subdir: str = '_antlr',
+            runtime_import: str = ANTLR_RUNTIME_VENDOR,
+    ) -> None:
+        super().__init__()
+        check.arg(not os.path.isabs(out_subdir) and '..' not in out_subdir)
+        self._dir = dir
+        self._out_subdir = out_subdir
+        self._runtime_import = runtime_import
+        self._out_dir = os.path.join(dir, out_subdir)
+
+    @lang.cached_function
+    def jar(self) -> str:
+        return dcache.default().get(ANTLR_JAR_CACHE)
+
+    def process_g4(self, g4_file: str) -> None:
+        subprocess.check_call([
+            'java',
+            '-jar', self.jar(),
+            '-Dlanguage=Python3',
+            '-visitor',
+            '-o', self._out_subdir,
+            g4_file,
+        ], cwd=self._dir)
+
+    def process_py(self, py_file: str) -> None:
+        ap = os.path.join(self._out_dir, py_file)
+        with open(ap) as f:
+            in_lines = list(f)
+
+        out_lines = [
+            '# type: ignore\n',
+            '# ruff: noqa\n',
+            '# flake8: noqa\n',
+        ]
+        for l in in_lines:
+            l = re.sub(r'^(from antlr4)(.*)', rf'from {self._runtime_import}\2', l)
+            out_lines.append(l)
+
+        with open(ap, 'w') as f:
+            f.write(''.join(out_lines))
+
+    def run(self) -> None:
+        if os.path.exists(self._out_dir):
+            shutil.rmtree(self._out_dir)
+        os.mkdir(self._out_dir)
+        with open(os.path.join(self._out_dir, '__init__.py'), 'w'):
+            pass
+
+        for f in os.listdir(self._dir):
+            fp = os.path.join(self._dir, f)
+            if not os.path.isfile(fp):
+                continue
+            if f.endswith('.g4'):
+                self.process_g4(f)
+
+        for f in list(os.listdir(self._out_dir)):
+            fp = os.path.join(self._out_dir, f)
+            if not os.path.isfile(fp):
+                continue
+            if f.split('.')[-1] in ('interp', 'tokens'):
+                os.unlink(fp)
+            elif f != '__init__.py' and f.endswith('.py'):
+                self.process_py(f)
+
+
+def _main() -> None:
+    parser = argparse.ArgumentParser()
+    parser.add_argument('roots', nargs='*')
+    args = parser.parse_args()
+
+    base_dir = os.getcwd()
+    if not os.path.isfile(os.path.join(base_dir, 'pyproject.toml')):
+        raise RuntimeError('Must run from project root')
+
+    for root_dir in args.roots:
+        print(f'Processing {root_dir}')
+        GenPy(
+            root_dir,
+        ).run()
+
+
+if __name__ == '__main__':
+    _main()

{omdev-0.0.0.dev57.dist-info → omdev-0.0.0.dev59.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: omdev
-Version: 0.0.0.dev57
+Version: 0.0.0.dev59
 Summary: omdev
 Author: wrmsr
 License: BSD-3-Clause
@@ -12,7 +12,7 @@ Classifier: Operating System :: OS Independent
 Classifier: Operating System :: POSIX
 Requires-Python: ~=3.12
 License-File: LICENSE
-Requires-Dist: omlish ==0.0.0.dev57
+Requires-Dist: omlish ==0.0.0.dev59
 Provides-Extra: all
 Requires-Dist: pycparser ~=2.22 ; extra == 'all'
 Requires-Dist: cffi ~=1.17 ; extra == 'all'

{omdev-0.0.0.dev57.dist-info → omdev-0.0.0.dev59.dist-info}/RECORD

@@ -14,6 +14,10 @@ omdev/wheelfile.py,sha256=yfupGcGkbFlmzGzKU64k_vmOKpaKnUlDWxeGn2KdekU,10005
 omdev/amalg/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 omdev/amalg/__main__.py,sha256=h94M-VqZ3AFBU2a8zOsjeKK7RF6uINhTHl6OiGbVMgw,163
 omdev/amalg/amalg.py,sha256=8tsrsKpgB54aOZ7CPQ6EoUIUYXylMvJpDOOJwbvSZ5E,14339
+omdev/antlr/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+omdev/antlr/consts.py,sha256=8pR6r0m0P3hAiyiAoJZ-nptd2GYbZ98mxwPL9cpaRuw,279
+omdev/antlr/gen.py,sha256=QnPyVWUrJYqHoOc3XsXA8fJdoiwAj6pyUwdG-DVrIeA,2953
+omdev/antlr/_antlr/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 omdev/cache/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 omdev/cache/compute/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 omdev/cache/compute/cache.py,sha256=pEbcTSQavhX7M0olzR8AOCtHx7tMnMnEDEfTPQVY5uE,3602
@@ -115,9 +119,9 @@ omdev/tools/piptools.py,sha256=-jR5q3w4sHqntxCLExFCBNIARB788FUsAbJ62PK2sBU,2774
 omdev/tools/proftools.py,sha256=xKSm_yPoCnfsvS3iT9MblDqFMuZmGfI3_koGj8amMyU,145
 omdev/tools/rst.py,sha256=6dWk8QZHoGiLSuBw3TKsXZjjFK6wWBEtPi9krdCLKKg,977
 omdev/tools/sqlrepl.py,sha256=tmFZh80-xsGM62dyQ7_UGLebChrj7IHbIPYBWDJMgVk,5741
-omdev-0.0.0.dev57.dist-info/LICENSE,sha256=B_hVtavaA8zCYDW99DYdcpDLKz1n3BBRjZrcbv8uG8c,1451
-omdev-0.0.0.dev57.dist-info/METADATA,sha256=Bc5g8JNkGTVplTCJJa3urQVje-2-pJgiAfOo1Z9-3mw,1252
-omdev-0.0.0.dev57.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
-omdev-0.0.0.dev57.dist-info/entry_points.txt,sha256=dHLXFmq5D9B8qUyhRtFqTGWGxlbx3t5ejedjrnXNYLU,33
-omdev-0.0.0.dev57.dist-info/top_level.txt,sha256=1nr7j30fEWgLYHW3lGR9pkdHkb7knv1U1ES1XRNVQ6k,6
-omdev-0.0.0.dev57.dist-info/RECORD,,
+omdev-0.0.0.dev59.dist-info/LICENSE,sha256=B_hVtavaA8zCYDW99DYdcpDLKz1n3BBRjZrcbv8uG8c,1451
+omdev-0.0.0.dev59.dist-info/METADATA,sha256=DwUCqPNW2EXhsYnskKg_oppQTYWF_XQnnJpzjA2eJzg,1252
+omdev-0.0.0.dev59.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
+omdev-0.0.0.dev59.dist-info/entry_points.txt,sha256=dHLXFmq5D9B8qUyhRtFqTGWGxlbx3t5ejedjrnXNYLU,33
+omdev-0.0.0.dev59.dist-info/top_level.txt,sha256=1nr7j30fEWgLYHW3lGR9pkdHkb7knv1U1ES1XRNVQ6k,6
+omdev-0.0.0.dev59.dist-info/RECORD,,