Permalink
Fetching contributors…
Cannot retrieve contributors at this time
652 lines (535 sloc) 22.7 KB
"""Tasks are the main abstractions managed by doit"""
import types
import os
import sys
import inspect
from collections import OrderedDict
from functools import partial
from pathlib import PurePath
from .cmdparse import CmdOption, TaskParse
from .exceptions import CatchedException, InvalidTask
from .action import create_action, PythonAction
from .dependency import UptodateCalculator
def first_line(doc):
"""extract first non-blank line from text, to extract docstring title"""
if doc is not None:
for line in doc.splitlines():
striped = line.strip()
if striped:
return striped
return ''
class DelayedLoader(object):
"""contains info for delayed creation of tasks from a task-creator
:ivar creator: reference to task-creator function
:ivar task_dep: (str) name of task that should be executed before the
the loader call the creator function
:ivar basename: (str) basename used when creating tasks
This is used when doit creates new tasks to handle
tasks and targets specified on command line
:ivar target_regex: (str) regex for all targets that this loader tasks
will create
:ivar created: (bool) whether this creator was already executed or not
"""
def __init__(self, creator, executed=None, target_regex=None, creates=None):
self.creator = creator
self.task_dep = executed
self.basename = None
self.created = False
self.target_regex = target_regex
self.creates = creates[:] if creates else []
self.regex_groups = OrderedDict() # task_name:RegexGroup
# used to indicate that a task had DelayedLoader but was already created
DelayedLoaded = False
class Stream():
"""Control task output stream verbosity
:ivar verbosity: 0,1,2 see Task.execute
Priority is given by:
1) command line -> force_global=True
2) task value
3) other config (INI, DOIT_CONFIG)
"""
def __init__(self, verbosity, force_global=False):
self.force_global = force_global
if verbosity is not None:
self.verbosity = verbosity
else:
self.verbosity = Task.DEFAULT_VERBOSITY
self.force_global = False
def effective_verbosity(self, task_verbosity):
"""return effective verbosity used on task"""
if self.force_global:
return self.verbosity
elif task_verbosity is not None:
return task_verbosity
else:
return self.verbosity
@staticmethod
def _get_out_err(verbosity):
"""return tuple (out, err) streams to be used
Replace stream with None if stream should be captured
:param verbosity: (int)
"""
if verbosity == 0:
return (None, None)
elif verbosity == 1:
return (None, sys.stderr)
else:
return (sys.stdout, sys.stderr)
class Task(object):
"""Task
@ivar name string
@ivar actions: list - L{BaseAction}
@ivar clean_actions: list - L{BaseAction}
@ivar loader (DelayedLoader)
@ivar teardown (list - L{BaseAction})
@ivar targets: (list -string)
@ivar task_dep: (list - string)
@ivar wild_dep: (list - string) task dependency using wildcard *
@ivar file_dep: (set - string)
@ivar calc_dep: (set - string) reference to a task
@ivar dep_changed (list - string): list of file-dependencies that changed
(are not up_to_date). this must be set before
@ivar uptodate: (list - bool/None) use bool/computed value instead of
checking dependencies
@ivar value_savers (list - callables) that return dicts to be added to
task values. Always executed on main process.
To be used by `uptodate` implementations.
@ivar setup_tasks (list - string): references to task-names
@ivar subtask_of: (string) indicate this task is a subtask of task name
@ivar has_subtask: (bool) indicate this task has subtasks
@ivar result: (str) last action "result". used to check task-result-dep
@ivar values: (dict) values saved by task that might be used by other tasks
@ivar getargs: (dict) values from other tasks
@ivar doc: (string) task documentation
@ivar options: (dict) calculated params values (from getargs and taskopt)
@ivar taskopt: (cmdparse.CmdParse)
@ivar pos_arg: (str) name of parameter in action to receive positional
parameters from command line
@ivar pos_arg_val: (list - str) list of positional parameters values
@ivar custom_title: function reference that takes a task object as
parameter and returns a string.
"""
DEFAULT_VERBOSITY = 1
string_types = (str, )
# list of valid types/values for each task attribute.
valid_attr = {'basename': (string_types, ()),
'name': (string_types, ()),
'actions': ((list, tuple), (None,)),
'file_dep': ((list, tuple), ()),
'task_dep': ((list, tuple), ()),
'uptodate': ((list, tuple), ()),
'calc_dep': ((list, tuple), ()),
'targets': ((list, tuple), ()),
'setup': ((list, tuple), ()),
'clean': ((list, tuple), (True,)),
'teardown': ((list, tuple), ()),
'doc': (string_types, (None,)),
'params': ((list, tuple,), ()),
'pos_arg': (string_types, (None,)),
'verbosity': ((), (None, 0, 1, 2,)),
'getargs': ((dict,), ()),
'title': ((types.FunctionType,), (None,)),
'watch': ((list, tuple), ()),
}
def __init__(self, name, actions, file_dep=(), targets=(),
task_dep=(), uptodate=(),
calc_dep=(), setup=(), clean=(), teardown=(),
subtask_of=None, has_subtask=False,
doc=None, params=(), pos_arg=None,
verbosity=None, title=None, getargs=None,
watch=(), loader=None):
"""sanity checks and initialization
@param params: (list of dict for parameters) see cmdparse.CmdOption
"""
getargs = getargs or {} #default
self.check_attr(name, 'name', name, self.valid_attr['name'])
self.check_attr(name, 'actions', actions, self.valid_attr['actions'])
self.check_attr(name, 'file_dep', file_dep, self.valid_attr['file_dep'])
self.check_attr(name, 'task_dep', task_dep, self.valid_attr['task_dep'])
self.check_attr(name, 'uptodate', uptodate, self.valid_attr['uptodate'])
self.check_attr(name, 'calc_dep', calc_dep, self.valid_attr['calc_dep'])
self.check_attr(name, 'targets', targets, self.valid_attr['targets'])
self.check_attr(name, 'setup', setup, self.valid_attr['setup'])
self.check_attr(name, 'clean', clean, self.valid_attr['clean'])
self.check_attr(name, 'teardown', teardown, self.valid_attr['teardown'])
self.check_attr(name, 'doc', doc, self.valid_attr['doc'])
self.check_attr(name, 'params', params, self.valid_attr['params'])
self.check_attr(name, 'pos_arg', pos_arg,
self.valid_attr['pos_arg'])
self.check_attr(name, 'verbosity', verbosity,
self.valid_attr['verbosity'])
self.check_attr(name, 'getargs', getargs, self.valid_attr['getargs'])
self.check_attr(name, 'title', title, self.valid_attr['title'])
self.check_attr(name, 'watch', watch, self.valid_attr['watch'])
if '=' in name:
msg = "Task '{}': name must not use the char '=' (equal sign)."
raise InvalidTask(msg.format(name))
self.name = name
self.params = params # save just for use on command `info`
self.options = None
self.pos_arg = pos_arg
self.pos_arg_val = None # to be set when parsing command line
self.setup_tasks = list(setup)
# actions
self._action_instances = None
if actions is None:
self._actions = []
else:
self._actions = list(actions[:])
self._init_deps(file_dep, task_dep, calc_dep)
# loaders create an implicity task_dep
self.loader = loader
if self.loader and self.loader.task_dep:
self.task_dep.append(loader.task_dep)
uptodate = uptodate if uptodate else []
self.getargs = getargs
if self.getargs:
uptodate.extend(self._init_getargs())
self.value_savers = []
self.uptodate = self._init_uptodate(uptodate)
self.targets = self._init_targets(targets)
self.subtask_of = subtask_of
self.has_subtask = has_subtask
self.result = None
self.values = {}
self.verbosity = verbosity
self.custom_title = title
# clean
if clean is True:
self._remove_targets = True
self.clean_actions = ()
else:
self._remove_targets = False
self.clean_actions = [create_action(a, self, 'clean') for a in clean]
self.teardown = [create_action(a, self, 'teardown') for a in teardown]
self.doc = self._init_doc(doc)
self.watch = watch
# just indicate if actions were executed at all
self.executed = False
def _init_deps(self, file_dep, task_dep, calc_dep):
"""init for dependency related attributes"""
self.dep_changed = None
# file_dep
self.file_dep = set()
self._expand_file_dep(file_dep)
# task_dep
self.task_dep = []
self.wild_dep = []
if task_dep:
self._expand_task_dep(task_dep)
# calc_dep
self.calc_dep = set()
if calc_dep:
self._expand_calc_dep(calc_dep)
def _init_targets(self, items):
"""convert valid targets to `str`"""
targets = []
for target in items:
if isinstance(target, str):
targets.append(target)
elif isinstance(target, PurePath):
targets.append(str(target))
else:
msg = ("%s. target must be a str or Path from pathlib. " +
"Got '%r' (%s)")
raise InvalidTask(msg % (self.name, target, type(target)))
return targets
def _init_uptodate(self, items):
"""wrap uptodate callables"""
uptodate = []
for item in items:
# configure task
if hasattr(item, 'configure_task'):
item.configure_task(self)
# check/append uptodate value to task
if isinstance(item, bool) or item is None:
uptodate.append((item, None, None))
elif hasattr(item, '__call__'):
uptodate.append((item, [], {}))
elif isinstance(item, tuple):
call = item[0]
args = list(item[1]) if len(item) > 1 else []
kwargs = item[2] if len(item) > 2 else {}
uptodate.append((call, args, kwargs))
elif isinstance(item, str):
uptodate.append((item, [], {}))
else:
msg = ("%s. task invalid 'uptodate' item '%r'. " +
"Must be bool, None, str, callable or tuple " +
"(callable, args, kwargs).")
raise InvalidTask(msg % (self.name, item))
return uptodate
def _expand_file_dep(self, file_dep):
"""put input into file_dep"""
for dep in file_dep:
if isinstance(dep, str):
self.file_dep.add(dep)
elif isinstance(dep, PurePath):
self.file_dep.add(str(dep))
else:
msg = ("%s. file_dep must be a str or Path from pathlib. " +
"Got '%r' (%s)")
raise InvalidTask(msg % (self.name, dep, type(dep)))
def _expand_task_dep(self, task_dep):
"""convert task_dep input into actaul task_dep and wild_dep"""
for dep in task_dep:
if "*" in dep:
self.wild_dep.append(dep)
else:
self.task_dep.append(dep)
def _expand_calc_dep(self, calc_dep):
"""calc_dep input"""
for dep in calc_dep:
if dep not in self.calc_dep:
self.calc_dep.add(dep)
def _extend_uptodate(self, uptodate):
"""add/extend uptodate values"""
self.uptodate.extend(self._init_uptodate(uptodate))
# FIXME should support setup also
_expand_map = {
'task_dep': _expand_task_dep,
'file_dep': _expand_file_dep,
'calc_dep': _expand_calc_dep,
'uptodate': _extend_uptodate,
}
def update_deps(self, deps):
"""expand all kinds of dep input"""
for dep, dep_values in deps.items():
if dep not in self._expand_map:
continue
self._expand_map[dep](self, dep_values)
def init_options(self):
"""Put default values on options.
This will only be used, if params options were not passed
on the command line.
"""
if self.options is None:
taskcmd = TaskParse([CmdOption(opt) for opt in self.params])
# ignore positional parameters
self.options = taskcmd.parse('')[0]
def _init_getargs(self):
"""task getargs attribute define implicit task dependencies"""
check_result = set()
for arg_name, desc in self.getargs.items():
# tuple (task_id, key_name)
parts = desc
if isinstance(parts, str) or len(parts) != 2:
msg = ("Taskid '%s' - Invalid format for getargs of '%s'.\n" %
(self.name, arg_name) +
"Should be tuple with 2 elements " +
"('<taskid>', '<key-name>') got '%s'\n" % desc)
raise InvalidTask(msg)
if parts[0] not in self.setup_tasks:
check_result.add(parts[0])
return [result_dep(t, setup_dep=True) for t in check_result]
@staticmethod
def _init_doc(doc):
"""process task "doc" attribute"""
# store just first non-empty line as documentation string
return first_line(doc)
@staticmethod
def check_attr(task, attr, value, valid):
"""check input task attribute is correct type/value
@param task (string): task name
@param attr (string): attribute name
@param value: actual input from user
@param valid (list): of valid types/value accepted
@raises InvalidTask if invalid input
"""
if type(value) in valid[0]:
return
if value in valid[1]:
return
# input value didnt match any valid type/value, raise exception
msg = "Task '%s' attribute '%s' must be " % (task, attr)
accept = ", ".join([getattr(v, '__name__', str(v)) for v in
(valid[0] + valid[1])])
msg += "{%s} got:%r %s" % (accept, value, type(value))
raise InvalidTask(msg)
@property
def actions(self):
"""lazy creation of action instances"""
if self._action_instances is None:
self._action_instances = [
create_action(a, self, 'actions') for a in self._actions]
return self._action_instances
def save_extra_values(self):
"""run value_savers updating self.values"""
for value_saver in self.value_savers:
self.values.update(value_saver())
def overwrite_verbosity(self, stream):
self.verbosity = stream.effective_verbosity(self.verbosity)
def execute(self, stream):
"""Executes the task.
@return failure: see CmdAction.execute
"""
self.executed = True
self.init_options()
task_stdout, task_stderr = stream._get_out_err(self.verbosity)
for action in self.actions:
action_return = action.execute(task_stdout, task_stderr)
if isinstance(action_return, CatchedException):
return action_return
self.result = action.result
self.values.update(action.values)
def execute_teardown(self, stream):
"""Executes task's teardown
@return failure: see CmdAction.execute
"""
task_stdout, task_stderr = stream._get_out_err(self.verbosity)
for action in self.teardown:
action_return = action.execute(task_stdout, task_stderr)
if isinstance(action_return, CatchedException):
return action_return
def clean(self, outstream, dryrun):
"""Execute task's clean
@ivar outstream: 'write' output into this stream
@ivar dryrun (bool): if True clean tasks are not executed
(just print out what would be executed)
"""
self.init_options()
# if clean is True remove all targets
if self._remove_targets is True:
clean_targets(self, dryrun)
else:
# clean contains a list of actions...
for action in self.clean_actions:
msg = "%s - executing '%s'\n"
outstream.write(msg % (self.name, action))
# add extra arguments used by clean actions
execute_on_dryrun = False
if isinstance(action, PythonAction):
action_sig = inspect.signature(action.py_callable)
if 'dryrun' in action_sig.parameters:
execute_on_dryrun = True
action.kwargs['dryrun'] = dryrun
if (not dryrun) or execute_on_dryrun:
result = action.execute(out=outstream)
if isinstance(result, CatchedException):
sys.stderr.write(str(result))
def title(self):
"""String representation on output.
@return: (str) Task name and actions
"""
if self.custom_title:
return self.custom_title(self)
return self.name
def __repr__(self):
return "<Task: %s>"% self.name
def __getstate__(self):
"""remove attributes that never used on process that only execute tasks
"""
to_pickle = self.__dict__.copy()
# never executed in sub-process
to_pickle['uptodate'] = None
to_pickle['value_savers'] = None
# can be re-recreated on demand
to_pickle['_action_instances'] = None
return to_pickle
# when using multiprocessing Tasks are pickled.
def pickle_safe_dict(self):
"""remove attributes that might contain unpickleble content
mostly probably closures
"""
to_pickle = self.__dict__.copy()
del to_pickle['_actions']
del to_pickle['_action_instances']
del to_pickle['clean_actions']
del to_pickle['teardown']
del to_pickle['custom_title']
del to_pickle['value_savers']
del to_pickle['uptodate']
return to_pickle
def update_from_pickle(self, pickle_obj):
"""update self with data from pickled Task"""
self.__dict__.update(pickle_obj)
def __eq__(self, other):
return self.name == other.name
def __lt__(self, other):
"""used on default sorting of tasks (alphabetically by name)"""
return self.name < other.name
def dict_to_task(task_dict):
"""Create a task instance from dictionary.
The dictionary has the same format as returned by task-generators
from dodo files.
@param task_dict (dict): task representation as a dict.
@raise InvalidTask: If unexpected fields were passed in task_dict
"""
# check required fields
if 'actions' not in task_dict:
raise InvalidTask("Task %s must contain 'actions' field. %s" %
(task_dict['name'], task_dict))
# user friendly. dont go ahead with invalid input.
task_attrs = list(task_dict.keys())
valid_attrs = set(Task.valid_attr.keys())
for key in task_attrs:
if key not in valid_attrs:
raise InvalidTask("Task %s contains invalid field: '%s'"%
(task_dict['name'], key))
return Task(**task_dict)
def clean_targets(task, dryrun):
"""remove all targets from a task"""
files = [path for path in task.targets if os.path.isfile(path)]
dirs = [path for path in task.targets if os.path.isdir(path)]
# remove all files
for file_ in files:
print("%s - removing file '%s'" % (task.name, file_))
if not dryrun:
os.remove(file_)
# remove all directories (if empty)
for dir_ in dirs:
if os.listdir(dir_):
msg = "%s - cannot remove (it is not empty) '%s'"
print(msg % (task.name, dir_))
else:
msg = "%s - removing dir '%s'"
print(msg % (task.name, dir_))
if not dryrun:
os.rmdir(dir_)
def _return_param(val):
'''just return passed parameter - make a callable from any value'''
return val
# uptodate
class result_dep(UptodateCalculator):
"""check if result of the given task was modified
"""
def __init__(self, dep_task_name, setup_dep=False):
'''
:param setup_dep: controls if dependent task is task_dep or setup
'''
self.dep_name = dep_task_name
self.setup_dep = setup_dep
self.result_name = '_result:%s' % self.dep_name
def configure_task(self, task):
"""to be called by doit when create the task"""
# result_dep creates an implicit task_dep
if self.setup_dep:
task.setup_tasks.append(self.dep_name)
else:
task.task_dep.append(self.dep_name)
def _result_single(self):
"""get result from a single task"""
return self.get_val(self.dep_name, 'result:')
def _result_group(self, dep_task):
"""get result from a group task
the result is the combination of results of all sub-tasks
"""
prefix = dep_task.name + ":"
sub_tasks = {}
for sub in dep_task.task_dep:
if sub.startswith(prefix):
sub_tasks[sub] = self.get_val(sub, 'result:')
return sub_tasks
def __call__(self, task, values):
"""return True if result is the same as last run"""
dep_task = self.tasks_dict[self.dep_name]
if not dep_task.has_subtask:
dep_result = self._result_single()
else:
dep_result = self._result_group(dep_task)
func = partial(_return_param, {self.result_name: dep_result})
task.value_savers.append(func)
last_success = values.get(self.result_name)
if last_success is None:
return False
return last_success == dep_result