python 2.7 / exec / 有什么问题？

Question

我的代码在 Python 2.5 中运行良好，但在 2.7 中运行不佳：

import sys
import traceback
try:
    from io import StringIO
except:
    from StringIO import StringIO

def CaptureExec(stmt):
    oldio = (sys.stdin, sys.stdout, sys.stderr)
    sio = StringIO()
    sys.stdout = sys.stderr = sio
    try:
        exec(stmt, globals(), globals())
        out = sio.getvalue()
    except Exception, e:
        out = str(e) + "\n" + traceback.format_exc()
    sys.stdin, sys.stdout, sys.stderr = oldio
    return out

print "%s" % CaptureExec("""
import random
print "hello world"
""")

我得到：

string argument expected, got 'str'
Traceback (most recent call last):
  File "D:\3.py", line 13, in CaptureExec
    exec(stmt, globals(), globals())
  File "", line 3, in 
TypeError: string argument expected, got 'str'

Answer 1

io.StringIO 在 Python 2.7 中很令人困惑，因为它是从 3.x 字节/字符串世界向后移植的。此代码得到与您的错误相同的错误：

from io import StringIO
sio = StringIO()
sio.write("Hello\n")

原因：

Traceback (most recent call last):
  File "so2.py", line 3, in <module>
    sio.write("Hello\n")
TypeError: string argument expected, got 'str'

如果您仅使用 Python 2.x，则完全跳过 io 模块，并坚持使用 StringIO。如果您确实想使用 io，请将导入更改为：

from io import BytesIO as StringIO

Answer 2

这是个坏消息

io.StringIO 想要使用 unicode。您可能认为可以通过在要打印的字符串前面放置一个 u 来修复它，如下所示

print "%s" % CaptureExec("""
import random
print u"hello world"
""")

，但是 print 确实会损坏此问题，因为它会导致 2 次写入字符串IO。第一个是 u"hello world" ，这很好，但后面跟着 "\n"

所以你需要写这样的东西

print "%s" % CaptureExec("""
import random
sys.stdout.write(u"hello world\n")
""")