blacken
This commit is contained in:
parent
fe44654c73
commit
9e89d7f524
|
@ -1,4 +1,4 @@
|
||||||
from .cli import main
|
from .cli import main
|
||||||
from .plugin import Plugin
|
from .plugin import Plugin
|
||||||
|
|
||||||
__version__ = '1.18.0'
|
__version__ = "1.18.0"
|
||||||
|
|
|
@ -11,39 +11,37 @@ from .util import module
|
||||||
import dotbot
|
import dotbot
|
||||||
import yaml
|
import yaml
|
||||||
|
|
||||||
|
|
||||||
def add_options(parser):
|
def add_options(parser):
|
||||||
parser.add_argument('-Q', '--super-quiet', action='store_true',
|
parser.add_argument("-Q", "--super-quiet", action="store_true", help="suppress almost all output")
|
||||||
help='suppress almost all output')
|
parser.add_argument("-q", "--quiet", action="store_true", help="suppress most output")
|
||||||
parser.add_argument('-q', '--quiet', action='store_true',
|
parser.add_argument("-v", "--verbose", action="store_true", help="enable verbose output")
|
||||||
help='suppress most output')
|
parser.add_argument("-d", "--base-directory", help="execute commands from within BASEDIR", metavar="BASEDIR")
|
||||||
parser.add_argument('-v', '--verbose', action='store_true',
|
parser.add_argument("-c", "--config-file", help="run commands given in CONFIGFILE", metavar="CONFIGFILE")
|
||||||
help='enable verbose output')
|
parser.add_argument(
|
||||||
parser.add_argument('-d', '--base-directory',
|
"-p", "--plugin", action="append", dest="plugins", default=[], help="load PLUGIN as a plugin", metavar="PLUGIN"
|
||||||
help='execute commands from within BASEDIR',
|
)
|
||||||
metavar='BASEDIR')
|
parser.add_argument("--disable-built-in-plugins", action="store_true", help="disable built-in plugins")
|
||||||
parser.add_argument('-c', '--config-file',
|
parser.add_argument(
|
||||||
help='run commands given in CONFIGFILE', metavar='CONFIGFILE')
|
"--plugin-dir",
|
||||||
parser.add_argument('-p', '--plugin', action='append', dest='plugins', default=[],
|
action="append",
|
||||||
help='load PLUGIN as a plugin', metavar='PLUGIN')
|
dest="plugin_dirs",
|
||||||
parser.add_argument('--disable-built-in-plugins',
|
default=[],
|
||||||
action='store_true', help='disable built-in plugins')
|
metavar="PLUGIN_DIR",
|
||||||
parser.add_argument('--plugin-dir', action='append', dest='plugin_dirs', default=[],
|
help="load all plugins in PLUGIN_DIR",
|
||||||
metavar='PLUGIN_DIR', help='load all plugins in PLUGIN_DIR')
|
)
|
||||||
parser.add_argument('--only', nargs='+',
|
parser.add_argument("--only", nargs="+", help="only run specified directives", metavar="DIRECTIVE")
|
||||||
help='only run specified directives', metavar='DIRECTIVE')
|
parser.add_argument("--except", nargs="+", dest="skip", help="skip specified directives", metavar="DIRECTIVE")
|
||||||
parser.add_argument('--except', nargs='+', dest='skip',
|
parser.add_argument("--force-color", dest="force_color", action="store_true", help="force color output")
|
||||||
help='skip specified directives', metavar='DIRECTIVE')
|
parser.add_argument("--no-color", dest="no_color", action="store_true", help="disable color output")
|
||||||
parser.add_argument('--force-color', dest='force_color', action='store_true',
|
parser.add_argument("--version", action="store_true", help="show program's version number and exit")
|
||||||
help='force color output')
|
|
||||||
parser.add_argument('--no-color', dest='no_color', action='store_true',
|
|
||||||
help='disable color output')
|
|
||||||
parser.add_argument('--version', action='store_true',
|
|
||||||
help='show program\'s version number and exit')
|
|
||||||
|
|
||||||
def read_config(config_file):
|
def read_config(config_file):
|
||||||
reader = ConfigReader(config_file)
|
reader = ConfigReader(config_file)
|
||||||
return reader.get_config()
|
return reader.get_config()
|
||||||
|
|
||||||
|
|
||||||
def main(additional_args=None):
|
def main(additional_args=None):
|
||||||
log = Messenger()
|
log = Messenger()
|
||||||
try:
|
try:
|
||||||
|
@ -54,7 +52,7 @@ def main(additional_args=None):
|
||||||
print("got explicit argumenets")
|
print("got explicit argumenets")
|
||||||
options = parser.parse_args(additional_args)
|
options = parser.parse_args(additional_args)
|
||||||
if options.version:
|
if options.version:
|
||||||
print('Dotbot version %s (yaml: %s)' % (dotbot.__version__, yaml.__version__))
|
print("Dotbot version %s (yaml: %s)" % (dotbot.__version__, yaml.__version__))
|
||||||
exit(0)
|
exit(0)
|
||||||
if options.super_quiet:
|
if options.super_quiet:
|
||||||
log.set_level(Level.WARNING)
|
log.set_level(Level.WARNING)
|
||||||
|
@ -78,23 +76,23 @@ def main(additional_args=None):
|
||||||
from .plugins import Clean, Create, Link, Shell
|
from .plugins import Clean, Create, Link, Shell
|
||||||
plugin_paths = []
|
plugin_paths = []
|
||||||
for directory in plugin_directories:
|
for directory in plugin_directories:
|
||||||
for plugin_path in glob.glob(os.path.join(directory, '*.py')):
|
for plugin_path in glob.glob(os.path.join(directory, "*.py")):
|
||||||
plugin_paths.append(plugin_path)
|
plugin_paths.append(plugin_path)
|
||||||
for plugin_path in options.plugins:
|
for plugin_path in options.plugins:
|
||||||
plugin_paths.append(plugin_path)
|
plugin_paths.append(plugin_path)
|
||||||
for plugin_path in plugin_paths:
|
for plugin_path in plugin_paths:
|
||||||
abspath = os.path.abspath(plugin_path)
|
abspath = os.path.abspath(plugin_path)
|
||||||
module.load(abspath)
|
module.load(abspath)
|
||||||
if not options.config_file:
|
if not options.config_file:
|
||||||
log.error('No configuration file specified')
|
log.error("No configuration file specified")
|
||||||
exit(1)
|
exit(1)
|
||||||
# read tasks from config file
|
# read tasks from config file
|
||||||
tasks = read_config(options.config_file)
|
tasks = read_config(options.config_file)
|
||||||
if tasks is None:
|
if tasks is None:
|
||||||
log.warning('Configuration file is empty, no work to do')
|
log.warning("Configuration file is empty, no work to do")
|
||||||
tasks = []
|
tasks = []
|
||||||
if not isinstance(tasks, list):
|
if not isinstance(tasks, list):
|
||||||
raise ReadingError('Configuration file must be a list of tasks')
|
raise ReadingError("Configuration file must be a list of tasks")
|
||||||
if options.base_directory:
|
if options.base_directory:
|
||||||
base_directory = os.path.abspath(options.base_directory)
|
base_directory = os.path.abspath(options.base_directory)
|
||||||
else:
|
else:
|
||||||
|
@ -104,12 +102,12 @@ def main(additional_args=None):
|
||||||
dispatcher = Dispatcher(base_directory, only=options.only, skip=options.skip)
|
dispatcher = Dispatcher(base_directory, only=options.only, skip=options.skip)
|
||||||
success = dispatcher.dispatch(tasks)
|
success = dispatcher.dispatch(tasks)
|
||||||
if success:
|
if success:
|
||||||
log.info('\n==> All tasks executed successfully')
|
log.info("\n==> All tasks executed successfully")
|
||||||
else:
|
else:
|
||||||
raise DispatchError('\n==> Some tasks were not executed successfully')
|
raise DispatchError("\n==> Some tasks were not executed successfully")
|
||||||
except (ReadingError, DispatchError) as e:
|
except (ReadingError, DispatchError) as e:
|
||||||
log.error('%s' % e)
|
log.error("%s" % e)
|
||||||
exit(1)
|
exit(1)
|
||||||
except KeyboardInterrupt:
|
except KeyboardInterrupt:
|
||||||
log.error('\n==> Operation aborted')
|
log.error("\n==> Operation aborted")
|
||||||
exit(1)
|
exit(1)
|
||||||
|
|
|
@ -3,6 +3,7 @@ import json
|
||||||
import os.path
|
import os.path
|
||||||
from .util import string
|
from .util import string
|
||||||
|
|
||||||
|
|
||||||
class ConfigReader(object):
|
class ConfigReader(object):
|
||||||
def __init__(self, config_file_path):
|
def __init__(self, config_file_path):
|
||||||
self._config = self._read(config_file_path)
|
self._config = self._read(config_file_path)
|
||||||
|
@ -11,17 +12,18 @@ class ConfigReader(object):
|
||||||
try:
|
try:
|
||||||
_, ext = os.path.splitext(config_file_path)
|
_, ext = os.path.splitext(config_file_path)
|
||||||
with open(config_file_path) as fin:
|
with open(config_file_path) as fin:
|
||||||
if ext == '.json':
|
if ext == ".json":
|
||||||
data = json.load(fin)
|
data = json.load(fin)
|
||||||
else:
|
else:
|
||||||
data = yaml.safe_load(fin)
|
data = yaml.safe_load(fin)
|
||||||
return data
|
return data
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
msg = string.indent_lines(str(e))
|
msg = string.indent_lines(str(e))
|
||||||
raise ReadingError('Could not read config file:\n%s' % msg)
|
raise ReadingError("Could not read config file:\n%s" % msg)
|
||||||
|
|
||||||
def get_config(self):
|
def get_config(self):
|
||||||
return self._config
|
return self._config
|
||||||
|
|
||||||
|
|
||||||
class ReadingError(Exception):
|
class ReadingError(Exception):
|
||||||
pass
|
pass
|
||||||
|
|
|
@ -1,10 +1,11 @@
|
||||||
import copy
|
import copy
|
||||||
import os
|
import os
|
||||||
|
|
||||||
|
|
||||||
class Context(object):
|
class Context(object):
|
||||||
'''
|
"""
|
||||||
Contextual data and information for plugins.
|
Contextual data and information for plugins.
|
||||||
'''
|
"""
|
||||||
|
|
||||||
def __init__(self, base_directory):
|
def __init__(self, base_directory):
|
||||||
self._base_directory = base_directory
|
self._base_directory = base_directory
|
||||||
|
|
|
@ -4,6 +4,7 @@ from .messenger import Messenger
|
||||||
from .context import Context
|
from .context import Context
|
||||||
import traceback
|
import traceback
|
||||||
|
|
||||||
|
|
||||||
class Dispatcher(object):
|
class Dispatcher(object):
|
||||||
def __init__(self, base_directory, only=None, skip=None):
|
def __init__(self, base_directory, only=None, skip=None):
|
||||||
self._log = Messenger()
|
self._log = Messenger()
|
||||||
|
@ -13,25 +14,27 @@ class Dispatcher(object):
|
||||||
self._skip = skip
|
self._skip = skip
|
||||||
|
|
||||||
def _setup_context(self, base_directory):
|
def _setup_context(self, base_directory):
|
||||||
path = os.path.abspath(
|
path = os.path.abspath(os.path.expanduser(base_directory))
|
||||||
os.path.expanduser(base_directory))
|
|
||||||
if not os.path.exists(path):
|
if not os.path.exists(path):
|
||||||
raise DispatchError('Nonexistent base directory')
|
raise DispatchError("Nonexistent base directory")
|
||||||
self._context = Context(path)
|
self._context = Context(path)
|
||||||
|
|
||||||
def dispatch(self, tasks):
|
def dispatch(self, tasks):
|
||||||
success = True
|
success = True
|
||||||
for task in tasks:
|
for task in tasks:
|
||||||
for action in task.keys():
|
for action in task.keys():
|
||||||
if (self._only is not None and action not in self._only \
|
if (
|
||||||
or self._skip is not None and action in self._skip) \
|
self._only is not None
|
||||||
and action != 'defaults':
|
and action not in self._only
|
||||||
self._log.info('Skipping action %s' % action)
|
or self._skip is not None
|
||||||
|
and action in self._skip
|
||||||
|
) and action != "defaults":
|
||||||
|
self._log.info("Skipping action %s" % action)
|
||||||
continue
|
continue
|
||||||
handled = False
|
handled = False
|
||||||
# print("\tcurrent action", action)
|
# print("\tcurrent action", action)
|
||||||
if action == 'defaults':
|
if action == "defaults":
|
||||||
self._context.set_defaults(task[action]) # replace, not update
|
self._context.set_defaults(task[action]) # replace, not update
|
||||||
handled = True
|
handled = True
|
||||||
# keep going, let other plugins handle this if they want
|
# keep going, let other plugins handle this if they want
|
||||||
for plugin in self._plugins:
|
for plugin in self._plugins:
|
||||||
|
@ -44,9 +47,7 @@ class Dispatcher(object):
|
||||||
except Exception as err:
|
except Exception as err:
|
||||||
print("failure", err)
|
print("failure", err)
|
||||||
traceback.print_exception(type(err), err, err.__traceback__)
|
traceback.print_exception(type(err), err, err.__traceback__)
|
||||||
self._log.error(
|
self._log.error('An error was encountered while executing action "%s"' % action)
|
||||||
'An error was encountered while executing action "%s"' %
|
|
||||||
action)
|
|
||||||
self._log.debug(err)
|
self._log.debug(err)
|
||||||
if not handled:
|
if not handled:
|
||||||
success = False
|
success = False
|
||||||
|
@ -54,8 +55,8 @@ class Dispatcher(object):
|
||||||
return success
|
return success
|
||||||
|
|
||||||
def _load_plugins(self):
|
def _load_plugins(self):
|
||||||
self._plugins = [plugin(self._context)
|
self._plugins = [plugin(self._context) for plugin in Plugin.__subclasses__()]
|
||||||
for plugin in Plugin.__subclasses__()]
|
|
||||||
|
|
||||||
class DispatchError(Exception):
|
class DispatchError(Exception):
|
||||||
pass
|
pass
|
||||||
|
|
|
@ -1,8 +1,8 @@
|
||||||
class Color(object):
|
class Color(object):
|
||||||
NONE = ''
|
NONE = ""
|
||||||
RESET = '\033[0m'
|
RESET = "\033[0m"
|
||||||
RED = '\033[91m'
|
RED = "\033[91m"
|
||||||
GREEN = '\033[92m'
|
GREEN = "\033[92m"
|
||||||
YELLOW = '\033[93m'
|
YELLOW = "\033[93m"
|
||||||
BLUE = '\033[94m'
|
BLUE = "\033[94m"
|
||||||
MAGENTA = '\033[95m'
|
MAGENTA = "\033[95m"
|
||||||
|
|
|
@ -3,8 +3,9 @@ from ..util.compat import with_metaclass
|
||||||
from .color import Color
|
from .color import Color
|
||||||
from .level import Level
|
from .level import Level
|
||||||
|
|
||||||
|
|
||||||
class Messenger(with_metaclass(Singleton, object)):
|
class Messenger(with_metaclass(Singleton, object)):
|
||||||
def __init__(self, level = Level.LOWINFO):
|
def __init__(self, level=Level.LOWINFO):
|
||||||
self.set_level(level)
|
self.set_level(level)
|
||||||
self.use_color(True)
|
self.use_color(True)
|
||||||
|
|
||||||
|
@ -15,8 +16,8 @@ class Messenger(with_metaclass(Singleton, object)):
|
||||||
self._use_color = yesno
|
self._use_color = yesno
|
||||||
|
|
||||||
def log(self, level, message):
|
def log(self, level, message):
|
||||||
if (level >= self._level):
|
if level >= self._level:
|
||||||
print('%s%s%s' % (self._color(level), message, self._reset()))
|
print("%s%s%s" % (self._color(level), message, self._reset()))
|
||||||
|
|
||||||
def debug(self, message):
|
def debug(self, message):
|
||||||
self.log(Level.DEBUG, message)
|
self.log(Level.DEBUG, message)
|
||||||
|
@ -34,13 +35,13 @@ class Messenger(with_metaclass(Singleton, object)):
|
||||||
self.log(Level.ERROR, message)
|
self.log(Level.ERROR, message)
|
||||||
|
|
||||||
def _color(self, level):
|
def _color(self, level):
|
||||||
'''
|
"""
|
||||||
Get a color (terminal escape sequence) according to a level.
|
Get a color (terminal escape sequence) according to a level.
|
||||||
'''
|
"""
|
||||||
if not self._use_color:
|
if not self._use_color:
|
||||||
return ''
|
return ""
|
||||||
elif level < Level.DEBUG:
|
elif level < Level.DEBUG:
|
||||||
return ''
|
return ""
|
||||||
elif Level.DEBUG <= level < Level.LOWINFO:
|
elif Level.DEBUG <= level < Level.LOWINFO:
|
||||||
return Color.YELLOW
|
return Color.YELLOW
|
||||||
elif Level.LOWINFO <= level < Level.INFO:
|
elif Level.LOWINFO <= level < Level.INFO:
|
||||||
|
@ -53,10 +54,10 @@ class Messenger(with_metaclass(Singleton, object)):
|
||||||
return Color.RED
|
return Color.RED
|
||||||
|
|
||||||
def _reset(self):
|
def _reset(self):
|
||||||
'''
|
"""
|
||||||
Get a reset color (terminal escape sequence).
|
Get a reset color (terminal escape sequence).
|
||||||
'''
|
"""
|
||||||
if not self._use_color:
|
if not self._use_color:
|
||||||
return ''
|
return ""
|
||||||
else:
|
else:
|
||||||
return Color.RESET
|
return Color.RESET
|
||||||
|
|
|
@ -1,25 +1,26 @@
|
||||||
from .messenger import Messenger
|
from .messenger import Messenger
|
||||||
from .context import Context
|
from .context import Context
|
||||||
|
|
||||||
|
|
||||||
class Plugin(object):
|
class Plugin(object):
|
||||||
'''
|
"""
|
||||||
Abstract base class for commands that process directives.
|
Abstract base class for commands that process directives.
|
||||||
'''
|
"""
|
||||||
|
|
||||||
def __init__(self, context):
|
def __init__(self, context):
|
||||||
self._context = context
|
self._context = context
|
||||||
self._log = Messenger()
|
self._log = Messenger()
|
||||||
|
|
||||||
def can_handle(self, directive):
|
def can_handle(self, directive):
|
||||||
'''
|
"""
|
||||||
Returns true if the Plugin can handle the directive.
|
Returns true if the Plugin can handle the directive.
|
||||||
'''
|
"""
|
||||||
raise NotImplementedError
|
raise NotImplementedError
|
||||||
|
|
||||||
def handle(self, directive, data):
|
def handle(self, directive, data):
|
||||||
'''
|
"""
|
||||||
Executes the directive.
|
Executes the directive.
|
||||||
|
|
||||||
Returns true if the Plugin successfully handled the directive.
|
Returns true if the Plugin successfully handled the directive.
|
||||||
'''
|
"""
|
||||||
raise NotImplementedError
|
raise NotImplementedError
|
||||||
|
|
|
@ -3,45 +3,45 @@ import dotbot
|
||||||
|
|
||||||
|
|
||||||
class Clean(dotbot.Plugin):
|
class Clean(dotbot.Plugin):
|
||||||
'''
|
"""
|
||||||
Cleans broken symbolic links.
|
Cleans broken symbolic links.
|
||||||
'''
|
"""
|
||||||
|
|
||||||
_directive = 'clean'
|
_directive = "clean"
|
||||||
|
|
||||||
def can_handle(self, directive):
|
def can_handle(self, directive):
|
||||||
return directive == self._directive
|
return directive == self._directive
|
||||||
|
|
||||||
def handle(self, directive, data):
|
def handle(self, directive, data):
|
||||||
if directive != self._directive:
|
if directive != self._directive:
|
||||||
raise ValueError('Clean cannot handle directive %s' % directive)
|
raise ValueError("Clean cannot handle directive %s" % directive)
|
||||||
return self._process_clean(data)
|
return self._process_clean(data)
|
||||||
|
|
||||||
def _process_clean(self, targets):
|
def _process_clean(self, targets):
|
||||||
success = True
|
success = True
|
||||||
defaults = self._context.defaults().get(self._directive, {})
|
defaults = self._context.defaults().get(self._directive, {})
|
||||||
for target in targets:
|
for target in targets:
|
||||||
force = defaults.get('force', False)
|
force = defaults.get("force", False)
|
||||||
recursive = defaults.get('recursive', False)
|
recursive = defaults.get("recursive", False)
|
||||||
if isinstance(targets, dict) and isinstance(targets[target], dict):
|
if isinstance(targets, dict) and isinstance(targets[target], dict):
|
||||||
force = targets[target].get('force', force)
|
force = targets[target].get("force", force)
|
||||||
recursive = targets[target].get('recursive', recursive)
|
recursive = targets[target].get("recursive", recursive)
|
||||||
success &= self._clean(target, force, recursive)
|
success &= self._clean(target, force, recursive)
|
||||||
if success:
|
if success:
|
||||||
self._log.info('All targets have been cleaned')
|
self._log.info("All targets have been cleaned")
|
||||||
else:
|
else:
|
||||||
self._log.error('Some targets were not successfully cleaned')
|
self._log.error("Some targets were not successfully cleaned")
|
||||||
return success
|
return success
|
||||||
|
|
||||||
def _clean(self, target, force, recursive):
|
def _clean(self, target, force, recursive):
|
||||||
'''
|
"""
|
||||||
Cleans all the broken symbolic links in target if they point to
|
Cleans all the broken symbolic links in target if they point to
|
||||||
a subdirectory of the base directory or if forced to clean.
|
a subdirectory of the base directory or if forced to clean.
|
||||||
'''
|
"""
|
||||||
target_normalised = os.path.normpath(os.path.expandvars(os.path.expanduser(target)))
|
target_normalised = os.path.normpath(os.path.expandvars(os.path.expanduser(target)))
|
||||||
isdir = os.path.isdir(target_normalised)
|
isdir = os.path.isdir(target_normalised)
|
||||||
if isdir is False:
|
if isdir is False:
|
||||||
self._log.debug('Ignoring nonexistent directory %s' % target)
|
self._log.debug("Ignoring nonexistent directory %s" % target)
|
||||||
return True
|
return True
|
||||||
for item in os.listdir(target_normalised):
|
for item in os.listdir(target_normalised):
|
||||||
path = os.path.join(target_normalised, item)
|
path = os.path.join(target_normalised, item)
|
||||||
|
@ -53,16 +53,16 @@ class Clean(dotbot.Plugin):
|
||||||
if not os.path.exists(path) and os.path.islink(path):
|
if not os.path.exists(path) and os.path.islink(path):
|
||||||
points_at = os.path.join(os.path.dirname(path), os.readlink(path))
|
points_at = os.path.join(os.path.dirname(path), os.readlink(path))
|
||||||
if self._in_directory(path, self._context.base_directory()) or force:
|
if self._in_directory(path, self._context.base_directory()) or force:
|
||||||
self._log.lowinfo('Removing invalid link %s -> %s' % (path, points_at))
|
self._log.lowinfo("Removing invalid link %s -> %s" % (path, points_at))
|
||||||
os.remove(path)
|
os.remove(path)
|
||||||
else:
|
else:
|
||||||
self._log.lowinfo('Link %s -> %s not removed.' % (path, points_at))
|
self._log.lowinfo("Link %s -> %s not removed." % (path, points_at))
|
||||||
return True
|
return True
|
||||||
|
|
||||||
def _in_directory(self, path, directory):
|
def _in_directory(self, path, directory):
|
||||||
'''
|
"""
|
||||||
Returns true if the path is in the directory.
|
Returns true if the path is in the directory.
|
||||||
'''
|
"""
|
||||||
directory = os.path.join(os.path.realpath(directory), '')
|
directory = os.path.join(os.path.realpath(directory), "")
|
||||||
path = os.path.realpath(path)
|
path = os.path.realpath(path)
|
||||||
return os.path.commonprefix([path, directory]) == directory
|
return os.path.commonprefix([path, directory]) == directory
|
||||||
|
|
|
@ -3,18 +3,18 @@ import dotbot
|
||||||
|
|
||||||
|
|
||||||
class Create(dotbot.Plugin):
|
class Create(dotbot.Plugin):
|
||||||
'''
|
"""
|
||||||
Create empty paths.
|
Create empty paths.
|
||||||
'''
|
"""
|
||||||
|
|
||||||
_directive = 'create'
|
_directive = "create"
|
||||||
|
|
||||||
def can_handle(self, directive):
|
def can_handle(self, directive):
|
||||||
return directive == self._directive
|
return directive == self._directive
|
||||||
|
|
||||||
def handle(self, directive, data):
|
def handle(self, directive, data):
|
||||||
if directive != self._directive:
|
if directive != self._directive:
|
||||||
raise ValueError('Create cannot handle directive %s' % directive)
|
raise ValueError("Create cannot handle directive %s" % directive)
|
||||||
return self._process_paths(data)
|
return self._process_paths(data)
|
||||||
|
|
||||||
def _process_paths(self, paths):
|
def _process_paths(self, paths):
|
||||||
|
@ -23,28 +23,28 @@ class Create(dotbot.Plugin):
|
||||||
path = os.path.normpath(os.path.expandvars(os.path.expanduser(path)))
|
path = os.path.normpath(os.path.expandvars(os.path.expanduser(path)))
|
||||||
success &= self._create(path)
|
success &= self._create(path)
|
||||||
if success:
|
if success:
|
||||||
self._log.info('All paths have been set up')
|
self._log.info("All paths have been set up")
|
||||||
else:
|
else:
|
||||||
self._log.error('Some paths were not successfully set up')
|
self._log.error("Some paths were not successfully set up")
|
||||||
return success
|
return success
|
||||||
|
|
||||||
def _exists(self, path):
|
def _exists(self, path):
|
||||||
'''
|
"""
|
||||||
Returns true if the path exists.
|
Returns true if the path exists.
|
||||||
'''
|
"""
|
||||||
path = os.path.expanduser(path)
|
path = os.path.expanduser(path)
|
||||||
return os.path.exists(path)
|
return os.path.exists(path)
|
||||||
|
|
||||||
def _create(self, path):
|
def _create(self, path):
|
||||||
success = True
|
success = True
|
||||||
if not self._exists(path):
|
if not self._exists(path):
|
||||||
self._log.debug('Trying to create path %s' % path)
|
self._log.debug("Trying to create path %s" % path)
|
||||||
try:
|
try:
|
||||||
self._log.lowinfo('Creating path %s' % path)
|
self._log.lowinfo("Creating path %s" % path)
|
||||||
os.makedirs(path)
|
os.makedirs(path)
|
||||||
except OSError:
|
except OSError:
|
||||||
self._log.warning('Failed to create path %s' % path)
|
self._log.warning("Failed to create path %s" % path)
|
||||||
success = False
|
success = False
|
||||||
else:
|
else:
|
||||||
self._log.lowinfo('Path exists %s' % path)
|
self._log.lowinfo("Path exists %s" % path)
|
||||||
return success
|
return success
|
||||||
|
|
|
@ -7,49 +7,49 @@ import subprocess
|
||||||
|
|
||||||
|
|
||||||
class Link(dotbot.Plugin):
|
class Link(dotbot.Plugin):
|
||||||
'''
|
"""
|
||||||
Symbolically links dotfiles.
|
Symbolically links dotfiles.
|
||||||
'''
|
"""
|
||||||
|
|
||||||
_directive = 'link'
|
_directive = "link"
|
||||||
|
|
||||||
def can_handle(self, directive):
|
def can_handle(self, directive):
|
||||||
return directive == self._directive
|
return directive == self._directive
|
||||||
|
|
||||||
def handle(self, directive, data):
|
def handle(self, directive, data):
|
||||||
if directive != self._directive:
|
if directive != self._directive:
|
||||||
raise ValueError('Link cannot handle directive %s' % directive)
|
raise ValueError("Link cannot handle directive %s" % directive)
|
||||||
return self._process_links(data)
|
return self._process_links(data)
|
||||||
|
|
||||||
def _process_links(self, links):
|
def _process_links(self, links):
|
||||||
# print("symlinking\n\t", links)
|
# print("symlinking\n\t", links)
|
||||||
success = True
|
success = True
|
||||||
defaults = self._context.defaults().get('link', {})
|
defaults = self._context.defaults().get("link", {})
|
||||||
for destination, source in links.items():
|
for destination, source in links.items():
|
||||||
destination = os.path.expandvars(destination)
|
destination = os.path.expandvars(destination)
|
||||||
relative = defaults.get('relative', False)
|
relative = defaults.get("relative", False)
|
||||||
canonical_path = defaults.get('canonicalize-path', True)
|
canonical_path = defaults.get("canonicalize-path", True)
|
||||||
force = defaults.get('force', False)
|
force = defaults.get("force", False)
|
||||||
relink = defaults.get('relink', False)
|
relink = defaults.get("relink", False)
|
||||||
create = defaults.get('create', False)
|
create = defaults.get("create", False)
|
||||||
use_glob = defaults.get('glob', False)
|
use_glob = defaults.get("glob", False)
|
||||||
test = defaults.get('if', None)
|
test = defaults.get("if", None)
|
||||||
ignore_missing = defaults.get('ignore-missing', False)
|
ignore_missing = defaults.get("ignore-missing", False)
|
||||||
if isinstance(source, dict):
|
if isinstance(source, dict):
|
||||||
# extended config
|
# extended config
|
||||||
test = source.get('if', test)
|
test = source.get("if", test)
|
||||||
relative = source.get('relative', relative)
|
relative = source.get("relative", relative)
|
||||||
canonical_path = source.get('canonicalize-path', canonical_path)
|
canonical_path = source.get("canonicalize-path", canonical_path)
|
||||||
force = source.get('force', force)
|
force = source.get("force", force)
|
||||||
relink = source.get('relink', relink)
|
relink = source.get("relink", relink)
|
||||||
create = source.get('create', create)
|
create = source.get("create", create)
|
||||||
use_glob = source.get('glob', use_glob)
|
use_glob = source.get("glob", use_glob)
|
||||||
ignore_missing = source.get('ignore-missing', ignore_missing)
|
ignore_missing = source.get("ignore-missing", ignore_missing)
|
||||||
path = self._default_source(destination, source.get('path'))
|
path = self._default_source(destination, source.get("path"))
|
||||||
else:
|
else:
|
||||||
path = self._default_source(destination, source)
|
path = self._default_source(destination, source)
|
||||||
if test is not None and not self._test_success(test):
|
if test is not None and not self._test_success(test):
|
||||||
self._log.lowinfo('Skipping %s' % destination)
|
self._log.lowinfo("Skipping %s" % destination)
|
||||||
continue
|
continue
|
||||||
path = os.path.expandvars(os.path.expanduser(path))
|
path = os.path.expandvars(os.path.expanduser(path))
|
||||||
if use_glob:
|
if use_glob:
|
||||||
|
@ -59,11 +59,12 @@ class Link(dotbot.Plugin):
|
||||||
self._log.warning("Globbing couldn't find anything matching " + str(path))
|
self._log.warning("Globbing couldn't find anything matching " + str(path))
|
||||||
success = False
|
success = False
|
||||||
continue
|
continue
|
||||||
glob_star_loc = path.find('*')
|
glob_star_loc = path.find("*")
|
||||||
if glob_star_loc == -1 and destination[-1] == '/':
|
if glob_star_loc == -1 and destination[-1] == "/":
|
||||||
self._log.error("Ambiguous action requested.")
|
self._log.error("Ambiguous action requested.")
|
||||||
self._log.error("No wildcard in glob, directory use undefined: " +
|
self._log.error(
|
||||||
destination + " -> " + str(glob_results))
|
"No wildcard in glob, directory use undefined: " + destination + " -> " + str(glob_results)
|
||||||
|
)
|
||||||
self._log.warning("Did you want to link the directory or into it?")
|
self._log.warning("Did you want to link the directory or into it?")
|
||||||
success = False
|
success = False
|
||||||
continue
|
continue
|
||||||
|
@ -78,13 +79,17 @@ class Link(dotbot.Plugin):
|
||||||
self._log.lowinfo("Globs from '" + path + "': " + str(glob_results))
|
self._log.lowinfo("Globs from '" + path + "': " + str(glob_results))
|
||||||
glob_base = path[:glob_star_loc]
|
glob_base = path[:glob_star_loc]
|
||||||
for glob_full_item in glob_results:
|
for glob_full_item in glob_results:
|
||||||
glob_item = glob_full_item[len(glob_base):]
|
glob_item = glob_full_item[len(glob_base) :]
|
||||||
glob_link_destination = os.path.join(destination, glob_item)
|
glob_link_destination = os.path.join(destination, glob_item)
|
||||||
if create:
|
if create:
|
||||||
success &= self._create(glob_link_destination)
|
success &= self._create(glob_link_destination)
|
||||||
if force or relink:
|
if force or relink:
|
||||||
success &= self._delete(glob_full_item, glob_link_destination, relative, canonical_path, force)
|
success &= self._delete(
|
||||||
success &= self._link(glob_full_item, glob_link_destination, relative, canonical_path, ignore_missing)
|
glob_full_item, glob_link_destination, relative, canonical_path, force
|
||||||
|
)
|
||||||
|
success &= self._link(
|
||||||
|
glob_full_item, glob_link_destination, relative, canonical_path, ignore_missing
|
||||||
|
)
|
||||||
else:
|
else:
|
||||||
if create:
|
if create:
|
||||||
success &= self._create(destination)
|
success &= self._create(destination)
|
||||||
|
@ -94,28 +99,27 @@ class Link(dotbot.Plugin):
|
||||||
# want to remove the original (this is tested by
|
# want to remove the original (this is tested by
|
||||||
# link-force-leaves-when-nonexistent.bash)
|
# link-force-leaves-when-nonexistent.bash)
|
||||||
success = False
|
success = False
|
||||||
self._log.warning('Nonexistent source %s -> %s' %
|
self._log.warning("Nonexistent source %s -> %s" % (destination, path))
|
||||||
(destination, path))
|
|
||||||
continue
|
continue
|
||||||
if force or relink:
|
if force or relink:
|
||||||
success &= self._delete(path, destination, relative, canonical_path, force)
|
success &= self._delete(path, destination, relative, canonical_path, force)
|
||||||
success &= self._link(path, destination, relative, canonical_path, ignore_missing)
|
success &= self._link(path, destination, relative, canonical_path, ignore_missing)
|
||||||
if success:
|
if success:
|
||||||
self._log.info('All links have been set up')
|
self._log.info("All links have been set up")
|
||||||
else:
|
else:
|
||||||
self._log.error('Some links were not successfully set up')
|
self._log.error("Some links were not successfully set up")
|
||||||
return success
|
return success
|
||||||
|
|
||||||
def _test_success(self, command):
|
def _test_success(self, command):
|
||||||
ret = dotbot.util.shell_command(command, cwd=self._context.base_directory())
|
ret = dotbot.util.shell_command(command, cwd=self._context.base_directory())
|
||||||
if ret != 0:
|
if ret != 0:
|
||||||
self._log.debug('Test \'%s\' returned false' % command)
|
self._log.debug("Test '%s' returned false" % command)
|
||||||
return ret == 0
|
return ret == 0
|
||||||
|
|
||||||
def _default_source(self, destination, source):
|
def _default_source(self, destination, source):
|
||||||
if source is None:
|
if source is None:
|
||||||
basename = os.path.basename(destination)
|
basename = os.path.basename(destination)
|
||||||
if basename.startswith('.'):
|
if basename.startswith("."):
|
||||||
return basename[1:]
|
return basename[1:]
|
||||||
else:
|
else:
|
||||||
return basename
|
return basename
|
||||||
|
@ -123,17 +127,17 @@ class Link(dotbot.Plugin):
|
||||||
return source
|
return source
|
||||||
|
|
||||||
def _is_link(self, path):
|
def _is_link(self, path):
|
||||||
'''
|
"""
|
||||||
Returns true if the path is a symbolic link.
|
Returns true if the path is a symbolic link.
|
||||||
'''
|
"""
|
||||||
return os.path.islink(os.path.expanduser(path))
|
return os.path.islink(os.path.expanduser(path))
|
||||||
|
|
||||||
def _link_destination(self, path):
|
def _link_destination(self, path):
|
||||||
'''
|
"""
|
||||||
Returns the destination of the symbolic link. Truncates the \\?\ start to a path if it
|
Returns the destination of the symbolic link. Truncates the \\?\ start to a path if it
|
||||||
is present. This is an identifier which allows >255 character file name links to work.
|
is present. This is an identifier which allows >255 character file name links to work.
|
||||||
Since this function is for the point of comparison, it is okay to truncate
|
Since this function is for the point of comparison, it is okay to truncate
|
||||||
'''
|
"""
|
||||||
# path = os.path.normpath(path)
|
# path = os.path.normpath(path)
|
||||||
path = os.path.expanduser(path)
|
path = os.path.expanduser(path)
|
||||||
try:
|
try:
|
||||||
|
@ -141,22 +145,21 @@ class Link(dotbot.Plugin):
|
||||||
# Read link can return paths starting with \\?\ - this allows over the 255 file name
|
# Read link can return paths starting with \\?\ - this allows over the 255 file name
|
||||||
# limit
|
# limit
|
||||||
except OSError as e:
|
except OSError as e:
|
||||||
if "[WinError 4390] The file or directory is not a reparse point" in str(e) and \
|
if "[WinError 4390] The file or directory is not a reparse point" in str(e) and os.path.isdir(path):
|
||||||
os.path.isdir(path):
|
|
||||||
return "UNLINKED_DIR"
|
return "UNLINKED_DIR"
|
||||||
return "OSERROR_READING_LINK"
|
return "OSERROR_READING_LINK"
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
print(e)
|
print(e)
|
||||||
return 'GENERAL_EXCEPTION_READING_LINK'
|
return "GENERAL_EXCEPTION_READING_LINK"
|
||||||
else:
|
else:
|
||||||
if read_link.startswith("\\\\?\\"):
|
if read_link.startswith("\\\\?\\"):
|
||||||
read_link = read_link.replace("\\\\?\\", "")
|
read_link = read_link.replace("\\\\?\\", "")
|
||||||
return read_link
|
return read_link
|
||||||
|
|
||||||
def _exists(self, path):
|
def _exists(self, path):
|
||||||
'''
|
"""
|
||||||
Returns true if the path exists. Returns false if contains dangling symbolic links.
|
Returns true if the path exists. Returns false if contains dangling symbolic links.
|
||||||
'''
|
"""
|
||||||
path = os.path.expanduser(path)
|
path = os.path.expanduser(path)
|
||||||
return os.path.exists(path)
|
return os.path.exists(path)
|
||||||
|
|
||||||
|
@ -168,10 +171,10 @@ class Link(dotbot.Plugin):
|
||||||
try:
|
try:
|
||||||
os.makedirs(parent)
|
os.makedirs(parent)
|
||||||
except OSError:
|
except OSError:
|
||||||
self._log.warning('Failed to create directory %s' % parent)
|
self._log.warning("Failed to create directory %s" % parent)
|
||||||
success = False
|
success = False
|
||||||
else:
|
else:
|
||||||
self._log.lowinfo('Creating directory %s' % parent)
|
self._log.lowinfo("Creating directory %s" % parent)
|
||||||
return success
|
return success
|
||||||
|
|
||||||
def _delete(self, source, path, relative, canonical_path, force):
|
def _delete(self, source, path, relative, canonical_path, force):
|
||||||
|
@ -180,8 +183,9 @@ class Link(dotbot.Plugin):
|
||||||
fullpath = os.path.expanduser(path)
|
fullpath = os.path.expanduser(path)
|
||||||
if relative:
|
if relative:
|
||||||
source = self._relative_path(source, fullpath)
|
source = self._relative_path(source, fullpath)
|
||||||
if ((self._is_link(path) and self._link_destination(path) != source) or
|
if (self._is_link(path) and self._link_destination(path) != source) or (
|
||||||
(self._exists(path) and not self._is_link(path))):
|
self._exists(path) and not self._is_link(path)
|
||||||
|
):
|
||||||
removed = False
|
removed = False
|
||||||
try:
|
try:
|
||||||
if os.path.islink(fullpath):
|
if os.path.islink(fullpath):
|
||||||
|
@ -195,29 +199,29 @@ class Link(dotbot.Plugin):
|
||||||
os.remove(fullpath)
|
os.remove(fullpath)
|
||||||
removed = True
|
removed = True
|
||||||
except OSError:
|
except OSError:
|
||||||
self._log.warning('Failed to remove %s' % path)
|
self._log.warning("Failed to remove %s" % path)
|
||||||
success = False
|
success = False
|
||||||
else:
|
else:
|
||||||
if removed:
|
if removed:
|
||||||
self._log.lowinfo('Removing %s' % path)
|
self._log.lowinfo("Removing %s" % path)
|
||||||
return success
|
return success
|
||||||
|
|
||||||
def _relative_path(self, source, destination):
|
def _relative_path(self, source, destination):
|
||||||
'''
|
"""
|
||||||
Returns the relative path to get to the source file from the
|
Returns the relative path to get to the source file from the
|
||||||
destination file.
|
destination file.
|
||||||
'''
|
"""
|
||||||
destination_dir = os.path.dirname(destination)
|
destination_dir = os.path.dirname(destination)
|
||||||
return os.path.relpath(source, destination_dir)
|
return os.path.relpath(source, destination_dir)
|
||||||
|
|
||||||
def _link(self, dotfile_source, target_path_to_link_at, relative, canonical_path, ignore_missing):
|
def _link(self, dotfile_source, target_path_to_link_at, relative, canonical_path, ignore_missing):
|
||||||
'''
|
"""
|
||||||
Links link_name to source.
|
Links link_name to source.
|
||||||
:param target_path_to_link_at is the file path where we are putting a symlink back to
|
:param target_path_to_link_at is the file path where we are putting a symlink back to
|
||||||
dotfile_source
|
dotfile_source
|
||||||
|
|
||||||
Returns true if successfully linked files.
|
Returns true if successfully linked files.
|
||||||
'''
|
"""
|
||||||
success = False
|
success = False
|
||||||
destination = os.path.expanduser(target_path_to_link_at)
|
destination = os.path.expanduser(target_path_to_link_at)
|
||||||
base_directory = self._context.base_directory(canonical_path=canonical_path)
|
base_directory = self._context.base_directory(canonical_path=canonical_path)
|
||||||
|
@ -230,10 +234,8 @@ class Link(dotbot.Plugin):
|
||||||
target_path_exists: bool = self._exists(target_path_to_link_at)
|
target_path_exists: bool = self._exists(target_path_to_link_at)
|
||||||
target_file_is_link: bool = self._is_link(target_path_to_link_at)
|
target_file_is_link: bool = self._is_link(target_path_to_link_at)
|
||||||
symlink_dest_at_target_path: str = self._link_destination(target_path_to_link_at)
|
symlink_dest_at_target_path: str = self._link_destination(target_path_to_link_at)
|
||||||
if (not target_path_exists and target_file_is_link and
|
if not target_path_exists and target_file_is_link and symlink_dest_at_target_path != dotfile_source:
|
||||||
symlink_dest_at_target_path != dotfile_source):
|
self._log.warning("Invalid link %s -> %s" % (target_path_to_link_at, symlink_dest_at_target_path))
|
||||||
self._log.warning('Invalid link %s -> %s' %
|
|
||||||
(target_path_to_link_at, symlink_dest_at_target_path))
|
|
||||||
# we need to use absolute_source below because our cwd is the dotfiles
|
# we need to use absolute_source below because our cwd is the dotfiles
|
||||||
# directory, and if source is relative, it will be relative to the
|
# directory, and if source is relative, it will be relative to the
|
||||||
# destination directory
|
# destination directory
|
||||||
|
@ -241,29 +243,26 @@ class Link(dotbot.Plugin):
|
||||||
try:
|
try:
|
||||||
os.symlink(dotfile_source, destination)
|
os.symlink(dotfile_source, destination)
|
||||||
except OSError:
|
except OSError:
|
||||||
self._log.warning('Linking failed %s -> %s' % (target_path_to_link_at, dotfile_source))
|
self._log.warning("Linking failed %s -> %s" % (target_path_to_link_at, dotfile_source))
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
print(f"SYMLINK FAILED with arguments os.symlink({dotfile_source}, {destination})", )
|
print(
|
||||||
|
f"SYMLINK FAILED with arguments os.symlink({dotfile_source}, {destination})",
|
||||||
|
)
|
||||||
raise e
|
raise e
|
||||||
else:
|
else:
|
||||||
self._log.lowinfo('Creating link %s -> %s' % (target_path_to_link_at, dotfile_source))
|
self._log.lowinfo("Creating link %s -> %s" % (target_path_to_link_at, dotfile_source))
|
||||||
success = True
|
success = True
|
||||||
elif target_path_exists and not target_file_is_link:
|
elif target_path_exists and not target_file_is_link:
|
||||||
self._log.warning(
|
self._log.warning("%s already exists but is a regular file or directory" % target_path_to_link_at)
|
||||||
'%s already exists but is a regular file or directory' %
|
|
||||||
target_path_to_link_at)
|
|
||||||
elif target_file_is_link and symlink_dest_at_target_path != dotfile_source:
|
elif target_file_is_link and symlink_dest_at_target_path != dotfile_source:
|
||||||
self._log.warning('Incorrect link %s -> %s' %
|
self._log.warning("Incorrect link %s -> %s" % (target_path_to_link_at, symlink_dest_at_target_path))
|
||||||
(target_path_to_link_at, symlink_dest_at_target_path))
|
|
||||||
# again, we use absolute_source to check for existence
|
# again, we use absolute_source to check for existence
|
||||||
elif not self._exists(absolute_source):
|
elif not self._exists(absolute_source):
|
||||||
if target_file_is_link:
|
if target_file_is_link:
|
||||||
self._log.warning('Nonexistent source %s -> %s' %
|
self._log.warning("Nonexistent source %s -> %s" % (target_path_to_link_at, dotfile_source))
|
||||||
(target_path_to_link_at, dotfile_source))
|
|
||||||
else:
|
else:
|
||||||
self._log.warning('Nonexistent source for %s : %s' %
|
self._log.warning("Nonexistent source for %s : %s" % (target_path_to_link_at, dotfile_source))
|
||||||
(target_path_to_link_at, dotfile_source))
|
|
||||||
else:
|
else:
|
||||||
self._log.lowinfo('Link exists %s -> %s' % (target_path_to_link_at, dotfile_source))
|
self._log.lowinfo("Link exists %s -> %s" % (target_path_to_link_at, dotfile_source))
|
||||||
success = True
|
success = True
|
||||||
return success
|
return success
|
||||||
|
|
|
@ -5,36 +5,35 @@ import dotbot.util
|
||||||
|
|
||||||
|
|
||||||
class Shell(dotbot.Plugin):
|
class Shell(dotbot.Plugin):
|
||||||
'''
|
"""
|
||||||
Run arbitrary shell commands.
|
Run arbitrary shell commands.
|
||||||
'''
|
"""
|
||||||
|
|
||||||
_directive = 'shell'
|
_directive = "shell"
|
||||||
|
|
||||||
def can_handle(self, directive):
|
def can_handle(self, directive):
|
||||||
return directive == self._directive
|
return directive == self._directive
|
||||||
|
|
||||||
def handle(self, directive, data):
|
def handle(self, directive, data):
|
||||||
if directive != self._directive:
|
if directive != self._directive:
|
||||||
raise ValueError('Shell cannot handle directive %s' %
|
raise ValueError("Shell cannot handle directive %s" % directive)
|
||||||
directive)
|
|
||||||
return self._process_commands(data)
|
return self._process_commands(data)
|
||||||
|
|
||||||
def _process_commands(self, data):
|
def _process_commands(self, data):
|
||||||
success = True
|
success = True
|
||||||
defaults = self._context.defaults().get('shell', {})
|
defaults = self._context.defaults().get("shell", {})
|
||||||
for item in data:
|
for item in data:
|
||||||
stdin = defaults.get('stdin', False)
|
stdin = defaults.get("stdin", False)
|
||||||
stdout = defaults.get('stdout', False)
|
stdout = defaults.get("stdout", False)
|
||||||
stderr = defaults.get('stderr', False)
|
stderr = defaults.get("stderr", False)
|
||||||
quiet = defaults.get('quiet', False)
|
quiet = defaults.get("quiet", False)
|
||||||
if isinstance(item, dict):
|
if isinstance(item, dict):
|
||||||
cmd = item['command']
|
cmd = item["command"]
|
||||||
msg = item.get('description', None)
|
msg = item.get("description", None)
|
||||||
stdin = item.get('stdin', stdin)
|
stdin = item.get("stdin", stdin)
|
||||||
stdout = item.get('stdout', stdout)
|
stdout = item.get("stdout", stdout)
|
||||||
stderr = item.get('stderr', stderr)
|
stderr = item.get("stderr", stderr)
|
||||||
quiet = item.get('quiet', quiet)
|
quiet = item.get("quiet", quiet)
|
||||||
elif isinstance(item, list):
|
elif isinstance(item, list):
|
||||||
cmd = item[0]
|
cmd = item[0]
|
||||||
msg = item[1] if len(item) > 1 else None
|
msg = item[1] if len(item) > 1 else None
|
||||||
|
@ -44,21 +43,17 @@ class Shell(dotbot.Plugin):
|
||||||
if msg is None:
|
if msg is None:
|
||||||
self._log.lowinfo(cmd)
|
self._log.lowinfo(cmd)
|
||||||
elif quiet:
|
elif quiet:
|
||||||
self._log.lowinfo('%s' % msg)
|
self._log.lowinfo("%s" % msg)
|
||||||
else:
|
else:
|
||||||
self._log.lowinfo('%s [%s]' % (msg, cmd))
|
self._log.lowinfo("%s [%s]" % (msg, cmd))
|
||||||
ret = dotbot.util.shell_command(
|
ret = dotbot.util.shell_command(
|
||||||
cmd,
|
cmd, cwd=self._context.base_directory(), enable_stdin=stdin, enable_stdout=stdout, enable_stderr=stderr
|
||||||
cwd=self._context.base_directory(),
|
|
||||||
enable_stdin=stdin,
|
|
||||||
enable_stdout=stdout,
|
|
||||||
enable_stderr=stderr
|
|
||||||
)
|
)
|
||||||
if ret != 0:
|
if ret != 0:
|
||||||
success = False
|
success = False
|
||||||
self._log.warning('Command [%s] failed' % cmd)
|
self._log.warning("Command [%s] failed" % cmd)
|
||||||
if success:
|
if success:
|
||||||
self._log.info('All commands have been executed')
|
self._log.info("All commands have been executed")
|
||||||
else:
|
else:
|
||||||
self._log.error('Some commands were not successfully executed')
|
self._log.error("Some commands were not successfully executed")
|
||||||
return success
|
return success
|
||||||
|
|
|
@ -4,12 +4,12 @@ import platform
|
||||||
|
|
||||||
|
|
||||||
def shell_command(command, cwd=None, enable_stdin=False, enable_stdout=False, enable_stderr=False):
|
def shell_command(command, cwd=None, enable_stdin=False, enable_stdout=False, enable_stderr=False):
|
||||||
with open(os.devnull, 'w') as devnull_w, open(os.devnull, 'r') as devnull_r:
|
with open(os.devnull, "w") as devnull_w, open(os.devnull, "r") as devnull_r:
|
||||||
stdin = None if enable_stdin else devnull_r
|
stdin = None if enable_stdin else devnull_r
|
||||||
stdout = None if enable_stdout else devnull_w
|
stdout = None if enable_stdout else devnull_w
|
||||||
stderr = None if enable_stderr else devnull_w
|
stderr = None if enable_stderr else devnull_w
|
||||||
executable = os.environ.get('SHELL')
|
executable = os.environ.get("SHELL")
|
||||||
if platform.system() == 'Windows':
|
if platform.system() == "Windows":
|
||||||
# We avoid setting the executable kwarg on Windows because it does
|
# We avoid setting the executable kwarg on Windows because it does
|
||||||
# not have the desired effect when combined with shell=True. It
|
# not have the desired effect when combined with shell=True. It
|
||||||
# will result in the correct program being run (e.g. bash), but it
|
# will result in the correct program being run (e.g. bash), but it
|
||||||
|
@ -24,11 +24,5 @@ def shell_command(command, cwd=None, enable_stdin=False, enable_stdout=False, en
|
||||||
# `bash -c "..."`.
|
# `bash -c "..."`.
|
||||||
executable = None
|
executable = None
|
||||||
return subprocess.call(
|
return subprocess.call(
|
||||||
command,
|
command, shell=True, executable=executable, stdin=stdin, stdout=stdout, stderr=stderr, cwd=cwd
|
||||||
shell=True,
|
|
||||||
executable=executable,
|
|
||||||
stdin=stdin,
|
|
||||||
stdout=stdout,
|
|
||||||
stderr=stderr,
|
|
||||||
cwd=cwd
|
|
||||||
)
|
)
|
||||||
|
|
|
@ -2,4 +2,5 @@ def with_metaclass(meta, *bases):
|
||||||
class metaclass(meta):
|
class metaclass(meta):
|
||||||
def __new__(cls, name, this_bases, d):
|
def __new__(cls, name, this_bases, d):
|
||||||
return meta(name, bases, d)
|
return meta(name, bases, d)
|
||||||
return type.__new__(metaclass, 'temporary_class', (), {})
|
|
||||||
|
return type.__new__(metaclass, "temporary_class", (), {})
|
||||||
|
|
|
@ -3,27 +3,33 @@ import sys, os.path
|
||||||
# We keep references to loaded modules so they don't get garbage collected.
|
# We keep references to loaded modules so they don't get garbage collected.
|
||||||
loaded_modules = []
|
loaded_modules = []
|
||||||
|
|
||||||
|
|
||||||
def load(path):
|
def load(path):
|
||||||
basename = os.path.basename(path)
|
basename = os.path.basename(path)
|
||||||
module_name, extension = os.path.splitext(basename)
|
module_name, extension = os.path.splitext(basename)
|
||||||
plugin = load_module(module_name, path)
|
plugin = load_module(module_name, path)
|
||||||
loaded_modules.append(plugin)
|
loaded_modules.append(plugin)
|
||||||
|
|
||||||
|
|
||||||
if sys.version_info >= (3, 5):
|
if sys.version_info >= (3, 5):
|
||||||
import importlib.util
|
import importlib.util
|
||||||
|
|
||||||
|
def load_module(module_name, path):
|
||||||
|
spec = importlib.util.spec_from_file_location(module_name, path)
|
||||||
|
module = importlib.util.module_from_spec(spec)
|
||||||
|
spec.loader.exec_module(module)
|
||||||
|
return module
|
||||||
|
|
||||||
|
|
||||||
def load_module(module_name, path):
|
|
||||||
spec = importlib.util.spec_from_file_location(module_name, path)
|
|
||||||
module = importlib.util.module_from_spec(spec)
|
|
||||||
spec.loader.exec_module(module)
|
|
||||||
return module
|
|
||||||
elif sys.version_info >= (3, 3):
|
elif sys.version_info >= (3, 3):
|
||||||
from importlib.machinery import SourceFileLoader
|
from importlib.machinery import SourceFileLoader
|
||||||
|
|
||||||
|
def load_module(module_name, path):
|
||||||
|
return SourceFileLoader(module_name, path).load_module()
|
||||||
|
|
||||||
|
|
||||||
def load_module(module_name, path):
|
|
||||||
return SourceFileLoader(module_name, path).load_module()
|
|
||||||
else:
|
else:
|
||||||
import imp
|
import imp
|
||||||
|
|
||||||
def load_module(module_name, path):
|
def load_module(module_name, path):
|
||||||
return imp.load_source(module_name, path)
|
return imp.load_source(module_name, path)
|
||||||
|
|
|
@ -1,5 +1,6 @@
|
||||||
class Singleton(type):
|
class Singleton(type):
|
||||||
_instances = {}
|
_instances = {}
|
||||||
|
|
||||||
def __call__(cls, *args, **kwargs):
|
def __call__(cls, *args, **kwargs):
|
||||||
if cls not in cls._instances:
|
if cls not in cls._instances:
|
||||||
cls._instances[cls] = super(Singleton, cls).__call__(*args, **kwargs)
|
cls._instances[cls] = super(Singleton, cls).__call__(*args, **kwargs)
|
||||||
|
|
|
@ -1,4 +1,4 @@
|
||||||
def indent_lines(string, amount=2, delimiter='\n'):
|
def indent_lines(string, amount=2, delimiter="\n"):
|
||||||
whitespace = ' ' * amount
|
whitespace = " " * amount
|
||||||
sep = '%s%s' % (delimiter, whitespace)
|
sep = "%s%s" % (delimiter, whitespace)
|
||||||
return '%s%s' % (whitespace, sep.join(string.split(delimiter)))
|
return "%s%s" % (whitespace, sep.join(string.split(delimiter)))
|
||||||
|
|
Loading…
Reference in a new issue