omdev 0.0.0.dev396__tar.gz → 0.0.0.dev397__tar.gz

{omdev-0.0.0.dev396/omdev.egg-info → omdev-0.0.0.dev397}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: omdev
-Version: 0.0.0.dev396
+Version: 0.0.0.dev397
 Summary: omdev
 Author: wrmsr
 License-Expression: BSD-3-Clause
@@ -14,7 +14,7 @@ Classifier: Programming Language :: Python :: 3.13
 Requires-Python: >=3.13
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: omlish==0.0.0.dev396
+Requires-Dist: omlish==0.0.0.dev397
 Provides-Extra: all
 Requires-Dist: black~=25.1; extra == "all"
 Requires-Dist: pycparser~=2.22; extra == "all"

{omdev-0.0.0.dev396 → omdev-0.0.0.dev397}/omdev/.manifests.json

@@ -515,7 +515,7 @@
     "module": ".tools.prof",
     "attr": "_CLI_MODULE",
     "file": "omdev/tools/prof.py",
-    "line": 57,
+    "line": 128,
     "value": {
       "$.cli.types.CliModule": {
         "cmd_name": "prof",

{omdev-0.0.0.dev396 → omdev-0.0.0.dev397}/omdev/dataclasses/codegen.py

@@ -7,7 +7,6 @@ TODO:
  - !! manifests for dataclass config?
   - more sparse / diffuse intent, not package-level
 """
-import json
 import logging
 import os.path
 import typing as ta
@@ -78,7 +77,7 @@ class DataclassCodeGen:
             for dp, _, fns in os.walk(root_dir):
                 if PACKAGE_CONFIG_FILE_NAME in fns:
                     with open(os.path.join(dp, PACKAGE_CONFIG_FILE_NAME)) as f:
-                        config = PackageConfig(**json.load(f))
+                        config = PackageConfig.loads(f.read())
                     pkg_parts = dp.split(os.sep)
                     trie[pkg_parts] = config
 

omdev-0.0.0.dev397/omdev/tools/prof.py

@@ -0,0 +1,133 @@
+"""
+TODO:
+ - a general purpose profile tool lol - or is this for / in bootstrap?
+"""
+import os.path
+import shutil
+import subprocess
+import sys
+import tempfile
+
+from omlish.argparse import all as ap
+
+from ..cli import CliModule
+
+
+##
+
+
+def pstats_to_pdf(
+        pstats_file: str,
+        pdf_file: str,
+) -> None:
+    dot = subprocess.check_output([
+        sys.executable,
+        '-m', 'gprof2dot',
+        '-f', 'pstats',
+        pstats_file,
+    ])
+
+    pdf = subprocess.check_output(
+        ['dot', '-Tpdf'],
+        input=dot,
+    )
+
+    with open(pdf_file, 'wb') as f:
+        f.write(pdf)
+
+
+class Cli(ap.Cli):
+    @ap.cmd(
+        ap.arg('pstats-file'),
+        ap.arg('pdf-file', nargs='?'),
+        ap.arg('-w', '--write', action='store_true'),
+        ap.arg('-o', '--open', action='store_true'),
+        ap.arg('-O', '--overwrite', action='store_true'),
+    )
+    def pstats_pdf(self) -> None:
+        pdf_file = self.args.pdf_file
+        if pdf_file is None and self.args.write:
+            pdf_file = self.args.pstats_file + '.pdf'
+
+        if pdf_file is not None:
+            if os.path.exists(pdf_file) and not self.args.overwrite:
+                raise OSError(f'File exists: {pdf_file}')
+        else:
+            pdf_file = tempfile.mktemp(suffix=os.path.basename(self.args.pstats_file) + '.pdf')  # noqa
+
+        pstats_to_pdf(
+            self.args.pstats_file,
+            self.args.pdf_file,
+        )
+
+        print(pdf_file)
+
+        if self.args.open:
+            subprocess.check_call(['open', pdf_file])
+
+    @ap.cmd(
+        ap.arg('src'),
+        ap.arg('out-file', nargs='?'),
+        ap.arg('-p', '--pdf', action='store_true'),
+        ap.arg('-o', '--open', action='store_true'),
+        ap.arg('-O', '--overwrite', action='store_true'),
+    )
+    def pstats_exec(self) -> None:
+        if self.args.out_file is not None:
+            if os.path.exists(self.args.out_file) and not self.args.overwrite:
+                raise OSError(f'File exists: {self.args.out_file}')
+
+        tmp_dir = tempfile.mkdtemp()
+
+        src_file = os.path.join(tmp_dir, 'pstats_exec.py')
+        with open(src_file, 'w') as f:
+            f.write(self.args.src)
+
+        pstats_file = os.path.join(tmp_dir, 'prof.pstats')
+
+        # TODO: --python - and handle env vars, unset venv and pythonpath stuff - helper for this, scrub env
+        #  - share with execstat, -x
+        subprocess.check_call([
+            sys.executable,
+            '-m', 'cProfile',
+            '-o', pstats_file,
+            os.path.abspath(src_file),
+        ])
+
+        if self.args.pdf:
+            out_file = os.path.join(tmp_dir, 'pstats.pdf')
+
+            pstats_to_pdf(
+                pstats_file,
+                out_file,
+            )
+
+        else:
+            out_file = pstats_file
+
+        if self.args.out_file is not None:
+            shutil.move(out_file, self.args.out_file)
+            out_file = self.args.out_file
+
+        print(out_file)
+
+        if self.args.open:
+            if self.args.pdf:
+                subprocess.check_call(['open', out_file])
+
+            else:
+                # Alt: python -i <setup.py> where setup.py is 'import pstats; stats = pstats.Stats(<out_file>)'
+                os.execl(
+                    sys.executable,
+                    sys.executable,
+                    '-m', 'pstats',
+                    out_file,
+                )
+
+
+# @omlish-manifest
+_CLI_MODULE = CliModule('prof', __name__)
+
+
+if __name__ == '__main__':
+    Cli()()

{omdev-0.0.0.dev396 → omdev-0.0.0.dev397/omdev.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: omdev
-Version: 0.0.0.dev396
+Version: 0.0.0.dev397
 Summary: omdev
 Author: wrmsr
 License-Expression: BSD-3-Clause
@@ -14,7 +14,7 @@ Classifier: Programming Language :: Python :: 3.13
 Requires-Python: >=3.13
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: omlish==0.0.0.dev396
+Requires-Dist: omlish==0.0.0.dev397
 Provides-Extra: all
 Requires-Dist: black~=25.1; extra == "all"
 Requires-Dist: pycparser~=2.22; extra == "all"

{omdev-0.0.0.dev396 → omdev-0.0.0.dev397}/omdev.egg-info/requires.txt

@@ -1,4 +1,4 @@
-omlish==0.0.0.dev396
+omlish==0.0.0.dev397
 
 [all]
 black~=25.1

{omdev-0.0.0.dev396 → omdev-0.0.0.dev397}/pyproject.toml

@@ -13,7 +13,7 @@ urls = {source = 'https://github.com/wrmsr/omlish'}
 license = 'BSD-3-Clause'
 readme = 'README.md'
 requires-python = '>=3.13'
-version = '0.0.0.dev396'
+version = '0.0.0.dev397'
 classifiers = [
     'Development Status :: 2 - Pre-Alpha',
     'Intended Audience :: Developers',
@@ -24,7 +24,7 @@ classifiers = [
 ]
 description = 'omdev'
 dependencies = [
-    'omlish == 0.0.0.dev396',
+    'omlish == 0.0.0.dev397',
 ]
 
 [project.optional-dependencies]
@@ -89,6 +89,7 @@ exclude = [
     '*.cu',
     '*.g4',
     '*.h',
+    '.dataclasses.json',
     '.manifests.json',
     'LICENSE',
     'LICENSE.txt',

omdev-0.0.0.dev396/omdev/tools/prof.py

@@ -1,62 +0,0 @@
-"""
-TODO:
- - a general purpose profile tool lol - or is this for / in bootstrap?
-"""
-import os.path
-import subprocess
-import sys
-import tempfile
-
-from omlish.argparse import all as ap
-
-from ..cli import CliModule
-
-
-##
-
-
-class Cli(ap.Cli):
-    @ap.cmd(
-        ap.arg('file'),
-        ap.arg('out-file', nargs='?'),
-        ap.arg('-w', '--write', action='store_true'),
-        ap.arg('-o', '--open', action='store_true'),
-        ap.arg('-O', '--overwrite', action='store_true'),
-    )
-    def pstats_pdf(self) -> None:
-        out_file = self.args.out_file
-        if out_file is None and self.args.write:
-            out_file = self.args.file + '.pdf'
-
-        if out_file is not None:
-            if os.path.exists(out_file) and not self.args.overwrite:
-                raise OSError(f'File exists: {out_file}')
-        else:
-            out_file = tempfile.mktemp(suffix=os.path.basename(self.args.file) + '.pdf')  # noqa
-
-        dot = subprocess.check_output([
-            sys.executable,
-            '-m', 'gprof2dot',
-            '-f', 'pstats',
-            self.args.file,
-        ])
-
-        pdf = subprocess.check_output(
-            ['dot', '-Tpdf'],
-            input=dot,
-        )
-
-        with open(out_file, 'wb') as f:
-            f.write(pdf)
-        print(out_file)
-
-        if self.args.open:
-            subprocess.check_call(['open', out_file])
-
-
-# @omlish-manifest
-_CLI_MODULE = CliModule('prof', __name__)
-
-
-if __name__ == '__main__':
-    Cli()()