在cmd.Cmd中完成时,sys.stdout已被替换
目前,我有一个使用cmd.Cmd模块作为命令行界面的应用程序,tab-completion非常完美。
现在,我想用另一个对象替换sys.stdout
(例如,为了捕获正在写入的内容)。
以下片段理论上应该是完全透明的,因为对Std对象的每个get / set操作都会重定向到sys.__stdout__
。
class Std(object):
def __getattribute__(self, name):
if name in ('__getattribute__', '__setattr__'):
return object.__getattribute__(self, name)
else:
return getattr(sys.__stdout__, name)
def __setattr__(self, name, value):
setattr(sys.__stdout__, name, value)
sys.stdout = Std()
例如, sys.stdout.fileno()
仍将打印1.但是, Cmd.cmd
的readline制表符完成现在不再有效......
好的,我们从文件继承。 (stdout是一个文件对象。)
class Std(file):
def __init__(self):
pass
def __getattribute__(self, name):
if name in ('__getattribute__', '__setattr__'):
return object.__getattribute__(self, name)
else:
return getattr(sys.__stdout__, name)
def __setattr__(self, name, value):
setattr(sys.__stdout__, name, value)
sys.stdout = Std()
现在我得到:
Traceback (most recent call last):
File "./shell.py", line 61, in <module>
print '#1'
ValueError: I/O operation on closed file
但是,下面的断言不会失败:
assert not sys.stdout.closed
不知何故,我猜,Python会过度优化某些东西,并绕过Std.write。
我应该如何替换stdout,而不会丢失readline支持...?
乔纳森
-编辑-
我也试图替换sys.stdin。 将它传递给cmd.Cmd不起作用,因为raw_input用于readline支持,而Python的raw_input不接受文件描述符。 它回退到分配给sys.stdin的任何pty。
当我创建一个新的pty(通过os.openpty()),并将这对分配给sys.stdin / out时,通过该pty的readline自动完成功能完美地工作,但是再次,当包装在代理对象中时,它工作,但没有自动完成。
试图了解readline的来源,但这并不容易:http://svn.python.org/projects/python/branches/release25-maint/Modules/readline.c
我不知道为什么替换sys.stdout
不起作用,但无论如何,通过将自己的文件对象传递给cmd.Cmd
的构造函数,可以解决您的即时问题。
这是一个示例脚本(部分来自PyMOTW),它演示了tab-completion:
import sys, cmd
class Std(object):
def __getattribute__(self, name):
if name in ('__getattribute__', '__setattr__'):
return object.__getattribute__(self, name)
else:
return getattr(sys.__stdout__, name)
def __setattr__(self, name, value):
setattr(sys.__stdout__, name, value)
class HelloWorld(cmd.Cmd):
FRIENDS = [ 'Alice', 'Adam', 'Barbara', 'Bob' ]
def do_greet(self, person):
"Greet the person"
if person and person in self.FRIENDS:
greeting = 'hi, %s!' % person
elif person:
greeting = "hello, " + person
else:
greeting = 'hello'
print greeting
def complete_greet(self, text, line, begidx, endidx):
if not text:
completions = self.FRIENDS[:]
else:
completions = [f for f in self.FRIENDS
if f.startswith(text)]
return completions
def do_EOF(self, line):
return True
if __name__ == '__main__':
HelloWorld(stdout=Std()).cmdloop()
出于您的目的,这是一种更好的做法,因为它可以确保您仅捕获Cmd
实例生成的输出。
请注意,您传入的文件对象也可用作Cmd
实例的stdout
属性。
上一篇: completion in cmd.Cmd, when sys.stdout has been replaced