README.md

@@ -1,9 +1,5 @@
-# Clintermission - Python CLI Intermission Library
-`clintermission` is a small Python library designed to quickly get a selection
-of a single or multiple items from a user. The menu shown is explicitly not a
-fullscreen menu so the user can use information in their console history to
-make a selection.
+# CLI-Intermission
+Tool to easily ask your user questions on the commandline. Interactive!
+...by abusing python prompt-toolkit's `Buffer` class.
 
-The menu is implemented by using `prompt-toolkit`'s `Buffer` class.
-
-Inspired by [go promptui](https://github.com/manifoldco/promptui).
+Highly inspired by [go promptui](https://github.com/manifoldco/promptui)

clintermission/__init__.py

@@ -1,13 +1,11 @@
 # Written by Sebastian Lohff <seba@someserver.de>
 # Licensed under Apache License 2.0
 
-from clintermission.climenu import CliMenu, CliMultiMenu, CliMenuStyle, CliSelectionStyle, CliMenuCursor, CliMenuTheme, cli_select_item
+from clintermission.climenu import CliMenu, CliMenuStyle, CliMenuCursor, CliMenuTheme, cli_select_item
 
 __all__ = [
     CliMenu,
-    CliMultiMenu,
     CliMenuStyle,
-    CliSelectionStyle,
     CliMenuCursor,
     CliMenuTheme,
     cli_select_item,

clintermission/climenu.py

@@ -12,23 +12,20 @@ from prompt_toolkit import search
 from prompt_toolkit.widgets import SearchToolbar
 
 
-class _CliMenuHeader:
+class CliMenuHeader:
     """Hold a menu header"""
-    def __init__(self, text, indent=False, style=None):
+    def __init__(self, text, indent=False):
         self.text = text
         self.indent = indent
-        self.style = style
         self.focusable = False
 
 
-class _CliMenuOption:
+class CliMenuOption:
     """Hold a menu option"""
-    def __init__(self, text, num, item=None, style=None, highlighted_style=None):
+    def __init__(self, text, num, item=None):
         self.text = text
         self.num = num
         self.item = item
-        self.style = style
-        self.highlighted_style = highlighted_style
         self.focusable = True
 
 
@@ -36,9 +33,9 @@ class CliMenuCursor:
     """Collection of cursors pointing at the active menu item"""
     BULLET = '●'
     TRIANGLE = '▶'
-    ASCII_STAR = '*'
-    ASCII_ARROW = '-->'
-    ASCII_CAT = '=^.^='
+    CLI_STAR = '*'
+    CLI_ARROW = '-->'
+    CLI_CAT = '=^.^='
     CAT = '😸'
     ARROW = '→'
 
@@ -48,95 +45,71 @@ class CliMenuStyle:
 
     Allows to select header, option and selected option color
     """
-    def __init__(self, option='', highlighted='', text='', selected=None, selected_highlighted=None):
-        self.option = option
-        self.highlighted = highlighted
-        self.text = text
-        self.selected = selected
-        self.selected_highlighted = selected_highlighted
-
-
-class CliSelectionStyle:
-    SQUARE_BRACKETS = ('[x]', '[ ]')
-    ROUND_BRACKETS = ('(x)', '( )')
-    CHECKMARK = ('✔', '✖')
-    THUMBS = ('👍', '👎')
-    SMILEY = ('🙂', '🙁')
-    SMILEY_EXTREME = ('😁', '😨')
+    def __init__(self, option_style='', highlight_style='', header_style=''):
+        self.option_style = option_style
+        self.highlight_style = highlight_style
+        self.header_style = header_style
+
+        # self.option_color = '#aa0000'
+        # #self.highlight_color = 'fg:ansiblue bg:ansired bold'
+        # self.highlight_color = 'bold'
+        # self.cursor = ' --> '
+        # self.cursor = ' ● '
+        # self.no_cursor = '     '
+        # self.header_color = '#aa22cc'
+        # self.option_indent = 4
+        # self.header_indent = 4
 
 
 class CliMenuTheme:
     BASIC = CliMenuStyle()
-    BASIC_BOLD = CliMenuStyle(text='bold', highlighted='bold')
+    BASIC_BOLD = CliMenuStyle(header_style='bold', highlight_style='bold')
     RED = CliMenuStyle('#aa0000', '#ee0000', '#aa0000')
     CYAN = CliMenuStyle('cyan', 'lightcyan', 'cyan')
     BLUE = CliMenuStyle('ansiblue', 'ansired', 'ansiblue')
     ANSI_CYAN = CliMenuStyle('ansicyan', 'ansibrightcyan', 'ansicyan')
-    BOLD_HIGHLIGHT = CliMenuStyle(text='bold', highlighted='bold fg:black bg:white')
-
-
-class _EmptyParameter:
-    pass
+    BOLD_HIGHLIGHT = CliMenuStyle(header_style='bold', highlight_style='bold fg:black bg:white')
 
 
 class CliMenu:
     default_style = CliMenuTheme.BASIC
     default_cursor = CliMenuCursor.TRIANGLE
 
-    @classmethod
-    def set_default_style(cls, style):
-        cls.default_style = style
-
-    @classmethod
-    def set_default_cursor(cls, cursor):
-        cls.default_cursor = cursor
-
     def __init__(self, options=None, header=None, cursor=None, style=None,
-                 indent=2, dedent_selection=False, initial_pos=0,
-                 option_prefix=' ', option_suffix='', right_pad_options=False):
+                 indent=2, dedent_selection=False):
         self._items = []
         self._item_num = 0
         self._ran = False
         self._success = None
         self._pos = 0
-        self._initial_pos = initial_pos
-        self._option_prefix = option_prefix
-        self._option_suffix = option_suffix
         self._option_indent = indent
         self._header_indent = indent
         self._dedent_selection = dedent_selection
-        self._right_pad_options = right_pad_options
 
-        self._cursor = cursor if cursor is not None else self.default_cursor
-        self._style = style if style is not None else self.default_style
+        self._cursor = cursor
+        if not self._cursor:
+            self._cursor = self.default_cursor
+
+        self._style = style
+        if not self._style:
+            self._style = self.default_style
 
         if header:
-            self.add_text(header, indent=False)
+            self.add_header(header, indent=False)
 
         if options:
             for option in options:
-                if isinstance(option, tuple):
+                if isinstance(option, tuple) and len(option) == 2:
                     self.add_option(*option)
-                elif isinstance(option, dict):
-                    self.add_option(**option)
-                elif isinstance(option, str):
-                    self.add_option(option)
                 else:
-                    raise ValueError("Option needs to be either tuple, dict or string, found '{}' of type {}"
-                                     .format(option, type(option)))
-
-    def add_header(self, *args, **kwargs):
-        return self.add_text(*args, **kwargs)
+                    self.add_option(option, option)
 
-    def add_text(self, title, indent=True, style=None):
+    def add_header(self, title, indent=True):
         for text in title.split('\n'):
-            self._items.append(_CliMenuHeader(text, indent=indent, style=style))
+            self._items.append(CliMenuHeader(text, indent=indent))
 
-    def add_option(self, text, item=_EmptyParameter, style=None, highlighted_style=None):
-        if item == _EmptyParameter:
-            item = text
-        self._items.append(_CliMenuOption(text, self._item_num, item=item,
-                                          style=style, highlighted_style=highlighted_style))
+    def add_option(self, text, item=None):
+        self._items.append(CliMenuOption(text, self._item_num, item=item))
         self._item_num += 1
 
     @property
@@ -147,7 +120,7 @@ class CliMenu:
         return self._success
 
     def get_options(self):
-        return [_item for _item in self._items if isinstance(_item, _CliMenuOption)]
+        return [_item for _item in self._items if isinstance(_item, CliMenuOption)]
 
     @property
     def num_options(self):
@@ -156,6 +129,7 @@ class CliMenu:
     def get_selection(self):
         if self.success:
             item = self._items[self._pos]
+
             return (item.num, item.item)
         else:
             return (None, None)
@@ -166,46 +140,41 @@ class CliMenu:
     def get_selection_item(self):
         return self.get_selection()[1]
 
-    def _transform_prefix(self, item, lineno, prefix):
-        return prefix
+    def cursor(self):
+        return '{} '.format(self._cursor)
 
-    def _get_style(self, item, lineno, highlighted):
-        s = self._style
-        if item.focusable:
-            if highlighted:
-                return item.highlighted_style if item.highlighted_style is not None else s.highlighted
-            else:
-                return item.style if item.style is not None else s.option
-        else:
-            return item.style if item.style is not None else s.text
+    @property
+    def no_cursor(self):
+        # cursor with spaces minus dedent
+        return ' ' * (len(self._cursor) + 1 * self._dedent_selection)
 
     def _transform_line(self, ti):
         if len(list(ti.fragments)) == 0:
             return Transformation(ti.fragments)
         style, text = list(ti.fragments)[0]
         item = self._items[ti.lineno]
-        style = self._get_style(item, ti.lineno, ti.lineno == self._pos)
+        s = self._style
 
         # cursor
         indent = ''
         prefix = ''
-        suffix = ''
         if item.focusable:
             indent += ' ' * self._option_indent
-            suffix = self._option_suffix
 
             if ti.lineno == self._pos:
-                prefix += '{}{}'.format(self._cursor, self._option_prefix)
+                prefix += '{} '.format(self._cursor)
+                style = s.highlight_style
             else:
-                prefix += ' ' * len(self._cursor) + self._option_prefix + ' ' * self._dedent_selection
+                prefix += ' ' * (len(self._cursor) + 1 + 1 * self._dedent_selection)
+                style = s.option_style
         else:
             if item.indent:
                 indent += ' ' * (self._header_indent + len(self._cursor) + 1)
+            style = s.header_style
 
         items = [(s if s else style, t) for s, t in ti.fragments]
-        prefix = self._transform_prefix(item, ti.lineno, prefix)
 
-        return Transformation([('', indent), (style, prefix)] + items + [(style, suffix)])
+        return Transformation([('', indent), (style, prefix)] + items)
 
     def next_item(self, direction):
         if not any(item.focusable for item in self._items):
@@ -252,28 +221,7 @@ class CliMenu:
 
         return lines
 
-    def _register_extra_kb_cbs(self, kb):
-        pass
-
-    def _preflight(self):
-        if self._initial_pos < 0 or self._initial_pos >= self._item_num:
-            raise ValueError("Initial position {} is out of range, needs to be in range of [0, {})"
-                             .format(self._initial_pos, self._item_num))
-
-        if self._right_pad_options:
-            # pad all item labels with spaces to have the same length
-            max_item_len = max([len(_item.text) for _item in self._items if _item.focusable])
-            for item in self._items:
-                if item.focusable:
-                    item.text += " " * (max_item_len - len(item.text))
-
-    def _accept(self, event):
-        self._success = True
-        event.app.exit()
-
     def _run(self):
-        self._preflight()
-
         class MenuColorizer(Processor):
             def apply_transformation(_self, ti):
                 return self._transform_line(ti)
@@ -298,7 +246,7 @@ class CliMenu:
 
         @self._kb.add('N', filter=~is_searching)
         @self._kb.add('n', filter=~is_searching)
-        def search_inc(event):
+        def search_inc(event, filter=is_searching):
             if not self._bufctrl.search_state.text:
                 return
 
@@ -310,8 +258,10 @@ class CliMenu:
 
         @self._kb.add('c-m', filter=~is_searching)
         @self._kb.add('right', filter=~is_searching)
+        @self._kb.add('c-space', filter=~is_searching)
         def accept(event):
-            self._accept(event)
+            self._success = True
+            event.app.exit()
 
         @self._kb.add('c-m', filter=is_searching)
         def accept_search(event):
@@ -319,8 +269,6 @@ class CliMenu:
             new_line, _ = self._doc.translate_index_to_position(self._buf.cursor_position)
             self.sync_cursor_to_line(new_line)
 
-        self._register_extra_kb_cbs(self._kb)
-
         self._searchbar = SearchToolbar(ignore_case=True)
 
         text = '\n'.join(map(lambda _x: _x.text, self._items))
@@ -336,8 +284,7 @@ class CliMenu:
                         self._searchbar])
 
         # set initial pos
-        for _ in range(self._initial_pos + 1):
-            self.next_item(1)
+        self.sync_cursor_to_line(0)
 
         app = Application(layout=Layout(split),
                           key_bindings=self._kb,
@@ -348,92 +295,6 @@ class CliMenu:
         self._ran = True
 
 
-class CliMultiMenu(CliMenu):
-    default_selection_icons = CliSelectionStyle.SQUARE_BRACKETS
-
-    @classmethod
-    def set_default_selector_icons(cls, selection_icons):
-        cls.default_selection_icons = selection_icons
-
-    def __init__(self, *args, selection_icons=None, min_selection_count=0, **kwargs):
-        self._multi_selected = []
-        self._min_selection_count = min_selection_count
-        self._selection_icons = selection_icons if selection_icons is not None else self.default_selection_icons
-        super().__init__(*args, **kwargs)
-
-    def add_option(self, text, item=_EmptyParameter, selected=False,
-                   style=None, highlighted_style=None, selected_style=None, selected_highlighted_style=None):
-        super().add_option(text, item, style, highlighted_style=highlighted_style)
-        self._items[-1].selected_style = selected_style
-        self._items[-1].selected_highlighted_style = selected_highlighted_style
-        if selected:
-            self._multi_selected.append(len(self._items) - 1)
-
-    def get_selection(self):
-        if self.success:
-            return [(self._items[n].num, self._items[n].item) for n in self._multi_selected]
-        else:
-            return None
-
-    def get_selection_num(self):
-        if self.success:
-            return [self._items[n].num for n in self._multi_selected]
-        else:
-            return None
-
-    def get_selection_item(self):
-        if self.success:
-            return [self._items[n].item for n in self._multi_selected]
-        else:
-            return None
-
-    def _register_extra_kb_cbs(self, kb):
-        @kb.add('space', filter=~is_searching)
-        @kb.add('right', filter=~is_searching)
-        def mark(event):
-            if self._pos not in self._multi_selected:
-                self._multi_selected.append(self._pos)
-            else:
-                self._multi_selected.remove(self._pos)
-
-    def _transform_prefix(self, item, lineno, prefix):
-        if item.focusable:
-            if lineno in self._multi_selected:
-                icon = self._selection_icons[0]
-            else:
-                icon = self._selection_icons[1]
-            return "{}{} ".format(prefix, icon)
-        else:
-            return prefix
-
-    def _get_style(self, item, lineno, highlighted):
-        s = self._style
-        if item.focusable and lineno in self._multi_selected:
-            if highlighted:
-                if item.selected_highlighted_style is not None:
-                    return item.selected_highlighted_style
-                if s.selected_highlighted is not None:
-                    return s.selected_highlighted
-            else:
-                if item.selected_style is not None:
-                    return item.selected_style
-                if s.selected is not None:
-                    return s.selected
-
-        # no style specified or no selected state, call parent
-        return super()._get_style(item, lineno, highlighted)
-
-    def _preflight(self):
-        super()._preflight()
-        if self._min_selection_count > self._item_num:
-            raise ValueError("A minimum of {} items was requested for successful selection but only {} exist"
-                             .format(self._min_selection_count, self._item_num))
-
-    def _accept(self, event):
-        if len(self._multi_selected) >= self._min_selection_count:
-            super()._accept(event)
-
-
 def cli_select_item(options, header=None, abort_exc=ValueError, abort_text="Selection aborted.", style=None,
                     return_single=True):
     """Helper function to quickly get a selection with just a few arguments"""

examples/basic_usage.py

@@ -1,88 +0,0 @@
-#!/usr/bin/env python3
-from clintermission import CliMenu, CliMenuTheme, CliMenuStyle, CliMenuCursor, cli_select_item
-
-# --- basic menu ---
-q = ["Foo", "Bar", "Baz"]
-m = CliMenu(q, "Time to choose:\n")
-
-print("You selected", m.get_selection())
-print()
-
-# basic menu with an item assigned to each option and detentation of selection
-q = [("Foo", 23), ("Bar", 34), ("Baz", 42)]
-m = CliMenu(q, "Time to choose:\n", cursor="--->", dedent_selection=True)
-
-print("You selected {} index {} item {}".format(m.get_selection(), m.get_selection_num(), m.get_selection_item()))
-print()
-
-# --- themes ---
-q = ["Foo", "Bar", "Baz"]
-m = CliMenu(q, "Time to choose:\n", style=CliMenuTheme.RED)
-print("You selected", m.get_selection())
-print()
-
-
-# --- custom themes ---
-style = CliMenuStyle(option='blue', highlighted='cyan', text='green')
-q = ["Foo", "Bar", "Baz"]
-m = CliMenu(q, "Choose in style:\n", style=style)
-print("You selected", m.get_selection())
-print()
-
-# --- theme defaults ---
-CliMenu.set_default_cursor(CliMenuCursor.BULLET)
-CliMenu.set_default_style(CliMenuTheme.BOLD_HIGHLIGHT)
-
-q = ["Foo", "Bar", "Baz"]
-m = CliMenu(q, "Time to choose:\n")
-
-print("You selected", m.get_selection())
-print()
-
-# --- multiple headers ---
-m = CliMenu()
-
-m.add_header("Time to choose:\n", indent=False)
-m.add_text("=== Category 1 ===")
-m.add_option("Foo")
-m.add_option("Bar")
-m.add_option("Baz")
-m.add_header('', indent=False)
-
-m.add_text("=== Category 2 ===")
-m.add_option("Cat 1")
-m.add_option("Cat 2")
-m.add_option("Cat 3")
-
-print("You selected", m.get_selection())
-print()
-
-# --- with shortcut ---
-try:
-    result = cli_select_item(["Foo", "Bar", "Baz"], abort_text="Selection faiiiled!")
-    print("You selected", result)
-except ValueError as e:
-    print(e)
-print()
-
-# --- with shortcut, shows no menu when only a single option is provided (can be disabled with return_single=False) ---
-result = cli_select_item(["Single Foo"])
-print("Directly selected for you as it was the only option:", result)
-print()
-
-# --- prefix/suffix ---
-q = ["Foo", "Bar", "Baz"]
-m = CliMenu(q, "Time to choose:\n", option_prefix=' <<<', option_suffix='>>>')
-print("You selected", m.get_selection())
-print()
-
-# --- colorize everything ---
-m = CliMenu()
-m.add_text("Time to choose:\n", style="green")
-m.add_option("Yellow foo", style="yellow", highlighted_style="black bg:yellow")
-m.add_option("Green bar", style="green", highlighted_style="black bg:green")
-m.add_option("Blue baz", style="blue", highlighted_style="black bg:blue")
-m.add_text("\n...go for it!")
-
-print("You selected", m.get_selection())
-print()

examples/cursorless.py

@@ -1,9 +0,0 @@
-#!/usr/bin/env python3
-from clintermission import CliMenu, CliMenuTheme
-
-
-q = ["Foo", "Bar", "Baz baz baz baz baz"]
-m = CliMenu(q, "Time to choose:\n", style=CliMenuTheme.BOLD_HIGHLIGHT,
-            cursor='', option_prefix=' ', option_suffix=' ', right_pad_options=True)
-m.get_selection()
-print("You selected", m.get_selection())

examples/multiselect.py

@@ -1,17 +0,0 @@
-#!/usr/bin/env python3
-from clintermission import CliMultiMenu, CliSelectionStyle, CliMenuCursor
-
-
-# --- simple multiselect ---
-q = [
-    "Option 1",
-    "Option 2",
-    ("Option 3 (preselected for your convenience)", "Option 3", True),
-    "Option 4"
-]
-m = CliMultiMenu(q, "Make your choice (<space> selects, <return> accepts):\n", cursor=CliMenuCursor.ASCII_ARROW,
-                 selection_icons=CliSelectionStyle.CHECKMARK)
-
-print("You selected", m.get_selection())
-print("You selected num:", m.get_selection_num())
-print("You selected item:", m.get_selection_item())

setup.cfg

@@ -3,11 +3,9 @@ name = clintermission
 version = 0.1.0
 author = Sebastian Lohff
 author-email = seba@someserver.de
-license = Apache-2.0
-home-page = https://github.com/sebageek/clintermission
-description = Non-fullscreen command-line selection menu
+home-page = https://git.someserver.de/seba/clintermission
+description = Non-fullscreen commandline selection menu
 long-description = file: README.md
-long-description-content-type = text/markdown
 platform = any
 keywords = cli, menu
 classifiers =
@@ -17,7 +15,6 @@ classifiers =
     Programming Language :: Python :: 3.5
     Programming Language :: Python :: 3.6
     Programming Language :: Python :: 3.7
-    Programming Language :: Python :: 3.8
     License :: OSI Approved :: Apache Software License
 
 [options]