マネジメントコマンドをview関数内から実行する

# -*- coding: utf-8 -*-

"""
management_command_launcher.py
マネジメントコマンドをview関数から実行
クラスにした方がよかったか？
マネジメントコマンド内で、出力を self.stdout.write() で行ってないと
正しく結果を得られない。
(print() などされると拾えない)
"""

import StringIO
from django.core.management.base import CommandError
from django.utils.encoding import smart_str

def launch_management_command(command_module, *args, **options):
    """
    @args
        (module) command_module:
            Command クラスを含むモジュール名。
            例:
                from appname.management.commands import hoge_command
                result = launch_management_command(hoge_command, ... )
    """
    log_buffer = StringIO.StringIO()
    command_instance = command_module.Command()
    command_instance.stdout=log_buffer
    command_instance.stderr=log_buffer
    try:
        output = command_instance.handle(*args, **options)
        if output:
            log_buffer.write(smart_str(output))
    except CommandError, e:
        log_buffer.write(smart_str(e))
    log_buffer.seek(0)
    return log_buffer.read()