Django manage.py 自定义命令可以返回值吗?如何实现,或为什么不可以?

25 投票
1 回答
8798 浏览
提问于 2025-04-17 23:22

根据文档的说明: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

但是显示的值是None。

1 个回答

35

如果你想获取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'

希望这对你有帮助。

撰写回答