def test_view_string():
s = String('aaa')
assert len(s) == 3
assert s[:2] == 'aa'
if six.PY2:
s = String(u'a啊啊')
assert len(s) == 5
assert s[:2] == 'a'
assert s[:3] == u'a啊啊'[:2]
else:
s = String('a啊啊')
assert len(s) == 5
assert s[:2] == 'a'
assert s[:3] == 'a啊'
def get_input(self, id, max_length=None):
"""
:param max_length: 输入字符数；注意区分max_width，max_width表示显示宽度，和输入无关
"""
self.view = self.ctl.find_view_by_id(id)
self.max_length = max_length
if not self.view or not isinstance(self.view, TextViewProxy):
return False, '-1'
self.x, self.y = self.get_view_x_y(id)
if self.x < 0 or self.y < 0:
return False, '-2'
# 停止绘制线程
if self.ctl.auto_re_draw:
self.ctl.refresh_thread_stop = True
self.ctl.refresh_thread.join()
self.max_width = self.get_view_max_width()
if self.max_width <= 0:
return False, '-3'
self.layout_height = self.ctl.layout.real_height
self.input_s = String(self.view.get_text())
if self.max_length and len(self.input_s.char_list) > self.max_length:
self.input_s.char_list_slice(stop=self.max_length)
self.input_char_list_index = len(self.input_s.char_list)
self.input_char_list_start = 0
self.input_char_list_end = len(self.input_s.char_list)
self.move_cursor_to_view()
self.update_index_and_show_s()
kl = KeyListener()
kl.bind_key('any', self.key_event, decorator=False)
kl.listen()
self.move_cursor_to_btm()
if self.ctl.auto_re_draw:
self.ctl.init_refresh_thread()
self.ctl.refresh_thread.start()
return True, str(self.input_s)
def get_final_text(self):
if self.visibility == Visibility.visible:
self.real_height = 1
show_text = self.text_string[:self.real_width]
elif self.visibility == Visibility.invisible:
self.real_height = 1
return ' ' * self.real_width
elif self.visibility == Visibility.gone:
return ''
# show_text是str 要转为String
show_string = String(show_text)
if self.real_width > len(show_string):
if self.gravity == Gravity.left:
show_text = show_text + ' ' * (self.real_width -
len(show_string))
elif self.gravity == Gravity.right:
show_text = ' ' * (self.real_width -
len(show_string)) + show_text
elif self.gravity == Gravity.center:
p = (self.real_width - len(show_string)) / 2.0
show_text = ' ' * int(p) + show_text + ' ' * (
int(p) if int(p) == p else int(p) + 1)
return str(self.fore) + str(self.back) + str(
self.style) + show_text + str(Style.reset_all)
def key_event(self, kl, event):
c = event.key.code
right = True
if len(repr(c)) > 1 and (repr(c).startswith("'\\")
or repr(c).startswith("u'\\")):
if event.key == Key.LEFT and self.input_char_list_index > 0:
self.input_char_list_index -= 1
if self.input_char_list_index < self.input_char_list_start:
self.input_char_list_start = self.input_char_list_index
right = False
elif event.key == Key.RIGHT and self.input_char_list_index < len(
self.input_s.char_list):
self.input_char_list_index += 1
if self.input_char_list_index > self.input_char_list_end - 1:
self.input_char_list_end = min(
self.input_char_list_index + 1,
len(self.input_s.char_list))
elif event.key == Key.BACKSPACE and self.input_char_list_index > 0:
self.input_char_list_index -= 1
self.input_s.pop(self.input_char_list_index)
self.input_char_list_end -= 1
elif event.key == Key.ENTER:
kl.stop()
return
else:
c_str = String(c)
if self.max_length:
left_length = self.max_length - len(self.input_s.char_list)
if left_length == 0:
return
if len(c_str.char_list) > left_length:
c_str.char_list_slice(stop=left_length)
self.input_s.insert_into_char_list(self.input_char_list_index,
c_str)
self.input_char_list_index += len(c_str.char_list)
self.input_char_list_end += len(c_str.char_list)
self.update_index_and_show_s(right=right)
class InputEx(object):
view = None # type:Union[None, TextViewProxy, LayoutProxy]
x = -1
y = -1
layout_height = 0
input_s = None # type: String
input_char_list_index = 0
input_char_list_start = 0
input_char_list_end = 0
cursor_index = 0
max_length = None
def __init__(self, _ctl, input_buffer=30):
"""
:param input_buffer: 读取字符的缓存；输入法一次输入多个字符时，若buffer过小会导致只读取到一部分
:type _ctl:LayoutCtl
"""
self.ctl = _ctl
self.input_buffer = input_buffer
def get_view_y(self, id, row):
y = 0
for v in row.data:
if v.id == id:
if v.visibility != Visibility.visible:
return True, -1
return True, y
y += v.real_width or 0
return False, -1
def get_view_x_y(self, id):
"""
(x, y)--------------
| 1 2 3 |
| 4 5 6 |
| 7 8 9 |
--------------(x, y)
(1,2) = 6
"""
x = 0
y = 0
layout = self.ctl.layout
if isinstance(layout, TextView):
return -1, -1 if layout.id != id else x, y
if isinstance(layout, TableRow):
ok, y = self.get_view_y(id, layout)
return x, y
for row in self.ctl.layout.data:
if row.visibility == Visibility.gone:
continue
if row.visibility == Visibility.visible:
ok, y = self.get_view_y(id, row)
if ok:
return x, y
x += 1
return -1, -1
def get_parent_max_width(self, view):
p = view.get_parent()
if p is None:
return self.ctl.get_terminal_size()[0]
else:
parent_max_width = self.get_parent_max_width(p)
width = p.get_width()
if isinstance(width, int) and width < parent_max_width:
return width
return parent_max_width
def get_view_max_width(self):
parent_max_width = self.get_parent_max_width(self.view)
parent = self.view.get_parent()
if parent is not None:
for v in parent.data:
if v.id == self.view.get_id():
continue
if v.weight is None:
parent_max_width -= v.real_width
width = self.view.get_width()
if isinstance(width, int) and width < parent_max_width:
return width
return parent_max_width
def get_input(self, id, max_length=None):
"""
:param max_length: 输入字符数；注意区分max_width，max_width表示显示宽度，和输入无关
"""
self.view = self.ctl.find_view_by_id(id)
self.max_length = max_length
if not self.view or not isinstance(self.view, TextViewProxy):
return False, '-1'
self.x, self.y = self.get_view_x_y(id)
if self.x < 0 or self.y < 0:
return False, '-2'
# 停止绘制线程
if self.ctl.auto_re_draw:
self.ctl.refresh_thread_stop = True
self.ctl.refresh_thread.join()
self.max_width = self.get_view_max_width()
if self.max_width <= 0:
return False, '-3'
self.layout_height = self.ctl.layout.real_height
self.input_s = String(self.view.get_text())
if self.max_length and len(self.input_s.char_list) > self.max_length:
self.input_s.char_list_slice(stop=self.max_length)
self.input_char_list_index = len(self.input_s.char_list)
self.input_char_list_start = 0
self.input_char_list_end = len(self.input_s.char_list)
self.move_cursor_to_view()
self.update_index_and_show_s()
kl = KeyListener()
kl.bind_key('any', self.key_event, decorator=False)
kl.listen()
self.move_cursor_to_btm()
if self.ctl.auto_re_draw:
self.ctl.init_refresh_thread()
self.ctl.refresh_thread.start()
return True, str(self.input_s)
def move_cursor_to_view(self):
s = ''
if self.layout_height - self.x > 0:
s += Cursor.UP(self.layout_height - self.x)
if self.y + self.cursor_index > 0:
s += Cursor.FORWARD(self.y + self.cursor_index)
if s:
sys.stdout.write(s)
sys.stdout.flush()
def char_list_iter(self, i, reverse):
for i in range(i - 1, -1, -1) if reverse else range(
i, len(self.input_s.char_list), 1):
yield i, self.input_s.char_list[i]
def update_index_and_show_s(self, right=True):
"""
right: 光标从左向右移动
"""
if len(self.input_s) >= self.max_width:
left_width = self.max_width
if right:
list_iter = self.char_list_iter(self.input_char_list_end, True)
else:
list_iter = self.char_list_iter(self.input_char_list_index,
False)
new_i = 0
for i, c in list_iter:
if left_width - len(c) < 0:
break
left_width -= len(c)
new_i = i
if right:
self.input_char_list_start = new_i
else:
self.input_char_list_end = new_i + 1
self.cursor_index = 0
show_s = ''
for i in range(self.input_char_list_start, self.input_char_list_end):
show_s += str(self.input_s.char_list[i])
if i < self.input_char_list_index:
self.cursor_index += len(self.input_s.char_list[i])
self.move_cursor_to_btm()
self.view.set_text(show_s)
self.ctl.re_draw()
self.move_cursor_to_view()
def move_cursor_to_btm(self):
s = ''
if self.layout_height - self.x > 0:
s += Cursor.DOWN(self.layout_height - self.x)
if self.y + self.cursor_index > 0:
s += Cursor.BACK(self.y + self.cursor_index + 2)
if s:
sys.stdout.write(s)
sys.stdout.flush()
def update_text(self):
self.view.set_text(str(self.input_s))
self.ctl.update_width()
def key_event(self, kl, event):
c = event.key.code
right = True
if len(repr(c)) > 1 and (repr(c).startswith("'\\")
or repr(c).startswith("u'\\")):
if event.key == Key.LEFT and self.input_char_list_index > 0:
self.input_char_list_index -= 1
if self.input_char_list_index < self.input_char_list_start:
self.input_char_list_start = self.input_char_list_index
right = False
elif event.key == Key.RIGHT and self.input_char_list_index < len(
self.input_s.char_list):
self.input_char_list_index += 1
if self.input_char_list_index > self.input_char_list_end - 1:
self.input_char_list_end = min(
self.input_char_list_index + 1,
len(self.input_s.char_list))
elif event.key == Key.BACKSPACE and self.input_char_list_index > 0:
self.input_char_list_index -= 1
self.input_s.pop(self.input_char_list_index)
self.input_char_list_end -= 1
elif event.key == Key.ENTER:
kl.stop()
return
else:
c_str = String(c)
if self.max_length:
left_length = self.max_length - len(self.input_s.char_list)
if left_length == 0:
return
if len(c_str.char_list) > left_length:
c_str.char_list_slice(stop=left_length)
self.input_s.insert_into_char_list(self.input_char_list_index,
c_str)
self.input_char_list_index += len(c_str.char_list)
self.input_char_list_end += len(c_str.char_list)
self.update_index_and_show_s(right=right)
def __setattr__(self, key, value):
super(TextView, self).__setattr__(key, value)
if key == 'text':
self.text_string = String(self.prompt + self.text)
def __setattr__(self, key, value):
super(TextView, self).__setattr__(key, value)
if key == 'text':
self.string_text = String(value)
def _(_, e):
global s, s_char_list_index, cursor_index
c = e.key.code
logger.debug(f's={s}, c={c}')
if repr(c).startswith("'\\x") or repr(c).startswith("u'\\x"):
if e.key == Key.LEFT and s_char_list_index > 0:
back = len(s.char_list[s_char_list_index - 1])
cursor_index -= back
s_char_list_index -= 1
sys.stdout.write(Cursor.BACK(back))
if e.key == Key.RIGHT and s_char_list_index < len(s.char_list):
forward = len(s.char_list[s_char_list_index])
cursor_index += forward
s_char_list_index += 1
sys.stdout.write(Cursor.FORWARD(forward))
if e.key == Key.BACKSPACE and s_char_list_index > 0:
back = len(s.char_list[s_char_list_index - 1])
cursor_index -= back
s_char_list_index -= 1
len_right_s = 0
show_s = ''
for c in s.get_char_list_item(s_char_list_index + 1):
len_right_s += len(c)
show_s += str(c)
s.pop(s_char_list_index)
logger.info(f'{back} - {len_right_s} - {show_s} - {repr(Cursor.BACK(len_right_s) if len_right_s else "")}')
sys.stdout.write(
Cursor.BACK(back) + ' ' * (len_right_s + back) + Cursor.BACK(len_right_s + back) + show_s)
if len_right_s > 0:
sys.stdout.write(Cursor.BACK(len_right_s))
else:
show_s = c
c_str = String(c)
after_show_s = ''
len_right_s = 0
for c in s.get_char_list_item(s_char_list_index):
len_right_s += len(c)
show_s += str(c)
if len_right_s > 0:
sys.stdout.write(' ' * len_right_s + Cursor.BACK(len_right_s))
after_show_s = Cursor.BACK(len_right_s)
sys.stdout.write(show_s + after_show_s)
s.insert_into_char_list(s_char_list_index, c_str)
s_char_list_index += len(c_str.char_list)
cursor_index += len(c_str)
logger.debug(f's_char_list_index={s_char_list_index}, cursor_index={cursor_index}, s={s}')
sys.stdout.flush()
from terminal_layout import KeyListener, Key
import logging
from terminal_layout.types import String
logger = logging.getLogger(__name__)
logger.setLevel(level=logging.DEBUG)
handler = logging.FileHandler("terminal_layout.log")
logger.addHandler(handler)
from terminal_layout.ansi import Cursor
k = KeyListener()
s = String('')
s_char_list_index = 0
cursor_index = 0
@k.bind_key('any')
def _(_, e):
global s, s_char_list_index, cursor_index
c = e.key.code
logger.debug(f's={s}, c={c}')
if repr(c).startswith("'\\x") or repr(c).startswith("u'\\x"):
if e.key == Key.LEFT and s_char_list_index > 0:
back = len(s.char_list[s_char_list_index - 1])