如何从 python 子例程中捕获 stdio？

发布于 2024-12-17 10:47:43 字数 248 浏览 0 评论 0原文

我有以下代码：

def fsub():
  print 'OK'

def fmain():
  a = fsub()

fmain()

显然 fsub() 不会返回“OK”并分配给 fmain() 中的 a 。然而，这就是我想要的。无论如何，我们是否可以不改变fsub()？

原文

I have the following codes:

def fsub():
  print 'OK'

def fmain():
  a = fsub()

fmain()

Apparently fsub() won't return 'OK' and assign to a in fmain(). However, this is what I want. Is there anyway we can make it without changing fsub()?

分享到QQ

分享到微博

如果你对这篇内容有疑问，欢迎到本站社区发帖提问参与讨论，获取更多帮助，或者扫码二维码加入 Web 技术交流群。

发布评论

需要登录才能够评论，你可以免费注册一个本站的账号。

塔塔猫 2024-12-24 10:47:43

当您执行 a = fsub() 时，您正在尝试将 a 返回到 fsub()，在本例中是 无（因为fsub() 不返回任何内容）。

正确的做法是将 stdout 重定向到文件，然后调用 fsub() 函数，并将 stdout 重定向回原始 stdout：

import sys

def fmain():
    sys.stdout = open('output','a')
    fsub()
    sys.stdout = sys.__stdout__
    print 'Output of fsub():'
    print open('output').read(),
    # added the coma (,) to avoid a new line

结果：

>>> fmain()
Output of fsub():
OK

When you do a = fsub(), you're trying to assing a the return of fsub(), in this case None (because fsub() doesn't return anything).

The correct thing to do is to redirect the stdout to a file, then call the fsub() function, and the redirect back the stdout to the original stdout:

import sys

def fmain():
    sys.stdout = open('output','a')
    fsub()
    sys.stdout = sys.__stdout__
    print 'Output of fsub():'
    print open('output').read(),
    # added the coma (,) to avoid a new line

Result: