omdev 0.0.0.dev89__tar.gz → 0.0.0.dev91__tar.gz

{omdev-0.0.0.dev89/omdev.egg-info → omdev-0.0.0.dev91}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: omdev
-Version: 0.0.0.dev89
+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.dev89
+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,7 +21,8 @@ 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.1; 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
 Requires-Dist: black~=24.10; extra == "black"
@@ -35,7 +36,9 @@ 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"
+Requires-Dist: tokenize-rt~=6.1; extra == "tokens"
 Provides-Extra: wheel
 Requires-Dist: wheel~=0.44; extra == "wheel"

{omdev-0.0.0.dev89 → omdev-0.0.0.dev91}/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-0.0.0.dev89 → omdev-0.0.0.dev91}/omdev/cli/_pathhack.py

@@ -7,20 +7,33 @@ problematic empty string is added, so a sys.meta_path hook is prepended.
 
 See:
   https://github.com/python/cpython/blob/da1e5526aee674bb33c17a498aa3781587b9850c/Python/sysmodule.c#L3939
+  https://github.com/python/cpython/blob/db0a1b8c1291bf1aa9e016e43bc2f7ed0acf83bd/Modules/getpath.py
+  https://github.com/python/cpython/blob/db0a1b8c1291bf1aa9e016e43bc2f7ed0acf83bd/Modules/getpath.c
+  https://github.com/python/cpython/blob/db0a1b8c1291bf1aa9e016e43bc2f7ed0acf83bd/Doc/library/sys_path_init.rst
 """
 import os.path
 import site
 import sys
 
 
-def _remove_empty_from_sys_path() -> None:
-    while '' in sys.path:
-        sys.path.remove('')
-
-
 class _PathHackMetaFinder:
+    def __init__(
+            self,
+            removed_paths=None,  # type: list[str] | None
+    ) -> None:
+        super().__init__()
+        if removed_paths is None:
+            removed_paths = ['', '.', os.getcwd()]
+        self._removed_paths = removed_paths
+        self.remove_paths()
+
+    def remove_paths(self) -> None:
+        for p in self._removed_paths:
+            while p in sys.path:
+                sys.path.remove(p)
+
     def find_spec(self, fullname, path, target=None):
-        _remove_empty_from_sys_path()
+        self.remove_paths()
         return None  # noqa
 
 

omdev-0.0.0.dev91/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-0.0.0.dev91/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-0.0.0.dev89 → omdev-0.0.0.dev91}/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-0.0.0.dev89 → omdev-0.0.0.dev91}/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.dev89 → omdev-0.0.0.dev91/omdev.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: omdev
-Version: 0.0.0.dev89
+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.dev89
+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,7 +21,8 @@ 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.1; 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
 Requires-Dist: black~=24.10; extra == "black"
@@ -35,7 +36,9 @@ 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"
+Requires-Dist: tokenize-rt~=6.1; extra == "tokens"
 Provides-Extra: wheel
 Requires-Dist: wheel~=0.44; extra == "wheel"

{omdev-0.0.0.dev89 → omdev-0.0.0.dev91}/omdev.egg-info/SOURCES.txt

@@ -108,6 +108,9 @@ omdev/precheck/lite.py
 omdev/precheck/main.py
 omdev/precheck/manifests.py
 omdev/precheck/scripts.py
+omdev/prompttoolkit/__init__.py
+omdev/prompttoolkit/apps/__init__.py
+omdev/prompttoolkit/apps/ncdu.py
 omdev/pyproject/__init__.py
 omdev/pyproject/__main__.py
 omdev/pyproject/cexts.py

{omdev-0.0.0.dev89 → omdev-0.0.0.dev91}/omdev.egg-info/requires.txt

@@ -1,4 +1,4 @@
-omlish==0.0.0.dev89
+omlish==0.0.0.dev91
 
 [all]
 black~=24.10
@@ -8,7 +8,8 @@ pcpp~=1.30
 docutils~=0.21
 mypy~=1.11
 gprof2dot~=2024.6
-tokenize_rt~=6.1
+prompt-toolkit~=3.0
+tokenize-rt~=6.1
 wheel~=0.44
 
 [black]
@@ -28,8 +29,11 @@ mypy~=1.11
 [prof]
 gprof2dot~=2024.6
 
+[prompttoolkit]
+prompt-toolkit~=3.0
+
 [tokens]
-tokenize_rt~=6.1
+tokenize-rt~=6.1
 
 [wheel]
 wheel~=0.44

{omdev-0.0.0.dev89 → omdev-0.0.0.dev91}/pyproject.toml

@@ -11,8 +11,8 @@ authors = [
 ]
 urls = {source = 'https://github.com/wrmsr/omlish'}
 license = {text = 'BSD-3-Clause'}
-requires-python = '~=3.12'
-version = '0.0.0.dev89'
+requires-python = '>=3.12'
+version = '0.0.0.dev91'
 classifiers = [
     'License :: OSI Approved :: BSD License',
     'Development Status :: 2 - Pre-Alpha',
@@ -22,7 +22,7 @@ classifiers = [
 ]
 description = 'omdev'
 dependencies = [
-    'omlish == 0.0.0.dev89',
+    'omlish == 0.0.0.dev91',
 ]
 
 [project.optional-dependencies]
@@ -34,7 +34,8 @@ all = [
     'docutils ~= 0.21',
     'mypy ~= 1.11',
     'gprof2dot ~= 2024.6',
-    'tokenize_rt ~= 6.1',
+    'prompt-toolkit ~= 3.0',
+    'tokenize-rt ~= 6.1',
     'wheel ~= 0.44',
 ]
 black = [
@@ -54,8 +55,11 @@ mypy = [
 prof = [
     'gprof2dot ~= 2024.6',
 ]
+prompttoolkit = [
+    'prompt-toolkit ~= 3.0',
+]
 tokens = [
-    'tokenize_rt ~= 6.1',
+    'tokenize-rt ~= 6.1',
 ]
 wheel = [
     'wheel ~= 0.44',