mirror of
https://github.com/YunoHost/yunohost.git
synced 2024-09-03 20:06:10 +02:00
[enh] Add pre/post script execution callbacks to hook_callback
This commit is contained in:
parent
a7657c1a39
commit
cf077a50aa
2 changed files with 40 additions and 10 deletions
|
@ -1380,8 +1380,15 @@ hook:
|
|||
nargs: "*"
|
||||
-a:
|
||||
full: --args
|
||||
help: Ordered list of arguments to pass to the script
|
||||
help: Ordered list of arguments to pass to the scripts
|
||||
nargs: "*"
|
||||
-q:
|
||||
full: --no-trace
|
||||
help: Do not print each command that will be executed
|
||||
action: store_true
|
||||
-d:
|
||||
full: --chdir
|
||||
help: The directory from where the scripts will be executed
|
||||
|
||||
### hook_exec()
|
||||
exec:
|
||||
|
@ -1391,7 +1398,8 @@ hook:
|
|||
help: Path of the script to execute
|
||||
-a:
|
||||
full: --args
|
||||
help: Arguments to pass to the script
|
||||
help: Ordered list of arguments to pass to the script
|
||||
nargs: "*"
|
||||
--raise-on-error:
|
||||
help: Raise if the script returns a non-zero exit code
|
||||
action: store_true
|
||||
|
|
|
@ -205,14 +205,22 @@ def hook_list(action, list_by='name', show_info=False):
|
|||
return { 'hooks': result }
|
||||
|
||||
|
||||
def hook_callback(action, hooks=[], args=None):
|
||||
def hook_callback(action, hooks=[], args=None, no_trace=False, chdir=None,
|
||||
pre_callback=None, post_callback=None):
|
||||
"""
|
||||
Execute all scripts binded to an action
|
||||
|
||||
Keyword argument:
|
||||
action -- Action name
|
||||
hooks -- List of hooks names to execute
|
||||
args -- Ordered list of arguments to pass to the script
|
||||
args -- Ordered list of arguments to pass to the scripts
|
||||
no_trace -- Do not print each command that will be executed
|
||||
chdir -- The directory from where the scripts will be executed
|
||||
pre_callback -- An object to call before each script execution with
|
||||
(name, priority, path, args) as arguments and which must return
|
||||
the arguments to pass to the script
|
||||
post_callback -- An object to call after each script execution with
|
||||
(name, priority, path, succeed) as arguments
|
||||
|
||||
"""
|
||||
result = { 'succeed': {}, 'failed': {} }
|
||||
|
@ -258,20 +266,34 @@ def hook_callback(action, hooks=[], args=None):
|
|||
elif not isinstance(args, list):
|
||||
args = [args]
|
||||
|
||||
# Validate callbacks
|
||||
if not callable(pre_callback):
|
||||
pre_callback = lambda name, priority, path, args: args
|
||||
if not callable(post_callback):
|
||||
post_callback = lambda name, priority, path, succeed: None
|
||||
|
||||
# Iterate over hooks and execute them
|
||||
for priority in sorted(hooks_dict):
|
||||
for name, info in iter(hooks_dict[priority].items()):
|
||||
state = 'succeed'
|
||||
filename = '%s-%s' % (priority, name)
|
||||
path = info['path']
|
||||
try:
|
||||
hook_exec(info['path'], args=args, raise_on_error=True)
|
||||
hook_args = pre_callback(name=name, priority=priority,
|
||||
path=path, args=args)
|
||||
hook_exec(path, args=hook_args, chdir=chdir,
|
||||
no_trace=no_trace, raise_on_error=True)
|
||||
except MoulinetteError as e:
|
||||
logger.error(str(e))
|
||||
state = 'failed'
|
||||
logger.error(str(e))
|
||||
post_callback(name=name, priority=priority, path=path,
|
||||
succeed=False)
|
||||
else:
|
||||
post_callback(name=name, priority=priority, path=path,
|
||||
succeed=True)
|
||||
try:
|
||||
result[state][name].append(info['path'])
|
||||
result[state][name].append(path)
|
||||
except KeyError:
|
||||
result[state][name] = [info['path']]
|
||||
result[state][name] = [path]
|
||||
return result
|
||||
|
||||
|
||||
|
@ -282,7 +304,7 @@ def hook_exec(path, args=None, raise_on_error=False, no_trace=False,
|
|||
|
||||
Keyword argument:
|
||||
path -- Path of the script to execute
|
||||
args -- A list of arguments to pass to the script
|
||||
args -- Ordered list of arguments to pass to the script
|
||||
raise_on_error -- Raise if the script returns a non-zero exit code
|
||||
no_trace -- Do not print each command that will be executed
|
||||
chdir -- The directory from where the script will be executed
|
||||
|
|
Loading…
Add table
Reference in a new issue