update connect.py

pull/26/head
kelianchun_miller 9 years ago
parent a143797ac9
commit 55503c89e9

@ -107,21 +107,13 @@ class Tty(object):
result_command = '' #最后的结果 result_command = '' #最后的结果
backspace_num = 0 #光标移动的个数 backspace_num = 0 #光标移动的个数
backspace_list = []
reach_backspace_flag = False #没有检测到光标键则为true reach_backspace_flag = False #没有检测到光标键则为true
reach_backspace_second_flag = False
pattern_list = []
pattern_str='' pattern_str=''
while str_r: while str_r:
tmp = re.match(r'\s*\w+\s*', str_r) #获取字符串,其它特殊字符匹配暂时还不知道。。 tmp = re.match(r'\s*\w+\s*', str_r)
if tmp: if tmp:
if reach_backspace_flag : if reach_backspace_flag :
if not reach_backspace_second_flag: pattern_str +=str(tmp.group(0))
pattern_str +=str(tmp.group(0))
else:
pattern_list.append(pattern_str)
pattern_str=str(tmp.group(0))
reach_backspace_second_flag=False
str_r = str_r[len(str(tmp.group(0))):] str_r = str_r[len(str(tmp.group(0))):]
continue continue
else: else:
@ -129,65 +121,46 @@ class Tty(object):
str_r = str_r[len(str(tmp.group(0))):] str_r = str_r[len(str(tmp.group(0))):]
continue continue
tmp = re.match(r'\x1b\[K[\x08]*', str_r) #遇到删除确认符,确定删除数据 tmp = re.match(r'\x1b\[K[\x08]*', str_r)
if tmp: if tmp:
for x in backspace_list:
backspace_num += int(x)
if backspace_num > 0: if backspace_num > 0:
if backspace_num > len(result_command) : if backspace_num > len(result_command) :
result_command += ''.join(pattern_list)
result_command += pattern_str result_command += pattern_str
result_command = result_command[0:-backspace_num] result_command = result_command[0:-backspace_num]
else: else:
result_command = result_command[0:-backspace_num] result_command = result_command[0:-backspace_num]
result_command += ''.join(pattern_list)
result_command += pattern_str result_command += pattern_str
del_len = len(str(tmp.group(0)))-3 del_len = len(str(tmp.group(0)))-3
if del_len > 0: if del_len > 0:
result_command = result_command[0:-del_len] result_command = result_command[0:-del_len]
reach_backspace_flag = False reach_backspace_flag = False
reach_backspace_second_flag =False
backspace_num =0 backspace_num =0
del pattern_list[:]
del backspace_list[:]
pattern_str='' pattern_str=''
str_r = str_r[len(str(tmp.group(0))):] str_r = str_r[len(str(tmp.group(0))):]
continue continue
tmp = re.match(r'\x08+', str_r) #将遇到的退格数字存放到队列中 tmp = re.match(r'\x08+', str_r)
if tmp: if tmp:
if reach_backspace_flag:
reach_backspace_second_flag = True
else:
reach_backspace_flag = True
str_r = str_r[len(str(tmp.group(0))):] str_r = str_r[len(str(tmp.group(0))):]
if len(str_r) != 0: #如果退格键在最后,则放弃 if len(str_r) != 0:
backspace_list.append(len(str(tmp.group(0)))) if reach_backspace_flag:
continue result_command = result_command[0:-backspace_num] + pattern_str
pattern_str = ''
else:
reach_backspace_flag = True
backspace_num = len(str(tmp.group(0)))
continue
else:
break
if reach_backspace_flag : if reach_backspace_flag :
if not reach_backspace_second_flag: pattern_str +=str_r[0]
pattern_str +=str_r[0]
else:
pattern_list.append(pattern_str)
pattern_str=str_r[0]
reach_backspace_second_flag=False
else : else :
result_command += str_r[0] result_command += str_r[0]
str_r = str_r[1:] str_r = str_r[1:]
if pattern_str !='': if backspace_num > 0 :
pattern_list.append(pattern_str) result_command = result_command[0:-backspace_num] + pattern_str
#退格队列中还有腿哥键,则进行删除操作
if len(backspace_list) > 0 :
for backspace in backspace_list:
if int(backspace) >= len(result_command):
result_command = pattern_list[0]
else:
result_command = result_command[:-int(backspace)]
result_command += pattern_list[0]
pattern_list = pattern_list[1:]
control_char = re.compile(r""" control_char = re.compile(r"""
\x1b[ #%()*+\-.\/]. | \x1b[ #%()*+\-.\/]. |
@ -200,21 +173,14 @@ class Tty(object):
""", re.X) """, re.X)
result_command = control_char.sub('', result_command.strip()) result_command = control_char.sub('', result_command.strip())
global VIM_FLAG global VIM_FLAG
global VIM_COMMAND
if not VIM_FLAG: if not VIM_FLAG:
if result_command.startswith('vi'): if result_command.startswith('vi'):
VIM_FLAG = True VIM_FLAG = True
VIM_COMMAND = result_command
return result_command.decode('utf8',"ignore") return result_command.decode('utf8',"ignore")
else: else:
if check_vim_status(VIM_COMMAND, ssh): if result_command.endswith(':wq') or result_command.endswith(':wq!') or result_command.endswith(':q!'):
VIM_FLAG = False VIM_FLAG = False
VIM_COMMAND='' return ''
if result_command.endswith(':wq') or result_command.endswith(':wq!') or result_command.endswith(':q!'):
return ''
return result_command.decode('utf8',"ignore")
else:
return ''
@staticmethod @staticmethod
def remove_control_char(str_r): def remove_control_char(str_r):

Loading…
Cancel
Save