Python ASTEVAL。 我如何捕獲標准輸出

Question

我考慮將asteval python軟件包用於我的個人Web應用程序。

使用Python的ast模塊，ASTEVAL是Python表達式和語句的安全（ish）評估器。 想法是提供一種可以處理用戶輸入的簡單，安全和健壯的微型數學語言。

我面臨的問題是我無法獲得asteval標准。 我嘗試使用以下代碼段捕獲它：

from asteval import Interpreter

aeval = Interpreter()

from cStringIO import StringIO
import sys

class Capturing(list):
    def __enter__(self):
        self._stdout = sys.stdout
        sys.stdout = self._stringio = StringIO()
        return self
    def __exit__(self, *args):
        self.extend(self._stringio.getvalue().splitlines())
        sys.stdout = self._stdout

接着：

with Capturing() as output:
    aeval('print "this should be captured"')

但沒有運氣， output是一個空列表。

Answer 1

您可以將文件對象（ writer ）傳遞給Interpreter()類：

output = StringIO()
aeval = Interpreter(writer=output)

如果未指定， writer默認為sys.stdout ，並在實例化Interpreter()時設置。 這就是為什么替換sys.stdout不起作用的原因。 該實例已經有自己的引用。

演示：

>>> from cStringIO import StringIO
>>> from asteval import Interpreter
>>> output = StringIO()
>>> aeval = Interpreter(writer=output)
>>> aeval('print "this should be captured"')
>>> output.getvalue()
'this should be captured\n'