nio Block for setting status

status_block.py

from nio.block.base import Block
from nio.command import command


@command('set_warning')
@command('set_error')
@command('set_ok')
class StatusBlock(Block):

    def set_warning(self):
        return {'status': self.set_status('warn')}

    def set_ok(self):
        return {'status': self.set_status('ok')}

    def set_error(self):
        return {'status': self.set_status('error')}