使用python模擬命令行終端的示例
更新時間：2019年08月13日 16:18:48 作者：halazi100
今天小編就為大家分享一篇使用python模擬命令行終端的示例，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
可以對？顯示幫助信息，需要立即獲取輸入的字符，因此需要用到termios模塊
另外需要對tab鍵做處理，當按下tab鍵時可以進行自動補全
#! /usr/bin/env python
# coding=utf-8
import os
import sys
import tty
import termios
'''
Enter: 13
Back: 127
?: 63
C-h: 8
C-w: 23
Tab: 9
C-u: 21
C-c: 3
C-d: 4
C-\: 28
SPACE: 32
'''
CLI_KEY_CNCR = 13
CLI_KEY_BACK = 127
CLI_KEY_QMARK = 63
CLI_KEY_CTRLH = 8
CLI_KEY_CTRLW = 23
CLI_KEY_TAB = 9
CLI_KEY_CTRLU = 21
CLI_KEY_CTRLC = 3
CLI_KEY_CTRLD = 4
CLI_KEY_QUIT = 28
CLI_KEY_SPACE = 32
CLI_KEY_TABLEN = 4
class CLI(object):
def __init__(self):
self.line = ''
self.line_complete = ''
self.completer_on = False
self.completer_dict = {}
self.completer_dict_keys = self.completer_dict.keys()
self.completer_id = 0
self.completer_cnt = len(self.completer_dict)
def getch(self):
fd = sys.stdin.fileno()
old_settings = termios.tcgetattr(fd)
try:
tty.setraw(fd)
ch = sys.stdin.read(1)
finally:
termios.tcsetattr(fd, termios.TCSADRAIN, old_settings)
return ch
def completer_kw_update(self):
self.completer_dict_keys = self.completer_dict.keys()
self.completer_dict_keys.sort()
self.completer_cnt = len(self.completer_dict)
def completer_kw_add(self, key, word):
self.completer_dict[key] = word
self.completer_kw_update()
def completer_kw_clear(self):
self.completer_dict.clear()
self.completer_id_update()
def completer_id_update(self):
self.completer_kw_update()
if self.completer_id < self.completer_cnt - 1:
self.completer_id += 1
else:
self.completer_id = 0
def completer_wd_select(self, word):
if not word:
return ''
cnt = self.completer_cnt
while cnt>0:
completer = self.completer_dict_keys[self.completer_id]
self.completer_id_update()
cnt -= 1
if word == completer[:len(word)]:
return completer[len(word):]
return ''
def printf(self, info=''):
sys.stdout.write(info)
def show_spec_len_str(self, info, maxlen, spacech=' '):
'display a string of the specified length'
maxlen = maxlen
infolen = len(info)
if maxlen < infolen:
maxlen = infolen
while infolen>0:
ch = info[-infolen]
for i in range(self.char_memory_len(ch)):
self.printf(info[i-infolen])
infolen -= self.char_memory_len(ch)
maxlen -= self.char_display_len(ch)
while maxlen>0:
self.printf(spacech)
maxlen -= self.char_display_len(spacech)
def show_help_info(self):
if self.completer_on:
line = self.line_complete
else:
line = self.line
lastwd = ''
show_all = False
if not line or line[-1] == ' ':
show_all = True
else:
lastwd = line.split()[-1]
if self.completer_dict:
maxlen = max([len(info) for info in self.completer_dict])
else:
maxlen = 12
for info in self.completer_dict:
if show_all or lastwd == info[:len(lastwd)]:
self.printf(' ')
self.show_spec_len_str(info, maxlen)
self.printf(' ')
self.printf(self.completer_dict[info])
self.printf('\r\n')
def is_chinese_char(self, ch):
return ord(ch) > 127
def char_display_len(self, ch):
if self.is_chinese_char(ch):
return 2
elif ord(ch) == CLI_KEY_TAB:
return CLI_KEY_TABLEN
else:
return 1
def char_memory_len(self, ch):
if self.is_chinese_char(ch):
return 3
else:
return 1
def rm_last_char(self, line):
lastch = ''
rmlen = 0
if len(line)>0:
lastch = line[-1]
self.printf('\b \b' * self.char_display_len(lastch))
rmlen = self.char_memory_len(lastch)
if len(line) >= rmlen:
line = line[:-(rmlen)]
else:
rmlen = len(line)
line = ''
return rmlen, line
def rm_last_word(self, line):
lastwd = ''
linelen = len(line)
rspacelen = linelen - len(line.rstrip())
if not linelen:
return line
lastwd = line.split()[-1]
backlen = len(lastwd) + rspacelen
rmlen = 0
while backlen>0 and line:
rmlen, line = self.rm_last_char(line)
backlen -= rmlen
return line
def rm_one_line(self, line):
rmlen = 0
while line:
rmlen, line = self.rm_last_char(line)
return line
def do_line_complete_proc(self):
line = self.line
line_complete = self.line_complete
lastwd = ''
self.printf('\r\n')
if self.line_complete:
self.printf(self.line_complete)
else:
self.printf(self.line)
if not line:
return line
lastwd = line.split()[-1]
completer = self.completer_wd_select(lastwd)
if not completer.strip():
self.line_complete = line
return line
backlen = len(line_complete) - len(line)
while backlen>0 and line_complete:
rmlen, line_complete = self.rm_last_char(line_complete)
backlen -= rmlen
self.printf(completer)
line_complete = line + completer
self.line_complete = line_complete
def do_line_complete_end(self):
if self.completer_on:
self.line = self.line_complete
self.line_complete = ''
self.completer_on = False
def get_line(self):
self.line = ''
self.line_complete = ''
self.completer_on = False
while True:
ch = self.getch()
if ch == '\r' or ch == '\n':
self.do_line_complete_end()
self.printf('\r\n')
break
elif ord(ch) == CLI_KEY_BACK or ord(ch) == CLI_KEY_CTRLH:
if self.completer_on:
rmlen, self.line_complete = self.rm_last_char(self.line_complete)
else:
rmlen, self.line = self.rm_last_char(self.line)
self.do_line_complete_end()
elif ord(ch) == CLI_KEY_QMARK:
self.printf('?')
self.printf('\r\n')
self.show_help_info()
if self.completer_on:
self.printf(self.line_complete)
else:
self.printf(self.line)
elif ord(ch) == CLI_KEY_CTRLW:
if self.completer_on:
self.line_complete = self.rm_last_word(self.line_complete)
else:
self.line = self.rm_last_word(self.line)
self.do_line_complete_end()
elif ord(ch) == CLI_KEY_TAB:
self.completer_on = True
self.do_line_complete_proc()
elif ord(ch) == CLI_KEY_CTRLD:
if self.line:
return self.line
else:
return ch
elif ord(ch) == CLI_KEY_QUIT:
self.printf('\r\n Interrupted by <Ctrl-\>.\r\n')
sys.exit()
elif ord(ch) == CLI_KEY_CTRLU:
if self.completer_on:
self.line_complete = self.rm_one_line(self.line_complete)
else:
self.line = self.rm_one_line(self.line)
self.do_line_complete_end()
elif ord(ch) == CLI_KEY_SPACE:
self.printf(ch)
if self.completer_on:
self.line_complete += ch
else:
self.line += ch
else:
self.printf(ch)
self.do_line_complete_end()
self.line += ch
# chinese qmask proc
if ord(ch) == 159 and len(self.line)>= 3 and self.line[-3:] == '\xef\xbc\x9f':
self.printf('\r\n')
self.line = self.line[:-3]
self.show_help_info()
self.printf(self.line)
return self.line
def get_raw_line(self):
self.raw_line = ''
while True:
ch = self.getch()
if ch == '\r' or ch == '\n':
self.printf('\r\n')
break
elif ord(ch) == CLI_KEY_BACK or ord(ch) == CLI_KEY_CTRLH:
rmlen, self.raw_line = self.rm_last_char(self.raw_line)
elif ord(ch) == CLI_KEY_CTRLW:
self.raw_line = self.rm_last_word(self.raw_line)
elif ord(ch) == CLI_KEY_TAB:
self.printf(' ' * self.char_display_len(ch))
self.raw_line += ch
elif ord(ch) == CLI_KEY_CTRLD:
if self.raw_line:
return self.raw_line
else:
return ch
elif ord(ch) == CLI_KEY_QUIT:
self.printf('\r\n Interrupted by <Ctrl-\>.\r\n')
sys.exit()
elif ord(ch) == CLI_KEY_CTRLU:
self.raw_line = self.rm_one_line(self.raw_line)
else:
self.raw_line += ch
self.printf(ch)
return self.raw_line
def test():
cli = CLI()
help_info = {
'hello0': 'say hello 0',
'hello1': 'say hello 1',
'hellohello': 'say hello hello',
'hellohehe': 'say hello hehe',
'hellohi': 'say hello hi',
'你好啊': 'say 你好啊',
'你好嗎': 'say 你好嗎',
'你好哈': 'say 你好哈',
}
for key in help_info:
cli.completer_kw_add(key, help_info[key])
while True:
line = cli.get_line()
if len(line) == 1 and ord(line[0]) == CLI_KEY_CTRLD:
break
if line == 'quit':
break
print(line)
if __name__ == "__main__":
test()
以上這篇使用python模擬命令行終端的示例就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關文章
使用Python內(nèi)置的模塊與函數(shù)進行不同進制的數(shù)的轉換
這篇文章主要介紹了使用Python內(nèi)置的模塊與函數(shù)進行不同進制的數(shù)的轉換的方法,Python也使得讀取純二進制文件內(nèi)容非常方便,需要的朋友可以參考下2016-03-03
在大數(shù)據(jù)分析的領域中,Google BigQuery 是一個被廣泛使用的云端數(shù)據(jù)倉庫解決方案,它由 Google Cloud 提供,并且專為處理大規(guī)模數(shù)據(jù)集、進行快速的數(shù)據(jù)分析和復雜的查詢而設計,本文給大家講解了如何使用Python與BigQuery進行交互,需要的朋友可以參考下2025-04-04
子域名枚舉是為一個或多個域查找子域的過程，它是信息收集階段的重要組成部分，這篇文章主要介紹了使用python實現(xiàn)子域名探測,需要的朋友可以參考下2022-07-07
Pytorch使用DataLoader實現(xiàn)批量加載數(shù)據(jù)
這篇文章主要介紹了Pytorch使用DataLoader實現(xiàn)批量加載數(shù)據(jù)方式,具有很好的參考價值,希望對大家有所幫助,如有錯誤或未考慮完全的地方,望不吝賜教2024-02-02
這篇文章主要介紹了Python 操作文件的基本方法總結的相關資料,這里說明了九種操作文件的方法，并附上實現(xiàn)方法，需要的朋友可以參考下2017-08-08
python實現(xiàn)人臉識別經(jīng)典算法（一） 特征臉法
這篇文章主要為大家詳細介紹了python實現(xiàn)人臉識別經(jīng)典算法，特征臉法，文中示例代碼介紹的非常詳細，具有一定的參考價值，感興趣的小伙伴們可以參考一下2018-03-03
python小程序基于Jupyter實現(xiàn)天氣查詢的方法
這篇文章主要介紹了python小程序基于Jupyter實現(xiàn)天氣查詢的方法，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧2020-03-03
這篇文章主要介紹了python 裝飾器的基本使用，幫助大家更好的理解和使用python，感興趣的朋友可以了解下2021-01-01
這篇文章主要介紹了Python?pluggy框架基礎用法總結,本文通過三個例子結合代碼給大家介紹的非常詳細，對大家的學習或工作具有一定的參考借鑒價值，需要的朋友可以參考下2022-05-05
PyTorch手寫數(shù)字數(shù)據(jù)集進行多分類
這篇文章主要介紹了PyTorch手寫數(shù)字數(shù)據(jù)集進行多分類，損失函數(shù)采用交叉熵，激活函數(shù)采用ReLU，優(yōu)化器采用帶有動量的mini-batchSGD算法,需要的朋友可以參考一下2022-03-03