Add 'create' directive to create directories

3 files changed, 55 insertions, 1 deletions

diff --git a/dotbot/cli.py b/dotbot/cli.py
index fdc2a13..2680acf 100644
--- a/dotbot/cli.py
+++ b/dotbot/cli.py
@@ -56,7 +56,7 @@ def main():
             log.use_color(False)
         plugin_directories = list(options.plugin_dirs)
         if not options.disable_built_in_plugins:
-            from .plugins import Clean, Link, Shell
+            from .plugins import Clean, Create, Link, Shell
         plugin_paths = []
         for directory in plugin_directories:
           for plugin_path in glob.glob(os.path.join(directory, '*.py')):
diff --git a/dotbot/plugins/__init__.py b/dotbot/plugins/__init__.py
index 93bd981..f75bef5 100644
--- a/dotbot/plugins/__init__.py
+++ b/dotbot/plugins/__init__.py
@@ -1,3 +1,4 @@
 from .clean import Clean
+from .create import Create
 from .link import Link
 from .shell import Shell
diff --git a/dotbot/plugins/create.py b/dotbot/plugins/create.py
new file mode 100644
index 0000000..dc119da
--- /dev/null
+++ b/dotbot/plugins/create.py
@@ -0,0 +1,53 @@
+import os
+import glob
+import shutil
+import dotbot
+import subprocess
+
+
+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.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