diff --git a/__init__.py b/__init__.py new file mode 100644 index 0000000..4fe47cb --- /dev/null +++ b/__init__.py @@ -0,0 +1,87 @@ +import bottombar as bb +import readchar +import string + + +class BottomBar(object): + FUNC_QUIT = 0 + FUNC_BOOL = 1 + FUNC_TEXT = 2 + # + F_KEYS = { + 1: '\x1bOP', + 2: '\x1bOQ', + 3: '\x1bOR', + 4: '\x1bOS', + 5: '\x1b[15~', + 6: '\x1b[17~', + 7: '\x1b[18~', + 8: '\x1b[19~', + 9: '\x1b[20~', + 12: '\x1b[24~' + } + + def __init__(self, *args, **kwargs): + self.__bb_args__ = {} + self.__bb_bar__ = {} + self.__bb_data__ = {} + self.__key_to_name__ = {} + self.__key_to_func__ = {} + self.edit_active = None + # + self.__bb_args__['__info__'] = (args, kwargs) + + def add_entry(self, name, f_key, function, *args, **kwargs): + if len(args) == 0: + args = ("", ) + # add bb arguments for later barentry creation + self.__bb_args__[name] = (args, kwargs) + # store data for entry + if function in [self.FUNC_BOOL]: + self.__bb_data__[name] = False + elif function in [self.FUNC_TEXT]: + self.__bb_data__[name] = "" + # store function and name for key + f_inp = self.F_KEYS.get(f_key) + if f_inp is not None: + self.__key_to_func__[f_inp] = function + self.__key_to_name__[f_inp] = name + + def get_entry(self, name, default=None): + return self.__bb_data__.get(name, default) + + def edit(self, data): + if data in string.ascii_letters or data in string.digits or data in string.punctuation: + self.__bb_data__[self.edit_active] += data + elif data == "\n": + self.edit_active = None + + def run(self): + for name in self.__bb_args__: + args, kwargs = self.__bb_args__.get(name, (None, None)) + self.__bb_bar__[name] = bb.add(*args, **kwargs) + while True: + # Update bar content + for name in self.__bb_data__: + data = self.__bb_data__[name] + if type(data) is type(True): + data = "on" if data else "off" + if name != self.edit_active: + if data == "": + data = "-" + else: + data += "#" + self.__bb_bar__[name].text = data + # keystroke action + data = readchar.readkey() + if self.edit_active is not None: + self.edit(data) + else: + if data in self.__key_to_func__: + name = self.__key_to_name__.get(data) + if self.__key_to_func__[data] is self.FUNC_QUIT: + break + elif self.__key_to_func__[data] is self.FUNC_BOOL: + self.__bb_data__[name] = not self.__bb_data__[name] + elif self.__key_to_func__[data] is self.FUNC_TEXT: + self.edit_active = name diff --git a/requirements.txt b/requirements.txt new file mode 100644 index 0000000..d75ec3f --- /dev/null +++ b/requirements.txt @@ -0,0 +1 @@ +bottombar