dotbot/dotbot/plugins/create.py

import os
import dotbot


class Create(dotbot.Plugin):
    """
    Create empty paths.
    """

    _directive = "create"

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

    def handle(self, directive, data):
        if directive != self._directive:
            raise ValueError("Create cannot handle directive %s" % directive)
        return self._process_paths(data)

    def _process_paths(self, paths):
        success = True
        for path in paths:
            path = os.path.normpath(os.path.expandvars(os.path.expanduser(path)))
            success &= self._create(path)
        if success:
            self._log.info("All paths have been set up")
        else:
            self._log.error("Some paths were not successfully set up")
        return success

    def _exists(self, path):
        """
        Returns true if the path exists.
        """
        path = os.path.expanduser(path)
        return os.path.exists(path)

    def _create(self, path):
        success = True
        if not self._exists(path):
            self._log.debug("Trying to create path %s" % path)
            try:
                self._log.lowinfo("Creating path %s" % path)
                os.makedirs(path)
            except OSError:
                self._log.warning("Failed to create path %s" % path)
                success = False
        else:
            self._log.lowinfo("Path exists %s" % path)
        return success