Могут ли пользовательские команды Django manage.py вернуть значение? Как или почему нет?

Следующие документы: 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 ответ

Решение

Если вы хотите получить вывод call_command()Вам нужно захватить стандартный вывод. Вот как вы можете это сделать:

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'

Надеюсь, это поможет.

Другие вопросы по тегам