omlish 0.0.0.dev112__tar.gz → 0.0.0.dev114__tar.gz

{omlish-0.0.0.dev112/omlish.egg-info → omlish-0.0.0.dev114}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: omlish
-Version: 0.0.0.dev112
+Version: 0.0.0.dev114
 Summary: omlish
 Author: wrmsr
 License: BSD-3-Clause

{omlish-0.0.0.dev112 → omlish-0.0.0.dev114}/omlish/__about__.py

@@ -1,5 +1,5 @@
-__version__ = '0.0.0.dev112'
-__revision__ = 'b12f9a554e6b2fb475bd820af89e0860c0b46215'
+__version__ = '0.0.0.dev114'
+__revision__ = '6b692f5ac328ea586f2988bf1bdb756c300d7765'
 
 
 #

{omlish-0.0.0.dev112 → omlish-0.0.0.dev114}/omlish/argparse.py

@@ -77,14 +77,23 @@ CommandFn = ta.Callable[[], int | None]
 class Command:
     name: str
     fn: CommandFn
-    args: ta.Sequence[Arg] = ()
+    args: ta.Sequence[Arg] = ()  # noqa
+
+    _: dc.KW_ONLY
+
+    aliases: ta.Sequence[str] | None = None
     parent: ta.Optional['Command'] = None
     accepts_unknown: bool = False
 
     def __post_init__(self) -> None:
-        check.isinstance(self.name, str)
-        check.not_in('_', self.name)
-        check.not_empty(self.name)
+        def check_name(s: str) -> None:
+            check.isinstance(s, str)
+            check.not_in('_', s)
+            check.not_empty(s)
+        check_name(self.name)
+        check.not_isinstance(self.aliases, str)
+        for a in self.aliases or []:
+            check_name(a)
 
         check.callable(self.fn)
         check.arg(all(isinstance(a, Arg) for a in self.args))
@@ -105,6 +114,7 @@ class Command:
 def command(
         *args: Arg,
         name: str | None = None,
+        aliases: ta.Iterable[str] | None = None,
         parent: Command | None = None,
         accepts_unknown: bool = False,
 ) -> ta.Any:  # ta.Callable[[CommandFn], Command]:  # FIXME
@@ -112,12 +122,14 @@ def command(
         check.isinstance(arg, Arg)
     check.isinstance(name, (str, None))
     check.isinstance(parent, (Command, None))
+    check.not_isinstance(aliases, str)
 
     def inner(fn):
         return Command(
             (name if name is not None else fn.__name__).replace('_', '-'),
             fn,
             args,
+            aliases=tuple(aliases) if aliases is not None else None,
             parent=parent,
             accepts_unknown=accepts_unknown,
         )
@@ -185,18 +197,23 @@ class _CliMeta(type):
             if isinstance(obj, Command):
                 if obj.parent is not None:
                     raise NotImplementedError
-                cparser = subparsers.add_parser(obj.name)
-                for arg in (obj.args or []):
-                    if (
-                            len(arg.args) == 1 and
-                            isinstance(arg.args[0], str) and
-                            not (n := check.isinstance(arg.args[0], str)).startswith('-') and
-                            'metavar' not in arg.kwargs
-                    ):
-                        cparser.add_argument(n.replace('-', '_'), **arg.kwargs, metavar=n)
-                    else:
-                        cparser.add_argument(*arg.args, **arg.kwargs)
-                cparser.set_defaults(_cmd=obj)
+                for cn in [obj.name, *(obj.aliases or [])]:
+                    cparser = subparsers.add_parser(cn)
+                    for arg in (obj.args or []):
+                        if (
+                                len(arg.args) == 1 and
+                                isinstance(arg.args[0], str) and
+                                not (n := check.isinstance(arg.args[0], str)).startswith('-') and
+                                'metavar' not in arg.kwargs
+                        ):
+                            cparser.add_argument(
+                                n.replace('-', '_'),
+                                **arg.kwargs,
+                                metavar=n,
+                            )
+                        else:
+                            cparser.add_argument(*arg.args, **arg.kwargs)
+                    cparser.set_defaults(_cmd=obj)
 
             elif isinstance(obj, Arg):
                 if att in anns:

omlish-0.0.0.dev114/omlish/diag/debug.py

@@ -0,0 +1,83 @@
+"""
+TODO:
+ - select best debugger: ipdb, pudb,
+"""
+import dataclasses as dc
+import functools
+import traceback
+import types
+import typing as ta
+
+from .. import check
+from .. import lang
+
+
+if ta.TYPE_CHECKING:
+    import pdb  # noqa
+
+    from . import pydevd
+
+else:
+    pdb = lang.proxy_import('pdb')
+
+    pydevd = lang.proxy_import('.pydevd', __package__)
+
+
+T = ta.TypeVar('T')
+P = ta.ParamSpec('P')
+
+ExceptionInfo: ta.TypeAlias = tuple[type[BaseException], BaseException, types.TracebackType]
+
+PostMortemDebugger: ta.TypeAlias = ta.Callable[[ExceptionInfo], None]
+
+
+##
+
+
+def pdb_post_mortem_debugger(ei: ExceptionInfo) -> None:
+    pdb.post_mortem(ei[2])
+
+
+def get_post_mortem_debugger() -> PostMortemDebugger:
+    if pydevd.is_present():
+        return pydevd.debug_unhandled_exception
+
+    else:
+        check.not_none(pdb.post_mortem)
+        return pdb_post_mortem_debugger
+
+
+##
+
+
+@dc.dataclass(kw_only=True)
+class DebuggingOnException:
+    silent: bool = False
+
+    fn: PostMortemDebugger | None = None
+
+    def __call__(self, fn: ta.Callable[P, T]) -> ta.Callable[P, T]:
+        @functools.wraps(fn)
+        def inner(*args, **kwargs):
+            with self:
+                return fn(*args, **kwargs)
+
+        return inner
+
+    def __enter__(self) -> ta.Self:
+        if self.fn is None:
+            self.fn = get_post_mortem_debugger()
+
+        return self
+
+    def __exit__(self, exc_type, exc_val, exc_tb):
+        if exc_val is None:
+            return
+
+        if not self.silent:
+            traceback.print_exc()
+
+        check.not_none(self.fn)((exc_type, exc_val, exc_tb))
+
+
+debugging_on_exception = DebuggingOnException

{omlish-0.0.0.dev112 → omlish-0.0.0.dev114}/omlish/diag/pydevd.py

@@ -5,7 +5,6 @@ an already-debugging PyCharm instance to debug PySpark jobs.
 
 TODO:
  - https://www.jetbrains.com/help/pycharm/remote-debugging-with-product.html#
- - PyCharm.app/Contents/plugins/python-ce/helpers/pydev/_pydevd_bundle/pydevd_constants.py -> USE_LOW_IMPACT_MONITORING
 
 ==
 

{omlish-0.0.0.dev112 → omlish-0.0.0.dev114/omlish.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: omlish
-Version: 0.0.0.dev112
+Version: 0.0.0.dev114
 Summary: omlish
 Author: wrmsr
 License: BSD-3-Clause

{omlish-0.0.0.dev112 → omlish-0.0.0.dev114}/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.dev112'
+version = '0.0.0.dev114'
 classifiers = [
     'License :: OSI Approved :: BSD License',
     'Development Status :: 2 - Pre-Alpha',

omlish-0.0.0.dev112/omlish/diag/debug.py

@@ -1,26 +0,0 @@
-"""
-TODO:
- - select best debugger: ipdb, pudb,
-"""
-import contextlib
-import sys
-import traceback
-import typing as ta
-
-
-@contextlib.contextmanager
-def debugging_on_exception(
-        *,
-        silent: bool = False,
-) -> ta.Iterator[None]:
-    import pdb  # noqa
-
-    try:
-        yield
-    except Exception:
-        if not silent:
-            traceback.print_exc()
-
-        pdb.post_mortem(sys.exc_info()[2])
-
-        raise