Source code for idmtools_platform_comps.ssmt_work_items.work_order

"""idmtools WorkOrder classes.

Copyright 2021, Bill & Melinda Gates Foundation. All rights reserved.
from abc import ABC
from dataclasses import dataclass, field
from typing import List, Dict

[docs]@dataclass() class IWorkOrder(ABC): """Base workorder type.""" WorkItem_Type: str = field()
[docs]@dataclass() class ExecutionDefinition: """Define the execution definition for workorders.""" Command: str = field() ImageName: str = field(default='DockerWorker')
[docs]@dataclass() class DockerWorkOrder(IWorkOrder): """Define the docker worker.""" WorkItem_Type: str = field(default='DockerWorker') Execution: ExecutionDefinition = field(default_factory=ExecutionDefinition)
[docs]@dataclass() class BuildFlags: """Define build flags.""" section: List[str] = field(default_factory=lambda: ['all']) library: str = field(default='') Switches: List['str'] = field(default_factory=lambda: [])
[docs]@dataclass() class BuildDefinition: """Define options for build definitions.""" Type: str = field(default='singularity') Input: str = field(default=None) Flags: BuildFlags = field(default_factory=BuildFlags)
[docs]@dataclass() class ImageBuilderWorkOrder(IWorkOrder): """Defines our Image Builder service workorder.""" WorkItem_Type: str = field(default='ImageBuilderWorker') Build: str = field(default=BuildDefinition()) Output: str = field(default='image.sif') Tags: Dict[str, str] = field(default_factory=lambda: dict(type='singularity')) AdditionalMounts: List[str] = field(default_factory=list) StaticEnvironment: Dict[str, str] = field(default_factory=dict)