jupyter-analysis-tools 1.7.0__py3-none-any.whl

jupyter_analysis_tools/widgets.py

@@ -0,0 +1,89 @@
+# -*- coding: utf-8 -*-
+# widgets.py
+
+import os
+
+import ipywidgets as ui
+
+
+def showBoolStatus(value, description, invertcolor=False):
+    """Pretty prints the status of a boolean variable *value* along with
+    the provided description in a green color for True and in a red color
+    for False values. *invertcolor* allows to flip the color assignment."""
+    from IPython.display import HTML, display
+
+    statuscolor = "darkgreen"
+    if value ^ invertcolor:
+        statuscolor = "darkred"
+    descr = description[0].lower() + description[1:-1]
+    if description[-1].isalnum():
+        descr += description[-1]
+    statustext = '<h4 style="color: {};">Yes, {}!</h4>'.format(statuscolor, descr)
+    if value:
+        statustext = '<h4 style="color: {};">Do not {}!</h4>'.format(statuscolor, descr)
+    display(
+        HTML(
+            f'<div style="border-style: solid; border-color: {statuscolor};'
+            " border-width: 1px; padding: 0em 1em .5em 1em; margin: 1em 0em;"
+            f' width: {len(descr) * 0.75}em;">'
+            + statustext
+            + "</div>"
+        )
+    )
+
+
+class PathSelector:
+    def __init__(self, start_dir, select_file=True):
+        self.file = None
+        self.select_file = select_file
+        self.cwd = start_dir
+        self.select = ui.SelectMultiple(options=["init"], value=(), rows=10, description="")
+        self.accord = ui.Accordion(children=[self.select])
+
+        self.accord.selected_index = None  # Start closed (showing path only)
+        self.refresh(self.cwd)
+        self.select.observe(self.on_update, "value")
+
+    def on_update(self, change):
+        if len(change["new"]) > 0:
+            self.refresh(change["new"][0])
+
+    def refresh(self, item):
+        path = os.path.abspath(os.path.join(self.cwd, item))
+
+        if os.path.isfile(path):
+            if self.select_file:
+                self.accord.set_title(0, path)
+                self.file = path
+                self.accord.selected_index = None
+            else:
+                self.select.value = ()
+
+        else:  # os.path.isdir(path)
+            self.file = None
+            self.cwd = path
+
+            # Build list of files and dirs
+            keys = ["[..]"]
+            for item in os.listdir(path):
+                if item[0] == ".":
+                    continue
+                elif os.path.isdir(os.path.join(path, item)):
+                    keys.append("[" + item + "]")
+                else:
+                    keys.append(item)
+
+            # Sort and create list of output values
+            keys.sort(key=str.lower)
+            vals = []
+            for k in keys:
+                if k[0] == "[":
+                    vals.append(k[1:-1])  # strip off brackets
+                else:
+                    vals.append(k)
+
+            # Update widget
+            self.accord.set_title(0, path)
+            self.select.options = list(zip(keys, vals))
+            with self.select.hold_trait_notifications():
+                self.select.value = ()