p3lib 1.1.116__tar.gz → 1.1.117__tar.gz

{p3lib-1.1.116 → p3lib-1.1.117}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: p3lib
-Version: 1.1.116
+Version: 1.1.117
 Summary: A group of python modules for networking, plotting data, config storage, automating boot scripts, ssh access and user input output.
 Home-page: https://github.com/pjaos/p3lib
 License: MIT

{p3lib-1.1.116 → p3lib-1.1.117}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "p3lib"
-version = "1.1.116"
+version = "1.1.117"
 description = "A group of python modules for networking, plotting data, config storage, automating boot scripts, ssh access and user input output."
 authors = ["Paul Austen <pjaos@gmail.com>"]
 license = "MIT License"

{p3lib-1.1.116 → p3lib-1.1.117}/src/p3lib/ngt.py

@@ -12,6 +12,7 @@ import platform
 from time import sleep
 from queue import Queue
 from time import time, strftime, localtime
+from pathlib import Path
 
 from p3lib.helper import getProgramVersion
 
@@ -836,3 +837,84 @@ class YesNoDialog(object):
     def close(self):
         """@brief Close the boolean dialog."""
         self._dialog.close()
+
+class local_file_picker(ui.dialog):
+    """@brief Allows the user to select local files and folders.
+       This is is a slight change on https://github.com/zauberzeug/nicegui/blob/main/examples/local_file_picker/local_file_picker.py"""
+
+    def __init__(self,
+                 directory: str,
+                 *,
+                 upper_limit: str = None,
+                 multiple: bool = False,
+                 show_hidden_files: bool = False) -> None:
+        """Local File Picker
+
+        This is a simple file picker that allows you to select a file from the local filesystem where NiceGUI is running.
+
+        :param directory: The directory to start in.
+        :param upper_limit: The directory to stop at (None: no limit, default: same as the starting directory).
+        :param multiple: Whether to allow multiple files to be selected.
+        :param show_hidden_files: Whether to show hidden files.
+        """
+        super().__init__()
+
+        self.path = Path(directory).expanduser()
+        if upper_limit is None:
+            self.upper_limit = None
+        else:
+            self.upper_limit = Path(directory if upper_limit == ... else upper_limit).expanduser()
+        self.show_hidden_files = show_hidden_files
+
+        with self, ui.card():
+            self.add_drives_toggle()
+            self.grid = ui.aggrid({
+                'columnDefs': [{'field': 'name', 'headerName': 'File'}],
+                'rowSelection': 'multiple' if multiple else 'single',
+            }, html_columns=[0]).classes('w-96').classes('h-96').on('cellDoubleClicked', self.handle_double_click)
+            with ui.row().classes('w-full justify-end'):
+                self._select_folder_checkbox = ui.switch('Select Folder')
+                #self._select_folder_checkbox = ui.checkbox('Folder')
+                self._select_folder_checkbox.tooltip("Select this if you wish to select a folder.")
+                ui.button('Cancel', on_click=self.close).props('outline')
+
+        self.update_grid()
+
+    def add_drives_toggle(self):
+        if platform.system() == 'Windows':
+            import win32api
+            drives = win32api.GetLogicalDriveStrings().split('\000')[:-1]
+            self.drives_toggle = ui.toggle(drives, value=drives[0], on_change=self.update_drive)
+
+    def update_drive(self):
+        self.path = Path(self.drives_toggle.value).expanduser()
+        self.update_grid()
+
+    def update_grid(self) -> None:
+        paths = list(self.path.glob('*'))
+        if not self.show_hidden_files:
+            paths = [p for p in paths if not p.name.startswith('.')]
+        paths.sort(key=lambda p: p.name.lower())
+        paths.sort(key=lambda p: not p.is_dir())
+
+        self.grid.options['rowData'] = [
+            {
+                'name': f'📁 <strong>{p.name}</strong>' if p.is_dir() else p.name,
+                'path': str(p),
+            }
+            for p in paths
+        ]
+        if (self.upper_limit is None and self.path != self.path.parent) or \
+                (self.upper_limit is not None and self.path != self.upper_limit):
+            self.grid.options['rowData'].insert(0, {
+                'name': '📁 <strong>..</strong>',
+                'path': str(self.path.parent),
+            })
+        self.grid.update()
+
+    def handle_double_click(self, e) -> None:
+        self.path = Path(e.args['data']['path'])
+        if self.path.is_dir() and not self._select_folder_checkbox.value:
+            self.update_grid()
+        else:
+            self.submit([str(self.path)])