idmtools.core.task_factory module#

Define our tasks factory. This is crucial to build tasks when fetching from the server.

Copyright 2021, Bill & Melinda Gates Foundation. All rights reserved.

class idmtools.core.task_factory.DynamicTaskSpecification(task_type: Type[ITask], description: str = '')[source]#

Bases: TaskSpecification

This class allows users to quickly define a spec for special tasks.

__init__(task_type: Type[ITask], description: str = '')[source]#

Initialize our specification.

Parameters:

  • task_type – Task type to register

  • description – Description to register with task

get(configuration: dict) → ITask[source]#

Get an instance of our task using configuration.

Parameters:

configuration – Configuration keyword args.

Returns:

Task with configuration specified

get_description() → str[source]#

Get description of our plugin.

Returns:

Returns the user-defined plugin description.

get_type() → Type[ITask][source]#

Get our task type.

Returns:

Returns our task type

class idmtools.core.task_factory.TaskFactory[source]#

Bases: object

TaskFactory allows creation of tasks that are derived from plugins.

DEFAULT_KEY = 'idmtools.entities.command_task.CommandTask'#
__init__()[source]#

Initialize our Factory.

register(spec: TaskSpecification) → NoReturn[source]#

Register a TaskSpecification dynamically.

Parameters:

spec – Specification to register

Returns:

None

register_task(task: Type[ITask]) → NoReturn[source]#

Dynamically register a class using the DynamicTaskSpecification.

Parameters:

task – Task to register

Returns:

None

create(key, fallback=None, **kwargs) → ITask[source]#

Create a task of type key.

Parameters:

  • key – Type of task to create

  • fallback – Fallback task type. Default to DEFAULT_KEY if not provided

  • **kwargs – Optional arguments to pass to the task

Returns:

Task with option specified