dv-flow-mgr 0.0.1.13468776985a1__py3-none-any.whl → 0.0.1.13489806632a1__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- dv_flow/mgr/package.py +1 -1
- dv_flow/mgr/package_def.py +2 -1
- dv_flow/mgr/task.py +113 -131
- dv_flow/mgr/task_ctor.py +43 -0
- dv_flow/mgr/task_data.py +15 -5
- dv_flow/mgr/task_def.py +1 -1
- dv_flow/mgr/task_graph_builder.py +3 -3
- dv_flow/mgr/task_graph_runner.py +4 -5
- dv_flow/mgr/task_graph_runner_local.py +1 -1
- dv_flow/mgr/task_node.py +80 -0
- dv_flow/mgr/task_params_ctor.py +31 -0
- dv_flow/mgr/task_runner.py +162 -0
- {dv_flow_mgr-0.0.1.13468776985a1.dist-info → dv_flow_mgr-0.0.1.13489806632a1.dist-info}/METADATA +1 -1
- {dv_flow_mgr-0.0.1.13468776985a1.dist-info → dv_flow_mgr-0.0.1.13489806632a1.dist-info}/RECORD +18 -14
- {dv_flow_mgr-0.0.1.13468776985a1.dist-info → dv_flow_mgr-0.0.1.13489806632a1.dist-info}/LICENSE +0 -0
- {dv_flow_mgr-0.0.1.13468776985a1.dist-info → dv_flow_mgr-0.0.1.13489806632a1.dist-info}/WHEEL +0 -0
- {dv_flow_mgr-0.0.1.13468776985a1.dist-info → dv_flow_mgr-0.0.1.13489806632a1.dist-info}/entry_points.txt +0 -0
- {dv_flow_mgr-0.0.1.13468776985a1.dist-info → dv_flow_mgr-0.0.1.13489806632a1.dist-info}/top_level.txt +0 -0
dv_flow/mgr/package.py
CHANGED
dv_flow/mgr/package_def.py
CHANGED
@@ -33,7 +33,8 @@ from typing import Any, Dict, List, Callable, Tuple, ClassVar
|
|
33
33
|
from .fragment_def import FragmentDef
|
34
34
|
from .package import Package
|
35
35
|
from .package_import_spec import PackageImportSpec, PackageSpec
|
36
|
-
from .task import
|
36
|
+
#from .task import TaskCtorCls, TaskCtorParam, TaskCtorParamCls
|
37
|
+
from .task_ctor import TaskCtor
|
37
38
|
from .task_def import TaskDef, TaskSpec
|
38
39
|
from .std.task_null import TaskNull
|
39
40
|
from .type_def import TypeDef
|
dv_flow/mgr/task.py
CHANGED
@@ -27,174 +27,156 @@ from pydantic import BaseModel
|
|
27
27
|
from typing import Any, Callable, ClassVar, Dict, List, Tuple
|
28
28
|
from .task_data import TaskData, TaskDataInput, TaskDataOutput, TaskDataResult
|
29
29
|
from .task_memento import TaskMemento
|
30
|
+
from .task_ctor import TaskParamsCtor
|
31
|
+
from .task_params_ctor import TaskParamsCtor
|
32
|
+
from .task_runner import TaskRunner
|
30
33
|
|
31
|
-
@dc.dataclass
|
32
|
-
class TaskSpec(object):
|
33
|
-
|
34
|
+
# @dc.dataclass
|
35
|
+
# class TaskSpec(object):
|
36
|
+
# name : str
|
34
37
|
|
35
|
-
class TaskParams(BaseModel):
|
36
|
-
|
38
|
+
# class TaskParams(BaseModel):
|
39
|
+
# pass
|
37
40
|
|
38
41
|
|
39
|
-
@dc.dataclass
|
40
|
-
class TaskCtor
|
41
|
-
|
42
|
-
uses : 'TaskCtor' = None
|
43
|
-
srcdir : str = None
|
44
|
-
depends : List[TaskSpec] = dc.field(default_factory=list)
|
45
|
-
|
46
|
-
_log : ClassVar = logging.getLogger("TaskCtor")
|
47
|
-
|
48
|
-
def mkTask(self, name : str, depends, rundir, srcdir=None, params=None):
|
49
|
-
if srcdir is None:
|
50
|
-
srcdir = self.srcdir
|
51
|
-
if params is None:
|
52
|
-
params = self.mkParams()
|
53
|
-
|
54
|
-
if self.uses is not None:
|
55
|
-
return self.uses.mkTask(name, depends, rundir, srcdir, params)
|
56
|
-
else:
|
57
|
-
raise NotImplementedError("TaskCtor.mkTask() not implemented for %s" % str(type(self)))
|
58
|
-
|
59
|
-
def mkParams(self):
|
60
|
-
self._log.debug("--> %s::mkParams" % self.name)
|
61
|
-
if self.uses is not None:
|
62
|
-
params = self.uses.mkParams()
|
63
|
-
else:
|
64
|
-
params = TaskParams()
|
65
|
-
self._log.debug("<-- %s::mkParams: %s" % (self.name, str(params)))
|
42
|
+
# @dc.dataclass
|
43
|
+
# class TaskCtorParam(TaskCtor):
|
44
|
+
# params : Dict[str,Any] = dc.field(default_factory=dict)
|
66
45
|
|
67
|
-
|
46
|
+
# _log : ClassVar = logging.getLogger("TaskCtorParam")
|
68
47
|
|
69
|
-
|
70
|
-
|
71
|
-
|
48
|
+
# def mkTask(self, name : str, depends, rundir, srcdir=None, params=None):
|
49
|
+
# self._log.debug("--> %s::mkTask" % self.name)
|
50
|
+
# if params is None:
|
51
|
+
# params = self.mkParams()
|
52
|
+
# if srcdir is None:
|
53
|
+
# srcdir = self.srcdir
|
72
54
|
|
55
|
+
# ret = self.uses.mkTask(name, depends, rundir, srcdir, params)
|
73
56
|
|
74
|
-
|
75
|
-
|
76
|
-
params : Dict[str,Any] = dc.field(default_factory=dict)
|
77
|
-
|
78
|
-
_log : ClassVar = logging.getLogger("TaskCtorParam")
|
79
|
-
|
80
|
-
def mkTask(self, name : str, depends, rundir, srcdir=None, params=None):
|
81
|
-
self._log.debug("--> %s::mkTask" % self.name)
|
82
|
-
if params is None:
|
83
|
-
params = self.mkParams()
|
84
|
-
if srcdir is None:
|
85
|
-
srcdir = self.srcdir
|
86
|
-
|
87
|
-
ret = self.uses.mkTask(name, depends, rundir, srcdir, params)
|
88
|
-
|
89
|
-
self.applyParams(ret.params)
|
90
|
-
self._log.debug("<-- %s::mkTask" % self.name)
|
91
|
-
|
92
|
-
return ret
|
93
|
-
|
94
|
-
def applyParams(self, params):
|
95
|
-
self._log.debug("--> %s::applyParams: %s %s" % (self.name, str(type(self.params)), str(type(params))))
|
96
|
-
if self.params is not None:
|
97
|
-
for k,v in self.params.items():
|
98
|
-
self._log.debug(" change %s %s=>%s" % (
|
99
|
-
k,
|
100
|
-
str(getattr(params, k)),
|
101
|
-
str(v)))
|
102
|
-
setattr(params, k, v)
|
103
|
-
else:
|
104
|
-
self._log.debug(" no params")
|
105
|
-
self._log.debug("<-- %s::applyParams: %s" % (self.name, str(self.params)))
|
57
|
+
# self.applyParams(ret.params)
|
58
|
+
# self._log.debug("<-- %s::mkTask" % self.name)
|
106
59
|
|
107
|
-
|
108
|
-
class TaskCtorParamCls(TaskCtor):
|
109
|
-
params_ctor : Callable = None
|
60
|
+
# return ret
|
110
61
|
|
111
|
-
|
62
|
+
# def applyParams(self, params):
|
63
|
+
# self._log.debug("--> %s::applyParams: %s %s" % (self.name, str(type(self.params)), str(type(params))))
|
64
|
+
# if self.params is not None:
|
65
|
+
# for k,v in self.params.items():
|
66
|
+
# self._log.debug(" change %s %s=>%s" % (
|
67
|
+
# k,
|
68
|
+
# str(getattr(params, k)),
|
69
|
+
# str(v)))
|
70
|
+
# setattr(params, k, v)
|
71
|
+
# else:
|
72
|
+
# self._log.debug(" no params")
|
73
|
+
# self._log.debug("<-- %s::applyParams: %s" % (self.name, str(self.params)))
|
112
74
|
|
113
|
-
|
114
|
-
|
115
|
-
|
116
|
-
self._log.debug("<-- %s::mkParams: %s" % (str(self.name), str(type(params))))
|
117
|
-
return params
|
75
|
+
# @dc.dataclass
|
76
|
+
# class TaskCtorParamCls(TaskCtor):
|
77
|
+
# params_ctor : Callable = None
|
118
78
|
|
119
|
-
|
120
|
-
class TaskCtorCls(TaskCtor):
|
121
|
-
task_ctor : Callable = None
|
79
|
+
# _log : ClassVar = logging.getLogger("TaskCtorParamType")
|
122
80
|
|
123
|
-
|
81
|
+
# def mkParams(self):
|
82
|
+
# self._log.debug("--> %s::mkParams" % str(self.name))
|
83
|
+
# params = self.params_ctor()
|
84
|
+
# self._log.debug("<-- %s::mkParams: %s" % (str(self.name), str(type(params))))
|
85
|
+
# return params
|
124
86
|
|
125
|
-
|
126
|
-
|
87
|
+
# @dc.dataclass
|
88
|
+
# class TaskCtorCls(TaskCtor):
|
89
|
+
# task_ctor : Callable = None
|
127
90
|
|
128
|
-
|
129
|
-
srcdir = self.srcdir
|
91
|
+
# _log : ClassVar = logging.getLogger("TaskCtorCls")
|
130
92
|
|
131
|
-
|
132
|
-
|
93
|
+
# def mkTask(self, name : str, depends, rundir, srcdir=None, params=None):
|
94
|
+
# self._log.debug("--> %s::mkTask (%s) srcdir=%s" % (self.name, str(self.task_ctor), srcdir))
|
133
95
|
|
134
|
-
|
135
|
-
|
136
|
-
depends=depends,
|
137
|
-
rundir=rundir,
|
138
|
-
srcdir=srcdir,
|
139
|
-
params=params)
|
96
|
+
# if srcdir is None:
|
97
|
+
# srcdir = self.srcdir
|
140
98
|
|
141
|
-
|
142
|
-
|
99
|
+
# if params is None:
|
100
|
+
# params = self.mkParams()
|
143
101
|
|
144
|
-
|
145
|
-
|
102
|
+
# ret = self.task_ctor(
|
103
|
+
# name=name,
|
104
|
+
# depends=depends,
|
105
|
+
# rundir=rundir,
|
106
|
+
# srcdir=srcdir,
|
107
|
+
# params=params)
|
146
108
|
|
147
|
-
|
148
|
-
|
149
|
-
|
150
|
-
|
109
|
+
# # Update parameters on the way back
|
110
|
+
# self.applyParams(ret.params)
|
111
|
+
|
112
|
+
# self._log.debug("<-- %s::mkTask" % self.name)
|
113
|
+
# return ret
|
151
114
|
|
152
|
-
|
115
|
+
# @dc.dataclass
|
116
|
+
# class TaskCtorProxy(TaskCtor):
|
117
|
+
# task_ctor : TaskCtor = None
|
118
|
+
# param_ctor : Callable = None
|
153
119
|
|
154
|
-
|
155
|
-
self._log.debug("--> %s::mkTask" % self.name)
|
156
|
-
ret = self.task_ctor.mkTask(*args, **kwargs)
|
157
|
-
self._log.debug("<-- %s::mkTask" % self.name)
|
158
|
-
return ret
|
120
|
+
# _log : ClassVar = logging.getLogger("TaskCtorProxy")
|
159
121
|
|
160
|
-
|
161
|
-
|
122
|
+
# def mkTask(self, *args, **kwargs):
|
123
|
+
# self._log.debug("--> %s::mkTask" % self.name)
|
124
|
+
# ret = self.task_ctor.mkTask(*args, **kwargs)
|
125
|
+
# self._log.debug("<-- %s::mkTask" % self.name)
|
126
|
+
# return ret
|
162
127
|
|
163
|
-
|
164
|
-
|
128
|
+
# def mkParams(self, params=None):
|
129
|
+
# self._log.debug("--> %s::mkParams: %s" % (self.name, str(self.params)))
|
165
130
|
|
166
|
-
|
131
|
+
# if params is None and self.param_ctor is not None:
|
132
|
+
# params = self.param_ctor()
|
167
133
|
|
168
|
-
|
169
|
-
|
170
|
-
|
171
|
-
|
172
|
-
|
173
|
-
|
174
|
-
|
175
|
-
|
176
|
-
|
134
|
+
# params = self.task_ctor.mkParams(params)
|
135
|
+
|
136
|
+
# if self.params is not None:
|
137
|
+
# for k,v in self.params.items():
|
138
|
+
# self._log.debug(" change %s %s=>%s" % (
|
139
|
+
# k,
|
140
|
+
# str(getattr(params, k)),
|
141
|
+
# str(v)))
|
142
|
+
# setattr(params, k, v)
|
143
|
+
# self._log.debug("<-- %s::mkParams: %s" % (self.name, str(self.params)))
|
144
|
+
# return params
|
177
145
|
|
178
146
|
|
179
147
|
@dc.dataclass
|
180
148
|
class Task(object):
|
181
149
|
"""Executable view of a task"""
|
150
|
+
# Ctor fields -- must specify on construction
|
182
151
|
name : str
|
152
|
+
srcdir : str
|
153
|
+
params : TaskParamsCtor
|
183
154
|
|
184
|
-
#
|
185
|
-
|
186
|
-
|
187
|
-
# This needs to be 'inside out' -- listed bottom to top
|
188
|
-
srcdir : str = dc.field(default=None)
|
189
|
-
rundir : str = dc.field(default=None)
|
190
|
-
depends : List['Task'] = dc.field(default_factory=list)
|
155
|
+
# Runtime fields -- these get populated during execution
|
156
|
+
changed : bool = False
|
157
|
+
needs : List['Task'] = dc.field(default_factory=list)
|
191
158
|
dependents : List['Task'] = dc.field(default_factory=list)
|
159
|
+
rundir : str = dc.field(default=None)
|
192
160
|
output : TaskDataOutput = dc.field(default=None)
|
193
161
|
|
194
162
|
_log : ClassVar = logging.getLogger("Task")
|
195
163
|
|
196
|
-
async def run(self,
|
197
|
-
|
198
|
-
|
164
|
+
async def run(self,
|
165
|
+
runner : 'TaskRunner',
|
166
|
+
input : TaskDataInput) -> TaskDataResult:
|
199
167
|
raise NotImplementedError("TaskImpl.run() not implemented")
|
168
|
+
|
169
|
+
@staticmethod
|
170
|
+
def ctor(paramT):
|
171
|
+
def wrapper(T):
|
172
|
+
def mk_task(name, srcdir, params):
|
173
|
+
return T(name, srcdir, params)
|
174
|
+
def mk_params(**params):
|
175
|
+
ctor = TaskParamsCtor(paramT)
|
176
|
+
ctor.values.append(paramT(**params))
|
177
|
+
return ctor
|
178
|
+
T.mkTask = mk_task
|
179
|
+
T.mkParams = mk_params
|
180
|
+
return T
|
181
|
+
return wrapper
|
200
182
|
|
dv_flow/mgr/task_ctor.py
ADDED
@@ -0,0 +1,43 @@
|
|
1
|
+
import os
|
2
|
+
import json
|
3
|
+
import dataclasses as dc
|
4
|
+
import logging
|
5
|
+
from pydantic import BaseModel
|
6
|
+
from typing import Any, Callable, ClassVar, Dict, List, Tuple
|
7
|
+
from .task_data import TaskDataOutput, TaskDataResult
|
8
|
+
|
9
|
+
# TaskParamsCtor accepts an evaluation context and returns a task parameter object
|
10
|
+
TaskParamsCtor = Callable[[object], Any]
|
11
|
+
|
12
|
+
@dc.dataclass
|
13
|
+
class TaskCtor(object):
|
14
|
+
name : str
|
15
|
+
uses : 'TaskCtor' = None
|
16
|
+
srcdir : str = None
|
17
|
+
|
18
|
+
_log : ClassVar = logging.getLogger("TaskCtor")
|
19
|
+
|
20
|
+
def mkTask(self, name : str, srcdir=None) -> 'Task':
|
21
|
+
"""Creates a task object"""
|
22
|
+
if srcdir is None:
|
23
|
+
srcdir = self.srcdir
|
24
|
+
|
25
|
+
if self.uses is not None:
|
26
|
+
return self.uses.mkTask(name, srcdir)
|
27
|
+
else:
|
28
|
+
raise NotImplementedError("TaskCtor.mkTask() not implemented for %s" % str(type(self)))
|
29
|
+
|
30
|
+
def mkParams(self, params=None) -> TaskParamsCtor:
|
31
|
+
"""Creates a list of task-parameter objects used produce the params object"""
|
32
|
+
self._log.debug("--> %s::mkParams" % self.name)
|
33
|
+
if self.uses is not None:
|
34
|
+
params = self.uses.mkParams()
|
35
|
+
else:
|
36
|
+
params = TaskParams()
|
37
|
+
self._log.debug("<-- %s::mkParams: %s" % (self.name, str(params)))
|
38
|
+
|
39
|
+
return params
|
40
|
+
|
41
|
+
def applyParams(self, params):
|
42
|
+
if self.uses is not None:
|
43
|
+
self.uses.applyParams(params)
|
dv_flow/mgr/task_data.py
CHANGED
@@ -27,6 +27,16 @@ from typing import Any, ClassVar, Dict, Set, List, Tuple
|
|
27
27
|
from .fileset import FileSet
|
28
28
|
from toposort import toposort
|
29
29
|
|
30
|
+
class TaskMarkerLoc(BaseModel):
|
31
|
+
path : str
|
32
|
+
line : int = dc.Field(default=-1)
|
33
|
+
pos : int = dc.Field(default=-1)
|
34
|
+
|
35
|
+
class TaskMarker(BaseModel):
|
36
|
+
msg : str
|
37
|
+
severity : str
|
38
|
+
loc : TaskMarkerLoc = dc.Field(default=None)
|
39
|
+
|
30
40
|
class TaskDataInput(BaseModel):
|
31
41
|
changed : bool
|
32
42
|
srcdir : str
|
@@ -35,11 +45,11 @@ class TaskDataInput(BaseModel):
|
|
35
45
|
memento : Any
|
36
46
|
|
37
47
|
class TaskDataResult(BaseModel):
|
38
|
-
changed : bool
|
39
|
-
output : List[Any]
|
40
|
-
memento : Any
|
41
|
-
markers : List[
|
42
|
-
status : int
|
48
|
+
changed : bool = dc.Field(default=True)
|
49
|
+
output : List[Any] = dc.Field(default_factory=list)
|
50
|
+
memento : Any = dc.Field(default=None)
|
51
|
+
markers : List[TaskMarker] = dc.Field(default_factory=list)
|
52
|
+
status : int = dc.Field(default=0)
|
43
53
|
|
44
54
|
class TaskDataOutput(BaseModel):
|
45
55
|
changed : bool
|
dv_flow/mgr/task_def.py
CHANGED
@@ -40,7 +40,7 @@ class TaskDef(BaseModel):
|
|
40
40
|
doc : str = dc.Field(default="")
|
41
41
|
depends : List[Union[str,TaskSpec]] = dc.Field(default_factory=list, alias="needs")
|
42
42
|
params: List[ParamDef] = dc.Field(default_factory=list, alias="with")
|
43
|
-
out: List[TaskOutput] = dc.Field(default_factory=list)
|
43
|
+
# out: List[TaskOutput] = dc.Field(default_factory=list)
|
44
44
|
|
45
45
|
def copy(self) -> 'TaskDef':
|
46
46
|
ret = TaskDef(
|
@@ -25,7 +25,7 @@ import logging
|
|
25
25
|
from .package import Package
|
26
26
|
from .package_def import PackageDef, PackageSpec
|
27
27
|
from .pkg_rgy import PkgRgy
|
28
|
-
from .task import Task
|
28
|
+
from .task import Task
|
29
29
|
from typing import Dict, List
|
30
30
|
|
31
31
|
@dc.dataclass
|
@@ -37,7 +37,7 @@ class TaskGraphBuilder(object):
|
|
37
37
|
_pkg_s : List[Package] = dc.field(default_factory=list)
|
38
38
|
_pkg_m : Dict[PackageSpec,Package] = dc.field(default_factory=dict)
|
39
39
|
_pkg_spec_s : List[PackageDef] = dc.field(default_factory=list)
|
40
|
-
_task_m : Dict[TaskSpec,Task] = dc.field(default_factory=dict)
|
40
|
+
_task_m : Dict['TaskSpec',Task] = dc.field(default_factory=dict)
|
41
41
|
_logger : logging.Logger = None
|
42
42
|
|
43
43
|
def __post_init__(self):
|
@@ -173,7 +173,7 @@ class TaskGraphBuilder(object):
|
|
173
173
|
|
174
174
|
return pkg
|
175
175
|
|
176
|
-
def getTaskCtor(self, spec : TaskSpec, pkg : PackageDef = None) -> 'TaskCtor':
|
176
|
+
def getTaskCtor(self, spec : 'TaskSpec', pkg : PackageDef = None) -> 'TaskCtor':
|
177
177
|
self._logger.debug("--> getTaskCtor %s" % spec.name)
|
178
178
|
spec_e = spec.name.split(".")
|
179
179
|
task_name = spec_e[-1]
|
dv_flow/mgr/task_graph_runner.py
CHANGED
@@ -21,20 +21,19 @@
|
|
21
21
|
#****************************************************************************
|
22
22
|
import asyncio
|
23
23
|
import os
|
24
|
-
import yaml
|
25
24
|
import dataclasses as dc
|
26
|
-
from typing import Any, Callable, ClassVar, Dict, List
|
25
|
+
from typing import Any, Callable, ClassVar, Dict, List, Tuple
|
27
26
|
from .task import Task
|
28
27
|
from .task_data import TaskData
|
28
|
+
from .task_runner import TaskRunner
|
29
29
|
|
30
30
|
@dc.dataclass
|
31
|
-
class TaskGraphRunner(
|
31
|
+
class TaskGraphRunner(TaskRunner):
|
32
32
|
"""Session manages execution of a task graph"""
|
33
33
|
|
34
|
-
rundir : str
|
35
|
-
|
36
34
|
_inst : ClassVar['TaskGraphRunner'] = None
|
37
35
|
|
36
|
+
|
38
37
|
# Search path for .dfs files
|
39
38
|
create_subprocess : Callable = asyncio.create_subprocess_exec
|
40
39
|
_root_dir : str = None
|
@@ -30,7 +30,7 @@ from .fragment_def import FragmentDef
|
|
30
30
|
from .package import Package
|
31
31
|
from .pkg_rgy import PkgRgy
|
32
32
|
from .package_def import PackageDef, PackageSpec
|
33
|
-
from .task import Task
|
33
|
+
from .task import Task
|
34
34
|
from .task_data import TaskData
|
35
35
|
from .task_graph_runner import TaskGraphRunner
|
36
36
|
|
dv_flow/mgr/task_node.py
ADDED
@@ -0,0 +1,80 @@
|
|
1
|
+
|
2
|
+
import dataclasses as dc
|
3
|
+
import logging
|
4
|
+
from typing import Any, Callable, ClassVar, Dict, List
|
5
|
+
from .task_data import TaskDataInput, TaskDataOutput, TaskDataResult
|
6
|
+
from .task_params_ctor import TaskParamsCtor
|
7
|
+
|
8
|
+
@dc.dataclass
|
9
|
+
class TaskNode(object):
|
10
|
+
"""Executable view of a task"""
|
11
|
+
# Ctor fields -- must specify on construction
|
12
|
+
name : str
|
13
|
+
srcdir : str
|
14
|
+
# This can be the resolved parameters
|
15
|
+
params : TaskParamsCtor
|
16
|
+
|
17
|
+
task : Callable[['TaskRunner','TaskDataInput'],'TaskDataResult']
|
18
|
+
|
19
|
+
# Runtime fields -- these get populated during execution
|
20
|
+
changed : bool = False
|
21
|
+
needs : List['TaskNode'] = dc.field(default_factory=list)
|
22
|
+
rundir : str = dc.field(default=None)
|
23
|
+
output : TaskDataOutput = dc.field(default=None)
|
24
|
+
|
25
|
+
_log : ClassVar = logging.getLogger("TaskNode")
|
26
|
+
|
27
|
+
def __hash__(self):
|
28
|
+
return id(self)
|
29
|
+
|
30
|
+
@staticmethod
|
31
|
+
def task(paramT):
|
32
|
+
def wrapper(T):
|
33
|
+
ctor = TaskNodeCtorWrapper(T.__name__, T, paramT)
|
34
|
+
return ctor
|
35
|
+
return wrapper
|
36
|
+
|
37
|
+
@dc.dataclass
|
38
|
+
class TaskNodeCtor(object):
|
39
|
+
"""
|
40
|
+
Factory for a specific task type
|
41
|
+
- Produces a task parameters object, applying value-setting instructions
|
42
|
+
- Produces a TaskNode
|
43
|
+
"""
|
44
|
+
name : str
|
45
|
+
|
46
|
+
|
47
|
+
def mkTaskNode(self, srcdir, params, name=None) -> TaskNode:
|
48
|
+
raise NotImplementedError("mkTaskNode in type %s" % str(type(self)))
|
49
|
+
|
50
|
+
def mkTaskParams(self, params : Dict) -> Any:
|
51
|
+
raise NotImplementedError("mkTaskParams in type %s" % str(type(self)))
|
52
|
+
|
53
|
+
@dc.dataclass
|
54
|
+
class TaskNodeCtorWrapper(TaskNodeCtor):
|
55
|
+
T : Any
|
56
|
+
paramT : Any
|
57
|
+
|
58
|
+
def __call__(self,
|
59
|
+
srcdir,
|
60
|
+
name=None,
|
61
|
+
params=None,
|
62
|
+
needs=None,
|
63
|
+
**kwargs):
|
64
|
+
"""Convenience method for direct creation of tasks"""
|
65
|
+
if params is None:
|
66
|
+
params = self.mkTaskParams(kwargs)
|
67
|
+
|
68
|
+
node = self.mkTaskNode(srcdir, params, name)
|
69
|
+
if needs is not None:
|
70
|
+
node.needs.extend(needs)
|
71
|
+
return node
|
72
|
+
|
73
|
+
def mkTaskNode(self, srcdir, params, name=None) -> TaskNode:
|
74
|
+
node = TaskNode(name, srcdir, params, self.T)
|
75
|
+
return node
|
76
|
+
|
77
|
+
def mkTaskParams(self, params : Dict) -> Any:
|
78
|
+
obj = self.paramT()
|
79
|
+
# TODO: apply user-specified params
|
80
|
+
return obj
|
@@ -0,0 +1,31 @@
|
|
1
|
+
import dataclasses as dc
|
2
|
+
from typing import Any, List
|
3
|
+
|
4
|
+
|
5
|
+
@dc.dataclass
|
6
|
+
class TaskParamsCtor(object):
|
7
|
+
|
8
|
+
paramT : Any = None
|
9
|
+
# List of parameter-setting sets, starting with inner-most
|
10
|
+
values : List[Any] = dc.field(default_factory=list)
|
11
|
+
|
12
|
+
def mk(self, input) -> Any:
|
13
|
+
params = self.paramT()
|
14
|
+
|
15
|
+
# Now, process each field
|
16
|
+
for field in dc.fields(self.paramT):
|
17
|
+
# Find the outer-most setting of the value
|
18
|
+
last_value_i = -1
|
19
|
+
for i in range(len(self.values)-1, -1, -1):
|
20
|
+
if hasattr(self.values[i], field.name) and getattr(self.values[i], field.name) is not None:
|
21
|
+
val = getattr(self.values[i], field.name)
|
22
|
+
# TODO: check if 'val' is a set or mutator
|
23
|
+
last_value_i = i
|
24
|
+
break
|
25
|
+
for i in range(last_value_i, -1, -1):
|
26
|
+
if hasattr(self.values[i], field.name):
|
27
|
+
val = getattr(self.values[i], field.name)
|
28
|
+
setattr(params, field.name, val)
|
29
|
+
|
30
|
+
print("params: %s" % str(params))
|
31
|
+
return params
|
@@ -0,0 +1,162 @@
|
|
1
|
+
import asyncio
|
2
|
+
import dataclasses as dc
|
3
|
+
from toposort import toposort
|
4
|
+
from typing import Any, Callable, List, Tuple, Union
|
5
|
+
from .task_data import TaskDataInput, TaskDataOutput, TaskDataResult
|
6
|
+
from .task_node import TaskNode
|
7
|
+
|
8
|
+
@dc.dataclass
|
9
|
+
class TaskRunner(object):
|
10
|
+
rundir : str
|
11
|
+
|
12
|
+
# List of [Listener:Callable[Task],Recurisve:bool]
|
13
|
+
listeners : List[Tuple[Callable['Task','Reason'], bool]] = dc.field(default_factory=list)
|
14
|
+
|
15
|
+
async def do_run(self,
|
16
|
+
task : 'Task',
|
17
|
+
memento : Any = None) -> 'TaskDataResult':
|
18
|
+
return await self.run(task, memento)
|
19
|
+
|
20
|
+
async def run(self,
|
21
|
+
task : 'Task',
|
22
|
+
memento : Any = None) -> 'TaskDataResult':
|
23
|
+
pass
|
24
|
+
|
25
|
+
@dc.dataclass
|
26
|
+
class TaskSetRunner(TaskRunner):
|
27
|
+
nproc : int = 8
|
28
|
+
|
29
|
+
async def run(self, task : Union[TaskNode,List[TaskNode]]):
|
30
|
+
# First, build a depedency map
|
31
|
+
tasks = task if isinstance(task, list) else [task]
|
32
|
+
dep_m = {}
|
33
|
+
for t in tasks:
|
34
|
+
self._buildDepMap(dep_m, t)
|
35
|
+
|
36
|
+
print("dep_m: %s" % str(dep_m))
|
37
|
+
|
38
|
+
order = list(toposort(dep_m))
|
39
|
+
|
40
|
+
active_task_l = []
|
41
|
+
done_task_s = set()
|
42
|
+
for active_s in order:
|
43
|
+
done = True
|
44
|
+
for t in active_s:
|
45
|
+
while len(active_task_l) >= self.nproc and t not in done_task_s:
|
46
|
+
# Wait for at least one job to complete
|
47
|
+
done, pending = await asyncio.wait(at[1] for at in active_task_l)
|
48
|
+
for d in done:
|
49
|
+
for i in range(len(active_task_l)):
|
50
|
+
if active_task_l[i][1] == d:
|
51
|
+
tt = active_task_l[i][0]
|
52
|
+
done_task_s.add(tt)
|
53
|
+
active_task_l.pop(i)
|
54
|
+
break
|
55
|
+
if t not in done_task_s:
|
56
|
+
coro = asyncio.Task(self.do_run(t))
|
57
|
+
active_task_l.append((t, coro))
|
58
|
+
|
59
|
+
# Now, wait for tasks to complete
|
60
|
+
if len(active_task_l):
|
61
|
+
coros = list(at[1] for at in active_task_l)
|
62
|
+
res = await asyncio.gather(*coros)
|
63
|
+
|
64
|
+
|
65
|
+
pass
|
66
|
+
|
67
|
+
async def do_run(self,
|
68
|
+
task : 'TaskNode',
|
69
|
+
memento : Any = None) -> 'TaskDataResult':
|
70
|
+
changed = False
|
71
|
+
for dep in task.needs:
|
72
|
+
changed |= dep.changed
|
73
|
+
|
74
|
+
# TODO: create an evaluator for substituting param values
|
75
|
+
eval = None
|
76
|
+
|
77
|
+
for field in dc.fields(task.params):
|
78
|
+
print("Field: %s" % field.name)
|
79
|
+
|
80
|
+
input = TaskDataInput(
|
81
|
+
changed=changed,
|
82
|
+
srcdir=task.srcdir,
|
83
|
+
rundir=self.rundir,
|
84
|
+
params=task.params,
|
85
|
+
memento=memento)
|
86
|
+
|
87
|
+
# TODO: notify of task start
|
88
|
+
ret : TaskDataResult = await task.task(self, input)
|
89
|
+
# TODO: notify of task complete
|
90
|
+
|
91
|
+
# Store the result
|
92
|
+
task.output = TaskDataOutput(
|
93
|
+
changed=ret.changed,
|
94
|
+
output=ret.output.copy())
|
95
|
+
|
96
|
+
# # By definition, none of this have run, since we just ran
|
97
|
+
# for dep in task.dependents:
|
98
|
+
# is_sat = True
|
99
|
+
# for need in dep.needs:
|
100
|
+
# if need.output is None:
|
101
|
+
# is_sat = False
|
102
|
+
# break
|
103
|
+
|
104
|
+
# if is_sat:
|
105
|
+
# # TODO: queue task for evaluation
|
106
|
+
# pass
|
107
|
+
# TODO:
|
108
|
+
|
109
|
+
return ret
|
110
|
+
|
111
|
+
def _buildDepMap(self, dep_m, task : TaskNode):
|
112
|
+
if task not in dep_m.keys():
|
113
|
+
dep_m[task] = set(task.needs)
|
114
|
+
for need in task.needs:
|
115
|
+
self._buildDepMap(dep_m, need)
|
116
|
+
|
117
|
+
@dc.dataclass
|
118
|
+
class SingleTaskRunner(TaskRunner):
|
119
|
+
|
120
|
+
async def run(self,
|
121
|
+
task : 'Task',
|
122
|
+
memento : Any = None) -> 'TaskDataResult':
|
123
|
+
changed = False
|
124
|
+
for dep in task.needs:
|
125
|
+
changed |= dep.changed
|
126
|
+
|
127
|
+
# TODO: create an evaluator for substituting param values
|
128
|
+
eval = None
|
129
|
+
|
130
|
+
for field in dc.fields(task.params):
|
131
|
+
print("Field: %s" % field.name)
|
132
|
+
|
133
|
+
input = TaskDataInput(
|
134
|
+
changed=changed,
|
135
|
+
srcdir=task.srcdir,
|
136
|
+
rundir=self.rundir,
|
137
|
+
params=task.params,
|
138
|
+
memento=memento)
|
139
|
+
|
140
|
+
# TODO: notify of task start
|
141
|
+
ret : TaskDataResult = await task.task(self, input)
|
142
|
+
# TODO: notify of task complete
|
143
|
+
|
144
|
+
# Store the result
|
145
|
+
task.output = TaskDataOutput(
|
146
|
+
changed=ret.changed,
|
147
|
+
output=ret.output.copy())
|
148
|
+
|
149
|
+
# # By definition, none of this have run, since we just ran
|
150
|
+
# for dep in task.dependents:
|
151
|
+
# is_sat = True
|
152
|
+
# for need in dep.needs:
|
153
|
+
# if need.output is None:
|
154
|
+
# is_sat = False
|
155
|
+
# break
|
156
|
+
|
157
|
+
# if is_sat:
|
158
|
+
# # TODO: queue task for evaluation
|
159
|
+
# pass
|
160
|
+
# TODO:
|
161
|
+
|
162
|
+
return ret
|
{dv_flow_mgr-0.0.1.13468776985a1.dist-info → dv_flow_mgr-0.0.1.13489806632a1.dist-info}/RECORD
RENAMED
@@ -5,22 +5,26 @@ dv_flow/mgr/expr_parser.py,sha256=aR8n6hgk5xmFknIaSVC02K-So-eU4pkkFtwb7zWsSPs,54
|
|
5
5
|
dv_flow/mgr/fileset.py,sha256=FNvC5sU2ArxJ0OO3v8dXTv8zX-bZ5t0a0ljne0fQQ1o,1150
|
6
6
|
dv_flow/mgr/fragment_def.py,sha256=cyzp1XeWtNOaagScmeS-BPsoXj9j2LTBbKq5ZUioz8I,1641
|
7
7
|
dv_flow/mgr/out,sha256=d8GGBi3J43fhdLBlnsUbzBfRe0TD0QTP3nOTz54l2bI,200
|
8
|
-
dv_flow/mgr/package.py,sha256=
|
9
|
-
dv_flow/mgr/package_def.py,sha256=
|
8
|
+
dv_flow/mgr/package.py,sha256=878twhPD-E1pFlDNUtuyeFEgJ_Y89b560og4St-Iwrs,1679
|
9
|
+
dv_flow/mgr/package_def.py,sha256=dwvs-7kGEp7wFrBK2Fybysb-vSKwtWOtxTrg1VIaL14,14975
|
10
10
|
dv_flow/mgr/package_import_spec.py,sha256=ah3r15v5Jdub2poc3sgi6Uar1L3oGoYsCPPNiOHV-a4,1760
|
11
11
|
dv_flow/mgr/param_def.py,sha256=e2WvRCMArbgcsKC4TKQqZTscZmCdo3WvVR6w3LN0VK8,727
|
12
12
|
dv_flow/mgr/parsetab.py,sha256=enSOnMQ-woIsMEzHyeYiefvhAl8enxfX9Ct_o8-jkqs,3780
|
13
13
|
dv_flow/mgr/pkg_rgy.py,sha256=2R_EaeBDJn5qUq9DzSnLc37wUP36MWSv-p0LgUjJRAg,4471
|
14
|
-
dv_flow/mgr/task.py,sha256=
|
15
|
-
dv_flow/mgr/
|
16
|
-
dv_flow/mgr/
|
14
|
+
dv_flow/mgr/task.py,sha256=kLQSvnVwj9ROIDtxq8lLu-4mJizTxOqvUeogmgN6QAA,5976
|
15
|
+
dv_flow/mgr/task_ctor.py,sha256=hlfl-UVvyjzLFN6D0Oel9eBs0xUQPqCX7gQ0uEHoL7o,1382
|
16
|
+
dv_flow/mgr/task_data.py,sha256=fkaVzQ81nJBXy4uPWPdnVCNoD9L7rvH5Abt8LXzvnJs,11098
|
17
|
+
dv_flow/mgr/task_def.py,sha256=WAW1fPXUfUiQcfmgCx0iCMddMBKATDa5RsRztJWRbUk,1819
|
17
18
|
dv_flow/mgr/task_exec_data.py,sha256=aT__kmVmNxqnS_GbTRS1vrYgKiv8cH-HUSmRb6YYKsI,640
|
18
|
-
dv_flow/mgr/task_graph_builder.py,sha256
|
19
|
-
dv_flow/mgr/task_graph_runner.py,sha256=
|
20
|
-
dv_flow/mgr/task_graph_runner_local.py,sha256=
|
19
|
+
dv_flow/mgr/task_graph_builder.py,sha256=-lRSjWU2UJf3euVIFtoVIU7Qdox7MI1sKERWg1k7U_g,7058
|
20
|
+
dv_flow/mgr/task_graph_runner.py,sha256=jUGI49QvxUCfQoKQDDk2psbeapIcCg72qNOW1JipHzM,2182
|
21
|
+
dv_flow/mgr/task_graph_runner_local.py,sha256=OrydPwtQ8E7hYWvSXx0h7lI3nfUNFyklULhsyMwz9dA,4687
|
21
22
|
dv_flow/mgr/task_impl_data.py,sha256=bFPijoKrh9x7fZN2DsvRJp0UHo-gGM0VjtDQISyfhFk,321
|
22
23
|
dv_flow/mgr/task_memento.py,sha256=C7VTQpBhDEoYuDmE6YTM-6TLMLnqHp6Y0Vat1aTgtCs,1096
|
24
|
+
dv_flow/mgr/task_node.py,sha256=wcvxdtrRNwPwV494DerrB79PFYAv6bPdWZDvRGQ0V6Q,2318
|
23
25
|
dv_flow/mgr/task_output.py,sha256=l-W-FvVo6YDah1RQS-I9N0KUtB3vp-kl7lxIdmNz0l4,178
|
26
|
+
dv_flow/mgr/task_params_ctor.py,sha256=aXgB8o9xFPjaEjGW_xYkEC0N0apzGzGUPDj7g2ZLvus,1112
|
27
|
+
dv_flow/mgr/task_runner.py,sha256=5vCKM2UFxdHB9OslUVIx5Ipc0r04Cizvrc2YSKGMqqI,5055
|
24
28
|
dv_flow/mgr/type.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
25
29
|
dv_flow/mgr/type_def.py,sha256=KdhuNlfw-NKU-4VZFCnMPyj775yEB7cpr5tz73a9yuQ,259
|
26
30
|
dv_flow/mgr/util.py,sha256=06eVyURF4ga-s8C9Sd3ZSDebwO4QS0XXaB8xADVbWRc,1437
|
@@ -30,9 +34,9 @@ dv_flow/mgr/std/fileset.py,sha256=uP7bGntRq-Tn5_GEFnt0_J_OAmfvep3GlCwCuE8by4o,27
|
|
30
34
|
dv_flow/mgr/std/flow.dv,sha256=j9wLrF3Ghh1ZLbJxmk7WiNiRYUYEer-8CCUA5hsgtfk,1409
|
31
35
|
dv_flow/mgr/std/message.py,sha256=BPTHnEMD4tBufQ9LvsS9Sa_0xjaJATbBpwqosWslvVA,193
|
32
36
|
dv_flow/mgr/std/task_null.py,sha256=KObmjG_4D08GJ1k6neqKIQrFY72Sj0jLnwXxEkq5HA0,321
|
33
|
-
dv_flow_mgr-0.0.1.
|
34
|
-
dv_flow_mgr-0.0.1.
|
35
|
-
dv_flow_mgr-0.0.1.
|
36
|
-
dv_flow_mgr-0.0.1.
|
37
|
-
dv_flow_mgr-0.0.1.
|
38
|
-
dv_flow_mgr-0.0.1.
|
37
|
+
dv_flow_mgr-0.0.1.13489806632a1.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
|
38
|
+
dv_flow_mgr-0.0.1.13489806632a1.dist-info/METADATA,sha256=iGQC00bCbcchWcesU4kXOBNBn-ggHWurn1YqLBKB610,13276
|
39
|
+
dv_flow_mgr-0.0.1.13489806632a1.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
|
40
|
+
dv_flow_mgr-0.0.1.13489806632a1.dist-info/entry_points.txt,sha256=1roy8wAFM48LabOvr6jiOw0MUs-qE8X3Vf8YykPazxk,50
|
41
|
+
dv_flow_mgr-0.0.1.13489806632a1.dist-info/top_level.txt,sha256=amfVTkggzYPtWwLqNmRukfz1Buu0pGS2SrYBBLhXm04,8
|
42
|
+
dv_flow_mgr-0.0.1.13489806632a1.dist-info/RECORD,,
|
{dv_flow_mgr-0.0.1.13468776985a1.dist-info → dv_flow_mgr-0.0.1.13489806632a1.dist-info}/LICENSE
RENAMED
File without changes
|
{dv_flow_mgr-0.0.1.13468776985a1.dist-info → dv_flow_mgr-0.0.1.13489806632a1.dist-info}/WHEEL
RENAMED
File without changes
|
File without changes
|
File without changes
|