omdev 0.0.0.dev90__py3-none-any.whl → 0.0.0.dev91__py3-none-any.whl

omdev/__about__.py

@@ -34,8 +34,12 @@ class Project(ProjectBase):
             'gprof2dot ~= 2024.6',
         ],
 
+        'prompttoolkit': [
+            'prompt-toolkit ~= 3.0',
+        ],
+
         'tokens': [
-            'tokenize_rt ~= 6.1',
+            'tokenize-rt ~= 6.1',
         ],
 
         'wheel': [

omdev/prompttoolkit/__init__.py

@@ -0,0 +1,11 @@
+# ruff: noqa: F401
+# flake8: noqa: F401
+
+from prompt_toolkit import Application
+from prompt_toolkit.key_binding import KeyBindings
+from prompt_toolkit.key_binding import KeyPressEvent
+from prompt_toolkit.layout import Layout
+from prompt_toolkit.layout.containers import Window
+from prompt_toolkit.styles import Style
+from prompt_toolkit.widgets import Frame
+from prompt_toolkit.widgets import TextArea

omdev/prompttoolkit/apps/ncdu.py

@@ -0,0 +1,140 @@
+import dataclasses as dc
+import os
+import typing as ta
+
+from omlish.lite.strings import format_num_bytes
+
+from ... import prompttoolkit as ptk
+
+
+def get_directory_size(path: str) -> int:
+    total_size = 0
+    for dirpath, _, filenames in os.walk(path):
+        for f in filenames:
+            fp = os.path.join(dirpath, f)
+            try:
+                total_size += os.path.getsize(fp)
+            except OSError:
+                pass
+    return total_size
+
+
+@dc.dataclass(frozen=True)
+class Entry:
+    name: str
+    size: int
+    type: ta.Literal['dir', 'file']
+
+
+def scan_directory(path: str) -> list[Entry]:
+    entries: list[Entry] = []
+
+    with os.scandir(path) as it:
+        for entry in it:
+            if entry.is_dir(follow_symlinks=False):
+                size = get_directory_size(entry.path)
+                entries.append(Entry(entry.name + '/', size, 'dir'))
+
+            elif entry.is_file(follow_symlinks=False):
+                size = entry.stat().st_size
+                entries.append(Entry(entry.name, size, 'file'))
+
+    entries.sort(key=lambda x: x.size, reverse=True)
+    return entries
+
+
+class NcduApp:
+    def __init__(self, root_path: str) -> None:
+        super().__init__()
+
+        self._root_path = root_path
+        self._current_path = root_path
+        self._entries = scan_directory(root_path)
+        self._cursor = 0
+
+        self._text_area = ptk.TextArea(focusable=True)
+        self.update_display()
+
+        self._kb = ptk.KeyBindings()
+        self._kb.add('q')(self.exit_app)
+        for k in ['up', 'p']:
+            self._kb.add(k)(self.move_up)
+        for k in ['down', 'n']:
+            self._kb.add(k)(self.move_down)
+        self._kb.add('enter')(self.enter_directory)
+        self._kb.add('backspace')(self.go_back)
+
+        self._layout = ptk.Layout(ptk.Frame(self._text_area))
+
+        self._style = ptk.Style.from_dict({
+            'frame': 'bg:#008800 bold',
+            'text-area': 'bg:#000000 fg:#ffffff',
+        })
+
+        self._app: ptk.Application = ptk.Application(
+            layout=self._layout,
+            key_bindings=self._kb,
+            style=self._style,
+            full_screen=True,
+        )
+
+    #
+
+    def update_display(self) -> None:
+        display_text = f'Current Directory: {self._current_path}\n\n'
+        for i, e in enumerate(self._entries):
+            indicator = '>' if i == self._cursor else ' '
+            display_text += f'{indicator} {e.name:<40} {format_num_bytes(e.size):>10}\n'
+        self._text_area.text = display_text
+
+    #
+
+    def move_up(self, event: ptk.KeyPressEvent) -> None:
+        if self._cursor > 0:
+            self._cursor -= 1
+            self.update_display()
+
+    def move_down(self, event: ptk.KeyPressEvent) -> None:
+        if self._cursor < len(self._entries) - 1:
+            self._cursor += 1
+            self.update_display()
+
+    def enter_directory(self, event: ptk.KeyPressEvent) -> None:
+        selected_entry = self._entries[self._cursor]
+        if selected_entry.type == 'dir':
+            self._current_path = os.path.join(self._current_path, selected_entry.name[:-1])
+            self._entries = scan_directory(self._current_path)
+            self._cursor = 0
+            self.update_display()
+
+    def go_back(self, event: ptk.KeyPressEvent) -> None:
+        if self._current_path != self._root_path:
+            self._current_path = os.path.dirname(self._current_path)
+            self._entries = scan_directory(self._current_path)
+            self._cursor = 0
+            self.update_display()
+
+    def exit_app(self, event: ptk.KeyPressEvent) -> None:
+        event.app.exit()
+
+    #
+
+    def run(self) -> None:
+        self._app.run()
+
+
+def _main() -> None:
+    import argparse
+
+    parser = argparse.ArgumentParser()
+
+    parser.add_argument('dir', default='.', nargs='?')
+
+    args = parser.parse_args()
+
+    ncdu_app = NcduApp(args.dir)
+    ncdu_app.run()
+
+
+if __name__ == '__main__':
+    _main()

omdev/scripts/interp.py

@@ -626,6 +626,9 @@ def deep_subclasses(cls: ta.Type[T]) -> ta.Iterator[ta.Type[T]]:
 # ../../../omlish/lite/strings.py
 
 
+##
+
+
 def camel_case(name: str, lower: bool = False) -> str:
     if not name:
         return ''
@@ -640,6 +643,9 @@ def snake_case(name: str) -> str:
     return '_'.join([name[l:r].lower() for l, r in zip([None, *uppers], [*uppers, None])]).strip('_')
 
 
+##
+
+
 def is_dunder(name: str) -> bool:
     return (
         name[:2] == name[-2:] == '__' and
@@ -658,10 +664,31 @@ def is_sunder(name: str) -> bool:
     )
 
 
+##
+
+
 def attr_repr(obj: ta.Any, *attrs: str) -> str:
     return f'{type(obj).__name__}({", ".join(f"{attr}={getattr(obj, attr)!r}" for attr in attrs)})'
 
 
+##
+
+
+FORMAT_NUM_BYTES_SUFFIXES: ta.Sequence[str] = ['B', 'kB', 'MB', 'GB', 'TB', 'PB', 'EB']
+
+
+def format_num_bytes(num_bytes: int) -> str:
+    for i, suffix in enumerate(FORMAT_NUM_BYTES_SUFFIXES):
+        value = num_bytes / 1024 ** i
+        if num_bytes < 1024 ** (i + 1):
+            if value.is_integer():
+                return f'{int(value)}{suffix}'
+            else:
+                return f'{value:.2f}{suffix}'
+
+    return f'{num_bytes / 1024 ** (len(FORMAT_NUM_BYTES_SUFFIXES) - 1):.2f}{FORMAT_NUM_BYTES_SUFFIXES[-1]}'
+
+
 ########################################
 # ../../packaging/specifiers.py
 # Copyright (c) Donald Stufft and individual contributors.

omdev/scripts/pyproject.py

@@ -2014,6 +2014,9 @@ def deep_subclasses(cls: ta.Type[T]) -> ta.Iterator[ta.Type[T]]:
 # ../../../omlish/lite/strings.py
 
 
+##
+
+
 def camel_case(name: str, lower: bool = False) -> str:
     if not name:
         return ''
@@ -2028,6 +2031,9 @@ def snake_case(name: str) -> str:
     return '_'.join([name[l:r].lower() for l, r in zip([None, *uppers], [*uppers, None])]).strip('_')
 
 
+##
+
+
 def is_dunder(name: str) -> bool:
     return (
         name[:2] == name[-2:] == '__' and
@@ -2046,10 +2052,31 @@ def is_sunder(name: str) -> bool:
     )
 
 
+##
+
+
 def attr_repr(obj: ta.Any, *attrs: str) -> str:
     return f'{type(obj).__name__}({", ".join(f"{attr}={getattr(obj, attr)!r}" for attr in attrs)})'
 
 
+##
+
+
+FORMAT_NUM_BYTES_SUFFIXES: ta.Sequence[str] = ['B', 'kB', 'MB', 'GB', 'TB', 'PB', 'EB']
+
+
+def format_num_bytes(num_bytes: int) -> str:
+    for i, suffix in enumerate(FORMAT_NUM_BYTES_SUFFIXES):
+        value = num_bytes / 1024 ** i
+        if num_bytes < 1024 ** (i + 1):
+            if value.is_integer():
+                return f'{int(value)}{suffix}'
+            else:
+                return f'{value:.2f}{suffix}'
+
+    return f'{num_bytes / 1024 ** (len(FORMAT_NUM_BYTES_SUFFIXES) - 1):.2f}{FORMAT_NUM_BYTES_SUFFIXES[-1]}'
+
+
 ########################################
 # ../../packaging/specifiers.py
 # Copyright (c) Donald Stufft and individual contributors.

{omdev-0.0.0.dev90.dist-info → omdev-0.0.0.dev91.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: omdev
-Version: 0.0.0.dev90
+Version: 0.0.0.dev91
 Summary: omdev
 Author: wrmsr
 License: BSD-3-Clause
@@ -10,9 +10,9 @@ Classifier: Development Status :: 2 - Pre-Alpha
 Classifier: Intended Audience :: Developers
 Classifier: Operating System :: OS Independent
 Classifier: Operating System :: POSIX
-Requires-Python: ~=3.12
+Requires-Python: >=3.12
 License-File: LICENSE
-Requires-Dist: omlish ==0.0.0.dev90
+Requires-Dist: omlish ==0.0.0.dev91
 Provides-Extra: all
 Requires-Dist: black ~=24.10 ; extra == 'all'
 Requires-Dist: pycparser ~=2.22 ; extra == 'all'
@@ -21,6 +21,7 @@ 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: prompt-toolkit ~=3.0 ; extra == 'all'
 Requires-Dist: tokenize-rt ~=6.1 ; extra == 'all'
 Requires-Dist: wheel ~=0.44 ; extra == 'all'
 Provides-Extra: black
@@ -35,6 +36,8 @@ Provides-Extra: mypy
 Requires-Dist: mypy ~=1.11 ; extra == 'mypy'
 Provides-Extra: prof
 Requires-Dist: gprof2dot ~=2024.6 ; extra == 'prof'
+Provides-Extra: prompttoolkit
+Requires-Dist: prompt-toolkit ~=3.0 ; extra == 'prompttoolkit'
 Provides-Extra: tokens
 Requires-Dist: tokenize-rt ~=6.1 ; extra == 'tokens'
 Provides-Extra: wheel

{omdev-0.0.0.dev90.dist-info → omdev-0.0.0.dev91.dist-info}/RECORD

@@ -1,5 +1,5 @@
 omdev/.manifests.json,sha256=juVlULcmZOz0t66Qf-GhpG_-4NGQU0XDmd7EjaRygxk,6257
-omdev/__about__.py,sha256=do_MjpnIpB_DX4GL4VV9gFQzs6bkoyDIYu29yB3rYxw,1131
+omdev/__about__.py,sha256=5MidJkdjpvZISIof4d7vMN7HhC7h5aUfl-FL0IpbY0A,1207
 omdev/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 omdev/bracepy.py,sha256=I8EdqtDvxzAi3I8TuMEW-RBfwXfqKbwp06CfOdj3L1o,2743
 omdev/classdot.py,sha256=YOvgy6x295I_8NKBbBlRVd3AN7Osirm_Lqt4Wj0j9rY,1631
@@ -98,6 +98,9 @@ omdev/precheck/lite.py,sha256=d8D-BY-Z0rqQigQ_aR9QXxdB3EBCHJc7WgBJngGiEiE,3871
 omdev/precheck/main.py,sha256=2YizDZSDhMeQaZKVH83s99pBmtOUL9B-8dXwIcw6fQQ,2674
 omdev/precheck/manifests.py,sha256=YfXqt6u0hlFXY0QkBMec6V_6Y9T4eCVAmrJDkQkB13U,774
 omdev/precheck/scripts.py,sha256=qq6MXkxgrYngPg5pWnXH4uRSuRkP3mFqbeml1UmvGBc,1265
+omdev/prompttoolkit/__init__.py,sha256=C4pFxNoIVo7cOH_jZ7-ruaYYd6DxEvuJr5ZGDrMFwXo,402
+omdev/prompttoolkit/apps/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+omdev/prompttoolkit/apps/ncdu.py,sha256=V2jZ-h3HN23KC5FB5gFjeMVZZbj-2j6zxyMca3O8OM0,3950
 omdev/pyproject/__init__.py,sha256=Y3l4WY4JRi2uLG6kgbGp93fuGfkxkKwZDvhsa0Rwgtk,15
 omdev/pyproject/__main__.py,sha256=gn3Rl1aYPYdiTtEqa9ifi0t-e4ZwPY0vhJ4UXvYdJDY,165
 omdev/pyproject/cexts.py,sha256=x13piOOnNrYbA17qZLDVuR0p1sqhgEwpk4FtImX-klM,4281
@@ -110,8 +113,8 @@ omdev/scripts/bumpversion.py,sha256=Kn7fo73Hs8uJh3Hi3EIyLOlzLPWAC6dwuD_lZ3cIzuY,
 omdev/scripts/execrss.py,sha256=hJTWBig_-XqKIj1DphYTfeVDtdQTM1XUl-Hvzs2x9iY,451
 omdev/scripts/exectime.py,sha256=dBdn3KV2jR6tCrzGvb9dTl2uGe2vNpLxmHMtExmnaiM,410
 omdev/scripts/importtrace.py,sha256=Jbo3Yk2RAbE8_tJ97iTcVNpoxCJxrRb2tl1W_CV3NG0,14067
-omdev/scripts/interp.py,sha256=2poCQ_45tO0-X16KeoxiAyuTfltSjnPETejlYEm9VY4,71527
-omdev/scripts/pyproject.py,sha256=wtmF5ji8fHw2qzc1aEve09Ozv0ncuZmTQ8IbKCJVAQ0,158223
+omdev/scripts/interp.py,sha256=ls04OE3g_TpiK5_inRfOzUvBGUtBgSt5vnkqC3eZhTM,72076
+omdev/scripts/pyproject.py,sha256=g5f0AyRii_DSY-D6mMDUVqKNPNxWXAQJoWrYi1F2KGM,158772
 omdev/scripts/slowcat.py,sha256=lssv4yrgJHiWfOiHkUut2p8E8Tq32zB-ujXESQxFFHY,2728
 omdev/toml/__init__.py,sha256=Y3l4WY4JRi2uLG6kgbGp93fuGfkxkKwZDvhsa0Rwgtk,15
 omdev/toml/parser.py,sha256=84bn09uhYHwQGyfww6Rw6y1RxPAE_HDltODOSakcqDM,29186
@@ -126,9 +129,9 @@ omdev/tools/piptools.py,sha256=-jR5q3w4sHqntxCLExFCBNIARB788FUsAbJ62PK2sBU,2774
 omdev/tools/proftools.py,sha256=8ZU9x_Dq8eT2ZFwU9sJpDIvxcIn9qBc8y2ELKPb5e5M,1382
 omdev/tools/rsttool.py,sha256=suwsfseUf8GH8rYmYygTUdif-Jk_bX1g9fYRLXaKkmM,1340
 omdev/tools/sqlrepl.py,sha256=tmFZh80-xsGM62dyQ7_UGLebChrj7IHbIPYBWDJMgVk,5741
-omdev-0.0.0.dev90.dist-info/LICENSE,sha256=B_hVtavaA8zCYDW99DYdcpDLKz1n3BBRjZrcbv8uG8c,1451
-omdev-0.0.0.dev90.dist-info/METADATA,sha256=plClp6X-AE56Ysn60nGpcnVUoKZ0VfnQ3VcgvGMU0YU,1492
-omdev-0.0.0.dev90.dist-info/WHEEL,sha256=P9jw-gEje8ByB7_hXoICnHtVCrEwMQh-630tKvQWehc,91
-omdev-0.0.0.dev90.dist-info/entry_points.txt,sha256=dHLXFmq5D9B8qUyhRtFqTGWGxlbx3t5ejedjrnXNYLU,33
-omdev-0.0.0.dev90.dist-info/top_level.txt,sha256=1nr7j30fEWgLYHW3lGR9pkdHkb7knv1U1ES1XRNVQ6k,6
-omdev-0.0.0.dev90.dist-info/RECORD,,
+omdev-0.0.0.dev91.dist-info/LICENSE,sha256=B_hVtavaA8zCYDW99DYdcpDLKz1n3BBRjZrcbv8uG8c,1451
+omdev-0.0.0.dev91.dist-info/METADATA,sha256=ETMAeQT2JIuqgfINKZYliLmHBYc5jW3pkOg09nSsmaY,1638
+omdev-0.0.0.dev91.dist-info/WHEEL,sha256=P9jw-gEje8ByB7_hXoICnHtVCrEwMQh-630tKvQWehc,91
+omdev-0.0.0.dev91.dist-info/entry_points.txt,sha256=dHLXFmq5D9B8qUyhRtFqTGWGxlbx3t5ejedjrnXNYLU,33
+omdev-0.0.0.dev91.dist-info/top_level.txt,sha256=1nr7j30fEWgLYHW3lGR9pkdHkb7knv1U1ES1XRNVQ6k,6
+omdev-0.0.0.dev91.dist-info/RECORD,,