如何将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‘仍然为none:目的:检查不是由django创建的遗留数据库中的现有表

Answer 1

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

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())

Answer 2

这在“从代码运行管理命令>输出重定向”下的中有说明。

要保存到变量，您可以执行以下操作：

import io
from django.core.management import call_command


with io.StringIO() as out:
   call_command('dumpdata', stdout=out)
   print(out.getvalue())

要保存到文件，您可以执行以下操作：

from django.core.management import call_command


with open('/path/to/command_output', 'w') as f:
    call_command('dumpdata', stdout=f)