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 CHANGED
@@ -22,7 +22,7 @@
22
22
  import dataclasses as dc
23
23
  import logging
24
24
  from typing import Any, ClassVar, Dict
25
- from .task import TaskCtor
25
+ from .task_ctor import TaskCtor
26
26
 
27
27
  @dc.dataclass
28
28
  class Package(object):
@@ -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 TaskCtor, TaskCtorProxy, TaskCtorCls, TaskCtorParam, TaskCtorParamCls
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
- name : str
34
+ # @dc.dataclass
35
+ # class TaskSpec(object):
36
+ # name : str
34
37
 
35
- class TaskParams(BaseModel):
36
- pass
38
+ # class TaskParams(BaseModel):
39
+ # pass
37
40
 
38
41
 
39
- @dc.dataclass
40
- class TaskCtor(object):
41
- name : str
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
- return params
46
+ # _log : ClassVar = logging.getLogger("TaskCtorParam")
68
47
 
69
- def applyParams(self, params):
70
- if self.uses is not None:
71
- self.uses.applyParams(params)
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
- @dc.dataclass
75
- class TaskCtorParam(TaskCtor):
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
- @dc.dataclass
108
- class TaskCtorParamCls(TaskCtor):
109
- params_ctor : Callable = None
60
+ # return ret
110
61
 
111
- _log : ClassVar = logging.getLogger("TaskCtorParamType")
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
- def mkParams(self):
114
- self._log.debug("--> %s::mkParams" % str(self.name))
115
- params = self.params_ctor()
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
- @dc.dataclass
120
- class TaskCtorCls(TaskCtor):
121
- task_ctor : Callable = None
79
+ # _log : ClassVar = logging.getLogger("TaskCtorParamType")
122
80
 
123
- _log : ClassVar = logging.getLogger("TaskCtorCls")
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
- def mkTask(self, name : str, depends, rundir, srcdir=None, params=None):
126
- self._log.debug("--> %s::mkTask (%s) srcdir=%s" % (self.name, str(self.task_ctor), srcdir))
87
+ # @dc.dataclass
88
+ # class TaskCtorCls(TaskCtor):
89
+ # task_ctor : Callable = None
127
90
 
128
- if srcdir is None:
129
- srcdir = self.srcdir
91
+ # _log : ClassVar = logging.getLogger("TaskCtorCls")
130
92
 
131
- if params is None:
132
- params = self.mkParams()
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
- ret = self.task_ctor(
135
- name=name,
136
- depends=depends,
137
- rundir=rundir,
138
- srcdir=srcdir,
139
- params=params)
96
+ # if srcdir is None:
97
+ # srcdir = self.srcdir
140
98
 
141
- # Update parameters on the way back
142
- self.applyParams(ret.params)
99
+ # if params is None:
100
+ # params = self.mkParams()
143
101
 
144
- self._log.debug("<-- %s::mkTask" % self.name)
145
- return ret
102
+ # ret = self.task_ctor(
103
+ # name=name,
104
+ # depends=depends,
105
+ # rundir=rundir,
106
+ # srcdir=srcdir,
107
+ # params=params)
146
108
 
147
- @dc.dataclass
148
- class TaskCtorProxy(TaskCtor):
149
- task_ctor : TaskCtor = None
150
- param_ctor : Callable = None
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
- _log : ClassVar = logging.getLogger("TaskCtorProxy")
115
+ # @dc.dataclass
116
+ # class TaskCtorProxy(TaskCtor):
117
+ # task_ctor : TaskCtor = None
118
+ # param_ctor : Callable = None
153
119
 
154
- def mkTask(self, *args, **kwargs):
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
- def mkParams(self, params=None):
161
- self._log.debug("--> %s::mkParams: %s" % (self.name, str(self.params)))
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
- if params is None and self.param_ctor is not None:
164
- params = self.param_ctor()
128
+ # def mkParams(self, params=None):
129
+ # self._log.debug("--> %s::mkParams: %s" % (self.name, str(self.params)))
165
130
 
166
- params = self.task_ctor.mkParams(params)
131
+ # if params is None and self.param_ctor is not None:
132
+ # params = self.param_ctor()
167
133
 
168
- if self.params is not None:
169
- for k,v in self.params.items():
170
- self._log.debug(" change %s %s=>%s" % (
171
- k,
172
- str(getattr(params, k)),
173
- str(v)))
174
- setattr(params, k, v)
175
- self._log.debug("<-- %s::mkParams: %s" % (self.name, str(self.params)))
176
- return params
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
- # Need TaskDef
185
-
186
- params : TaskParams # Might need to be set of instructions for building...
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
- input : TaskDataInput,
198
- runner : 'TaskGraphRunner') -> TaskDataResult:
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
 
@@ -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[Any]
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, TaskCtor, TaskSpec
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]
@@ -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(object):
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, TaskSpec, TaskCtor
33
+ from .task import Task
34
34
  from .task_data import TaskData
35
35
  from .task_graph_runner import TaskGraphRunner
36
36
 
@@ -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
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.2
2
2
  Name: dv-flow-mgr
3
- Version: 0.0.1.13468776985a1
3
+ Version: 0.0.1.13489806632a1
4
4
  Summary: DV Flow Manager is a build system for silicon design
5
5
  Author-email: Matthew Ballance <matt.ballance@gmail.com>
6
6
  License: Apache License
@@ -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=MDnt9q4pD_Aex_hSK_EUQ38IiaxOihgxbMmk5dH1deQ,1674
9
- dv_flow/mgr/package_def.py,sha256=Sv3KS5pDysUrtPBWElCvmWt2pnkTipZcZGYwKr5N7Mc,14967
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=jZrxTevSulUQ56RdgD0TQXrQ05859FgmXahOdxhdyT8,6349
15
- dv_flow/mgr/task_data.py,sha256=TCc5KYuKoQNYfemVOzIlNKkMKIw-cqVl5_EmDs03W7M,10718
16
- dv_flow/mgr/task_def.py,sha256=6qDDgge5RXssL5TAZR01dGb_Vjvbj72tJirg66O-Yt4,1818
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=4dfy_T26MYI7ls2xiofr4V0ItZHDehU2iw6UBPt-SZQ,7074
19
- dv_flow/mgr/task_graph_runner.py,sha256=NwNYcOJ952lPMLwIIlYE9CoDdedqvcw2fWHYUsKFXuU,2164
20
- dv_flow/mgr/task_graph_runner_local.py,sha256=66JYJcu-W60LjfuT9UkbcKslNqDcD5q4UR7ZAWrSaGM,4707
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.13468776985a1.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
34
- dv_flow_mgr-0.0.1.13468776985a1.dist-info/METADATA,sha256=C2BOmVhhEQfwS9QQ9reuEWFsP9uWupVItzxnL5lilJc,13276
35
- dv_flow_mgr-0.0.1.13468776985a1.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
36
- dv_flow_mgr-0.0.1.13468776985a1.dist-info/entry_points.txt,sha256=1roy8wAFM48LabOvr6jiOw0MUs-qE8X3Vf8YykPazxk,50
37
- dv_flow_mgr-0.0.1.13468776985a1.dist-info/top_level.txt,sha256=amfVTkggzYPtWwLqNmRukfz1Buu0pGS2SrYBBLhXm04,8
38
- dv_flow_mgr-0.0.1.13468776985a1.dist-info/RECORD,,
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,,