遵循文件: 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"
问题是如何回归像#34; Yay&#34;不起作用? 我做错了还是不可能?
当我从我的视图中调用自定义命令时,我会执行以下操作:
value = call_command('call_custom_command', parameter)
print value
但该值显示为无。
答案 0 :(得分:24)
如果您想获得call_command()
的输出,则需要捕获标准输出。以下是如何做到的:
out = StringIO()
call_command('call_custom_command', stdout=out)
value = out.getvalue()
print value
此技术实际上在django tests中用于测试管理命令。
演示:
>>> from django.core.management import call_command
>>> from StringIO import StringIO
>>> out = StringIO()
>>> call_command('validate', stdout=out)
>>> out.getvalue()
'0 errors found\n'
希望有所帮助。