python - Django manage.py 自定义命令可以返回值吗？如何，或者为什么不？

Question

按照文档： https ://docs.djangoproject.com/en/dev/howto/custom-management-commands/

我创建了自己的自定义命令（称为其他命令，但示例如下所示）：

from django.core.management.base import BaseCommand, CommandError
from polls.models import Poll

class Command(BaseCommand):
    args = '<poll_id poll_id ...>'
    help = 'Closes the specified poll for voting'

    def handle(self, *args, **options):
        for poll_id in args:
            try:
                poll = Poll.objects.get(pk=int(poll_id))
            except Poll.DoesNotExist:
                raise CommandError('Poll "%s" does not exist' % poll_id)

            poll.opened = False
            poll.save()

            self.stdout.write('Successfully closed poll "%s"' % poll_id)

        return "Yay"

问题是为什么返回像“Yay”这样的字符串不起作用？我做错了还是不可能？

当我从视图中调用自定义命令时，我会执行以下操作：

     value = call_command('call_custom_command', parameter)
     print value

但该值显示为无。

Answer 1

如果要获取 的输出call_command()，则需要捕获 stdout。以下是您的操作方法：

out = StringIO()
call_command('call_custom_command', stdout=out)

value = out.getvalue()
print value

这种技术实际上在django 测试中用于测试管理命令。

演示：

>>> from django.core.management import call_command
>>> from StringIO import StringIO
>>> out = StringIO()
>>> call_command('validate', stdout=out)
>>> out.getvalue()
'0 errors found\n'

希望有帮助。