dotbot/plugins/shell.py

import os, subprocess, dotbot

class Shell(dotbot.Plugin):
    '''
    Run arbitrary shell commands.
    '''

    _directive = 'shell'

    def can_handle(self, directive):
        return directive == self._directive

    def handle(self, directive, data):
        if directive != self._directive:
            raise ValueError('Shell cannot handle directive %s' %
                directive)
        return self._process_commands(data)

    def _process_commands(self, data):
        success = True
        defaults = self._context.defaults().get('shell', {})
        with open(os.devnull, 'w') as devnull:
            for item in data:
                stdin = stdout = stderr = devnull
                if defaults.get('stdin', False) == True:
                    stdin = None
                if defaults.get('stdout', False) == True:
                    stdout = None
                if defaults.get('stderr', False) == True:
                    stderr = None
                if isinstance(item, dict):
                    cmd = item['command']
                    msg = item.get('description', None)
                    if 'stdin' in item:
                        stdin = None if item['stdin'] == True else devnull
                    if 'stdout' in item:
                        stdout = None if item['stdout'] == True else devnull
                    if 'stderr' in item:
                        stderr = None if item['stderr'] == True else devnull
                elif isinstance(item, list):
                    cmd = item[0]
                    msg = item[1] if len(item) > 1 else None
                else:
                    cmd = item
                    msg = None
                if msg is None:
                    self._log.lowinfo(cmd)
                else:
                    self._log.lowinfo('%s [%s]' % (msg, cmd))
                executable = os.environ.get('SHELL')
                ret = subprocess.call(cmd, shell=True, stdin=stdin, stdout=stdout,
                    stderr=stderr, cwd=self._context.base_directory(),
                    executable=executable)
                if ret != 0:
                    success = False
                    self._log.warning('Command [%s] failed' % cmd)
        if success:
            self._log.info('All commands have been executed')
        else:
            self._log.error('Some commands were not successfully executed')
        return success