python - 如何将 django call_command 的输出保存到变量或文件中

Question

我在 Django 中从类似于以下的脚本中调用命令:

#!/usr/bin/python
from django.core.management import call_command
call_command('syncdb')
call_command('runserver')
call_command('inspectdb')

如何将例如 call_command('inspectdb') 的输出分配给变量或文件？

我试过了

var = call_command('inspectdb')

但是 'var' 仍然没有:目的:检查不是由 django 创建的遗留数据库中的现有表

Answer 1

您必须重定向 call_command 的输出，否则它只会打印到标准输出但不返回任何内容。您可以尝试将其保存到文件中，然后像这样读取它:

with open('/tmp/inspectdb', 'w+') as f:
    call_command('inspectdb', stdout=f)
    var = f.readlines()

编辑:几年后再看这个，更好的解决方案是创建一个 StringIO 来重定向输出，而不是一个真实的文件。这是来自 Django's test suites 之一的示例:

from io import StringIO

def test_command(self):
    out = StringIO()
    management.call_command('dance', stdout=out)
    self.assertIn("I don't feel like dancing Rock'n'Roll.\n", out.getvalue())