黑帽联盟

标题: python raw_input中使用tab键补全命令 [打印本页]

---

作者: 定位    时间: 2018-2-8 16:11
标题: python raw_input中使用tab键补全命令
用 readline, 以下是一个简单的例子

1. import readline
   
2. 
   
3. CMD = ['foo1', 'foo2', 'bar1', 'bar2', 'exit']
   
4. 
   
5. def completer(text, state):
   
6.     options = [cmd for cmd in CMD if cmd.startswith(text)]
   
7.     if state < len(options):
   
8.         return options[state]
   
9.     else:
   
10.         return None
    
11. 
    
12. readline.parse_and_bind("tab: complete")
    
13. readline.set_completer(completer)
    
14. 
    
15. while True:
    
16.     cmd = raw_input('==> ')
    
17.     if cmd=='exit':
    
18.         break
    
19.     print(cmd)

复制代码

测试：

1. ==> <TAB><TAB>
   
2. bar1  bar2  exit  foo1  foo2
   
3. ==> b<TAB>
   
4. ==> bar
   
5. ==> bar<TAB><TAB>
   
6. bar1  bar2  
   
7. ==> bar1
   
8. bar1
   
9. ==> exit

复制代码

---

欢迎光临 黑帽联盟 (https://bbs.cnblackhat.com/)

Powered by Discuz! X2.5