omdev 0.0.0.dev68__tar.gz → 0.0.0.dev70__tar.gz

{omdev-0.0.0.dev68/omdev.egg-info → omdev-0.0.0.dev70}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: omdev
-Version: 0.0.0.dev68
+Version: 0.0.0.dev70
 Summary: omdev
 Author: wrmsr
 License: BSD-3-Clause
@@ -12,15 +12,19 @@ Classifier: Operating System :: OS Independent
 Classifier: Operating System :: POSIX
 Requires-Python: ~=3.12
 License-File: LICENSE
-Requires-Dist: omlish==0.0.0.dev68
+Requires-Dist: omlish==0.0.0.dev70
 Provides-Extra: all
+Requires-Dist: black~=24.10; extra == "all"
 Requires-Dist: pycparser~=2.22; extra == "all"
 Requires-Dist: cffi~=1.17; extra == "all"
 Requires-Dist: pcpp~=1.30; extra == "all"
 Requires-Dist: docutils~=0.21; extra == "all"
 Requires-Dist: mypy~=1.11; extra == "all"
+Requires-Dist: gprof2dot~=2024.6; extra == "all"
 Requires-Dist: tokenize_rt~=6.0; extra == "all"
 Requires-Dist: wheel~=0.44; extra == "all"
+Provides-Extra: black
+Requires-Dist: black~=24.10; extra == "black"
 Provides-Extra: c
 Requires-Dist: pycparser~=2.22; extra == "c"
 Requires-Dist: cffi~=1.17; extra == "c"
@@ -29,6 +33,8 @@ Provides-Extra: docutils
 Requires-Dist: docutils~=0.21; extra == "docutils"
 Provides-Extra: mypy
 Requires-Dist: mypy~=1.11; extra == "mypy"
+Provides-Extra: prof
+Requires-Dist: gprof2dot~=2024.6; extra == "prof"
 Provides-Extra: tokens
 Requires-Dist: tokenize_rt~=6.0; extra == "tokens"
 Provides-Extra: wheel

{omdev-0.0.0.dev68 → omdev-0.0.0.dev70}/omdev/.manifests.json

@@ -171,7 +171,7 @@
     "module": ".tools.nbtools",
     "attr": "_CLI_MODULE",
     "file": "omdev/tools/nbtools.py",
-    "line": 44,
+    "line": 115,
     "value": {
       "$.cli.types.CliModule": {
         "cmd_name": [
@@ -194,6 +194,30 @@
       }
     }
   },
+  {
+    "module": ".tools.proftools",
+    "attr": "_CLI_MODULE",
+    "file": "omdev/tools/proftools.py",
+    "line": 50,
+    "value": {
+      "$.cli.types.CliModule": {
+        "cmd_name": "prof",
+        "mod_name": "omdev.tools.proftools"
+      }
+    }
+  },
+  {
+    "module": ".tools.rst",
+    "attr": "_CLI_MODULE",
+    "file": "omdev/tools/rst.py",
+    "line": 45,
+    "value": {
+      "$.cli.types.CliModule": {
+        "cmd_name": "rst",
+        "mod_name": "omdev.tools.rst"
+      }
+    }
+  },
   {
     "module": ".tools.sqlrepl",
     "attr": "_CLI_MODULE",

{omdev-0.0.0.dev68 → omdev-0.0.0.dev70}/omdev/__about__.py

@@ -12,6 +12,10 @@ class Project(ProjectBase):
     ]
 
     optional_dependencies = {
+        'black': [
+            'black ~= 24.10',
+        ],
+
         'c': [
             'pycparser ~= 2.22',
             'cffi ~= 1.17',
@@ -26,6 +30,10 @@ class Project(ProjectBase):
             'mypy ~= 1.11',
         ],
 
+        'prof': [
+            'gprof2dot ~= 2024.6',
+        ],
+
         'tokens': [
             'tokenize_rt ~= 6.0',
         ],

{omdev-0.0.0.dev68 → omdev-0.0.0.dev70}/omdev/cli/managers.py

@@ -50,5 +50,5 @@ def setup_install_manager(cli_pkg: str) -> None:
     if detect_install_manager(cli_pkg) is None:
         return
 
-    from .pathhack import _install_pth_file
+    from ._pathhack import _install_pth_file
     _install_pth_file()

omdev-0.0.0.dev70/omdev/tools/nbtools.py

@@ -0,0 +1,120 @@
+import io
+import json
+import os.path
+import subprocess
+import sys
+import urllib.parse
+import urllib.request
+
+from omlish import argparse as ap
+from omlish import check
+
+from ..cli import CliModule
+
+
+class Cli(ap.Cli):
+    @ap.command(
+        ap.arg('file'),
+        ap.arg('-w', '--write', action='store_true'),
+        ap.arg('-o', '--overwrite', action='store_true'),
+        ap.arg('--no-header', action='store_true'),
+        ap.arg('--keep-bangs', action='store_true'),
+        ap.arg('--black', action='store_true'),
+    )
+    def strip_code(self) -> None:
+        out_file: str | None = None
+        out_header: list[str] = []
+
+        if ':' in self.args.file:
+            url = self.args.file
+            pu = urllib.parse.urlparse(url)
+            check.equal(pu.scheme, 'https')
+
+            out_header.append(f'# Origin: {url}')
+
+            if pu.hostname == 'github.com':
+                _, user, repo, blob, *rest = pu.path.split('/')
+                check.equal(blob, 'blob')
+                url = f'https://raw.githubusercontent.com/{user}/{repo}/refs/heads/{"/".join(rest)}'
+
+            elif pu.hostname == 'raw.githubusercontent.com':
+                pass
+
+            else:
+                raise ValueError(pu.hostname)
+
+            file = pu.path.split('/')[-1]
+            check.state(file.endswith('.ipynb'))
+
+            if self.args.write:
+                out_file = file.rpartition('.')[0] + '.py'
+
+            with urllib.request.urlopen(url) as resp:  # noqa
+                buf = resp.read()
+
+            dct = json.loads(buf.decode('utf-8'))
+
+        elif self.args.file == '-':
+            if self.args.write:
+                raise Exception('Write not supported with stdin')
+
+            dct = json.load(sys.stdin)
+
+        else:
+            with open(self.args.file) as f:
+                dct = json.load(f)
+
+            if self.args.write:
+                out_file = self.args.file.rpartition('.')[0] + '.py'
+
+        if out_file is not None and os.path.isfile(out_file) and not self.args.overwrite:
+            raise OSError(f'File exists: {out_file}')
+
+        delim = '\n##\n\n'
+
+        out = io.StringIO()
+        if not self.args.no_header and out_header:
+            out.write('\n'.join(out_header))
+            out.write(delim)
+
+        code_cells = [c for c in dct['cells'] if c['cell_type'] == 'code']
+        for i, c in enumerate(code_cells):
+            if i:
+                out.write(delim)
+
+            src = ''.join(c['source'])
+            if not self.args.keep_bangs:
+                src = ''.join(
+                    ('# ' + l) if l.startswith('!') else l
+                    for l in src.splitlines(keepends=True)
+                )
+
+            out.write(src)
+            out.write('\n')
+
+        out_src = out.getvalue()
+
+        if self.args.black:
+            proc = subprocess.run(
+                [sys.executable, '-m', 'black', '-'],
+                input=out_src.encode('utf-8'),
+                stdout=subprocess.PIPE,
+                check=True,
+            )
+            out_src = proc.stdout.decode('utf-8')
+
+        if out_file is not None:
+            with open(out_file, 'w') as f:
+                f.write(out_src)
+            print(f'Wrote {out_file}')
+
+        else:
+            sys.stdout.write(out_src)
+
+
+# @omlish-manifest
+_CLI_MODULE = CliModule(['notebook', 'nb'], __name__)
+
+
+if __name__ == '__main__':
+    Cli()()

omdev-0.0.0.dev70/omdev/tools/proftools.py

@@ -0,0 +1,55 @@
+import os.path
+import subprocess
+import sys
+import tempfile
+
+from omlish import argparse as ap
+
+from ..cli import CliModule
+
+
+class Cli(ap.Cli):
+    @ap.command(
+        ap.arg('file'),
+        ap.arg('out-file', nargs='?'),
+        ap.arg('-w', '--write', action='store_true'),
+        ap.arg('-o', '--overwrite', action='store_true'),
+        ap.arg('-O', '--open', 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()()

{omdev-0.0.0.dev68 → omdev-0.0.0.dev70}/omdev/tools/rst.py

@@ -5,6 +5,8 @@ import sys
 
 import docutils.core
 
+from ..cli import CliModule
+
 
 def rst2html(rst, report_level=None):
     kwargs = {
@@ -40,5 +42,9 @@ def _main() -> None:
     print(html)
 
 
+# @omlish-manifest
+_CLI_MODULE = CliModule('rst', __name__)
+
+
 if __name__ == '__main__':
     _main()

{omdev-0.0.0.dev68 → omdev-0.0.0.dev70/omdev.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: omdev
-Version: 0.0.0.dev68
+Version: 0.0.0.dev70
 Summary: omdev
 Author: wrmsr
 License: BSD-3-Clause
@@ -12,15 +12,19 @@ Classifier: Operating System :: OS Independent
 Classifier: Operating System :: POSIX
 Requires-Python: ~=3.12
 License-File: LICENSE
-Requires-Dist: omlish==0.0.0.dev68
+Requires-Dist: omlish==0.0.0.dev70
 Provides-Extra: all
+Requires-Dist: black~=24.10; extra == "all"
 Requires-Dist: pycparser~=2.22; extra == "all"
 Requires-Dist: cffi~=1.17; extra == "all"
 Requires-Dist: pcpp~=1.30; extra == "all"
 Requires-Dist: docutils~=0.21; extra == "all"
 Requires-Dist: mypy~=1.11; extra == "all"
+Requires-Dist: gprof2dot~=2024.6; extra == "all"
 Requires-Dist: tokenize_rt~=6.0; extra == "all"
 Requires-Dist: wheel~=0.44; extra == "all"
+Provides-Extra: black
+Requires-Dist: black~=24.10; extra == "black"
 Provides-Extra: c
 Requires-Dist: pycparser~=2.22; extra == "c"
 Requires-Dist: cffi~=1.17; extra == "c"
@@ -29,6 +33,8 @@ Provides-Extra: docutils
 Requires-Dist: docutils~=0.21; extra == "docutils"
 Provides-Extra: mypy
 Requires-Dist: mypy~=1.11; extra == "mypy"
+Provides-Extra: prof
+Requires-Dist: gprof2dot~=2024.6; extra == "prof"
 Provides-Extra: tokens
 Requires-Dist: tokenize_rt~=6.0; extra == "tokens"
 Provides-Extra: wheel

{omdev-0.0.0.dev68 → omdev-0.0.0.dev70}/omdev.egg-info/SOURCES.txt

@@ -69,11 +69,11 @@ omdev/cexts/_distutils/compilers/options.py
 omdev/cexts/_distutils/compilers/unixccompiler.py
 omdev/cli/__init__.py
 omdev/cli/__main__.py
+omdev/cli/_pathhack.py
 omdev/cli/clicli.py
 omdev/cli/install.py
 omdev/cli/main.py
 omdev/cli/managers.py
-omdev/cli/pathhack.py
 omdev/cli/types.py
 omdev/interp/__init__.py
 omdev/interp/__main__.py

{omdev-0.0.0.dev68 → omdev-0.0.0.dev70}/omdev.egg-info/requires.txt

@@ -1,14 +1,19 @@
-omlish==0.0.0.dev68
+omlish==0.0.0.dev70
 
 [all]
+black~=24.10
 pycparser~=2.22
 cffi~=1.17
 pcpp~=1.30
 docutils~=0.21
 mypy~=1.11
+gprof2dot~=2024.6
 tokenize_rt~=6.0
 wheel~=0.44
 
+[black]
+black~=24.10
+
 [c]
 pycparser~=2.22
 cffi~=1.17
@@ -20,6 +25,9 @@ docutils~=0.21
 [mypy]
 mypy~=1.11
 
+[prof]
+gprof2dot~=2024.6
+
 [tokens]
 tokenize_rt~=6.0
 

{omdev-0.0.0.dev68 → omdev-0.0.0.dev70}/pyproject.toml

@@ -12,7 +12,7 @@ authors = [
 urls = {source = 'https://github.com/wrmsr/omlish'}
 license = {text = 'BSD-3-Clause'}
 requires-python = '~=3.12'
-version = '0.0.0.dev68'
+version = '0.0.0.dev70'
 classifiers = [
     'License :: OSI Approved :: BSD License',
     'Development Status :: 2 - Pre-Alpha',
@@ -22,19 +22,24 @@ classifiers = [
 ]
 description = 'omdev'
 dependencies = [
-    'omlish == 0.0.0.dev68',
+    'omlish == 0.0.0.dev70',
 ]
 
 [project.optional-dependencies]
 all = [
+    'black ~= 24.10',
     'pycparser ~= 2.22',
     'cffi ~= 1.17',
     'pcpp ~= 1.30',
     'docutils ~= 0.21',
     'mypy ~= 1.11',
+    'gprof2dot ~= 2024.6',
     'tokenize_rt ~= 6.0',
     'wheel ~= 0.44',
 ]
+black = [
+    'black ~= 24.10',
+]
 c = [
     'pycparser ~= 2.22',
     'cffi ~= 1.17',
@@ -46,6 +51,9 @@ docutils = [
 mypy = [
     'mypy ~= 1.11',
 ]
+prof = [
+    'gprof2dot ~= 2024.6',
+]
 tokens = [
     'tokenize_rt ~= 6.0',
 ]

omdev-0.0.0.dev68/omdev/tools/nbtools.py

@@ -1,49 +0,0 @@
-import io
-import json
-import os.path
-import sys
-
-from omlish import argparse as ap
-
-from ..cli import CliModule
-
-
-class Cli(ap.Cli):
-    @ap.command(
-        ap.arg('file'),
-        ap.arg('-w', '--write', action='store_true'),
-        ap.arg('-o', '--overwrite', action='store_true'),
-    )
-    def code(self) -> None:
-        if self.args.write:
-            out_file = self.args.file.rpartition('.')[0] + '.py'
-            if os.path.isfile(out_file) and not self.args.overwrite:
-                raise OSError(f'File exists: {out_file}')
-        else:
-            out_file = None
-
-        with open(self.args.file) as f:
-            dct = json.load(f)
-
-        out = io.StringIO()
-        code_cells = [c for c in dct['cells'] if c['cell_type'] == 'code']
-        for i, c in enumerate(code_cells):
-            if i:
-                out.write('\n##\n\n')
-            out.write(''.join(c['source']))
-            out.write('\n')
-
-        if out_file is not None:
-            with open(out_file, 'w') as f:
-                f.write(out.getvalue())
-            print(f'Wrote {out_file}')
-        else:
-            sys.stdout.write(out.getvalue())
-
-
-# @omlish-manifest
-_CLI_MODULE = CliModule(['notebook', 'nb'], __name__)
-
-
-if __name__ == '__main__':
-    Cli()()

omdev-0.0.0.dev68/omdev/tools/proftools.py

@@ -1,4 +0,0 @@
-"""
-TODO:
- - rm prof.pstats.pdf ; ./python -m gprof2dot -f pstats prof.pstats | dot -Tpdf -o prof.pstats.pdf && open prof.pstats.pdf
-"""  # noqa