57 lines
1.7 KiB
Python
57 lines
1.7 KiB
Python
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
|
|
defaults = self._context.defaults().get('create', {})
|
|
for key in paths:
|
|
path = os.path.expandvars(os.path.expanduser(key))
|
|
mode = defaults.get('mode', 0o777) # same as the default for os.makedirs
|
|
if isinstance(paths, dict):
|
|
options = paths[key]
|
|
if options:
|
|
mode = options.get('mode', mode)
|
|
success &= self._create(path, mode)
|
|
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, mode):
|
|
success = True
|
|
if not self._exists(path):
|
|
self._log.debug('Trying to create path %s with mode %o' % (path, mode))
|
|
try:
|
|
self._log.lowinfo('Creating path %s' % path)
|
|
os.makedirs(path, mode)
|
|
except OSError:
|
|
self._log.warning('Failed to create path %s' % path)
|
|
success = False
|
|
else:
|
|
self._log.lowinfo('Path exists %s' % path)
|
|
return success
|