some extended functionality

This commit is contained in:
Dirk Alders 2023-07-31 19:18:19 +02:00
parent e47647d7e9
commit ff0434370a

View File

@ -2,11 +2,13 @@ import bottombar as bb
import readchar import readchar
import string import string
DEBUG = False
class BottomBar(object): class BottomBar(object):
FUNC_QUIT = 0 FUNC_QUIT = 0
FUNC_BOOL = 1 FUNC_INFO = 1
FUNC_TEXT = 2 FUNC_BOOL = 2
FUNC_TEXT = 3
# #
F_KEYS = { F_KEYS = {
1: '\x1bOP', 1: '\x1bOP',
@ -34,18 +36,20 @@ class BottomBar(object):
def add_entry(self, name, f_key, function, *args, **kwargs): def add_entry(self, name, f_key, function, *args, **kwargs):
if len(args) == 0: if len(args) == 0:
args = ("", ) args = ("", )
# add bb arguments for later barentry creation
self.__bb_args__[name] = (args, kwargs)
# store data for entry # store data for entry
if function in [self.FUNC_BOOL]: if function in [self.FUNC_BOOL]:
self.__bb_data__[name] = False self.__bb_data__[name] = False
elif function in [self.FUNC_TEXT]: elif function in [self.FUNC_TEXT]:
self.__bb_data__[name] = "" self.__bb_data__[name] = ""
elif function in [self.FUNC_INFO]:
self.__bb_data__[name] = kwargs.pop('infotext')
# store function and name for key # store function and name for key
f_inp = self.F_KEYS.get(f_key) f_inp = self.F_KEYS.get(f_key)
if f_inp is not None: if f_inp is not None:
self.__key_to_func__[f_inp] = function self.__key_to_func__[f_inp] = function
self.__key_to_name__[f_inp] = name self.__key_to_name__[f_inp] = name
# add bb arguments for later barentry creation
self.__bb_args__[name] = (args, kwargs)
def get_entry(self, name, default=None): def get_entry(self, name, default=None):
return self.__bb_data__.get(name, default) return self.__bb_data__.get(name, default)
@ -53,8 +57,12 @@ class BottomBar(object):
def edit(self, data): def edit(self, data):
if data in string.ascii_letters or data in string.digits or data in string.punctuation: if data in string.ascii_letters or data in string.digits or data in string.punctuation:
self.__bb_data__[self.edit_active] += data self.__bb_data__[self.edit_active] += data
elif data == "\x7f":
self.__bb_data__[self.edit_active] = self.__bb_data__[self.edit_active][:-1]
elif data == "\n": elif data == "\n":
self.edit_active = None self.edit_active = None
elif DEBUG:
print("unused keystroke in edit method:", repr(data))
def run(self): def run(self):
for name in self.__bb_args__: for name in self.__bb_args__:
@ -64,13 +72,16 @@ class BottomBar(object):
# Update bar content # Update bar content
for name in self.__bb_data__: for name in self.__bb_data__:
data = self.__bb_data__[name] data = self.__bb_data__[name]
hotkey = [k for k, v in self.__key_to_name__.items() if v == name][0]
func = self.__key_to_func__[hotkey]
if func in [self.FUNC_BOOL, self.FUNC_TEXT]:
if type(data) is type(True): if type(data) is type(True):
data = "on" if data else "off" data = "on" if data else "off"
if name != self.edit_active: if name != self.edit_active:
if data == "": if data == "":
data = "-" data = "-"
else: else:
data += "#" data += "\u2588"
self.__bb_bar__[name].text = data self.__bb_bar__[name].text = data
# keystroke action # keystroke action
data = readchar.readkey() data = readchar.readkey()
@ -81,7 +92,11 @@ class BottomBar(object):
name = self.__key_to_name__.get(data) name = self.__key_to_name__.get(data)
if self.__key_to_func__[data] is self.FUNC_QUIT: if self.__key_to_func__[data] is self.FUNC_QUIT:
break break
elif self.__key_to_func__[data] is self.FUNC_INFO:
print(self.__bb_data__[name])
elif self.__key_to_func__[data] is self.FUNC_BOOL: elif self.__key_to_func__[data] is self.FUNC_BOOL:
self.__bb_data__[name] = not self.__bb_data__[name] self.__bb_data__[name] = not self.__bb_data__[name]
elif self.__key_to_func__[data] is self.FUNC_TEXT: elif self.__key_to_func__[data] is self.FUNC_TEXT:
self.edit_active = name self.edit_active = name
elif DEBUG:
print("unused keystroke in run method:", repr(data))