44from typing import Any , Callable , List
55
66from dotflow .log import logger
7- from dotflow .config import Config
7+ from dotflow .core . config import Config
88from dotflow .core .action import Action
99from dotflow .core .context import Context
1010from dotflow .core .module import Module
1111from dotflow .core .exception import MissingActionDecorator
1212from dotflow .core .models .status import TaskStatus
13- from dotflow .core .utils import (
14- basic_callback ,
15- traceback_error ,
16- message_error
17- )
13+ from dotflow .settings import Settings as settings
14+ from dotflow .core .utils import basic_callback , traceback_error , message_error , copy_file
1815
1916
2017class TaskInstance :
2118
22- def __init__ (
23- self ,
24- * args ,
25- ** kwargs
26- ) -> None :
19+ def __init__ (self , * args , ** kwargs ) -> None :
2720 self .task_id = None
2821 self .workflow_id = None
2922 self .step = None
@@ -45,15 +38,9 @@ def __init__(
4538 callback : Callable = basic_callback ,
4639 initial_context : Any = None ,
4740 workflow_id : UUID = None ,
48- config : Config = None
41+ config : Config = None ,
4942 ) -> None :
50- super ().__init__ (
51- task_id ,
52- step ,
53- callback ,
54- initial_context ,
55- workflow_id
56- )
43+ super ().__init__ (task_id , step , callback , initial_context , workflow_id )
5744 self .config = config
5845 self .task_id = task_id
5946 self .workflow_id = workflow_id
@@ -72,10 +59,7 @@ def status(self):
7259 def status (self , value : TaskStatus ) -> None :
7360 self ._status = value
7461
75- logger .info (
76- "ID %s - %s - %s" ,
77- self .workflow_id , self .task_id , value
78- )
62+ logger .info ("ID %s - %s - %s" , self .workflow_id , self .task_id , value )
7963
8064 @property
8165 def error (self ):
@@ -90,8 +74,10 @@ def error(self, value: Exception) -> None:
9074
9175 logger .error (
9276 "ID %s - %s - %s \n %s" ,
93- self .workflow_id , self .task_id ,
94- self .status , task_error .traceback
77+ self .workflow_id ,
78+ self .task_id ,
79+ self .status ,
80+ task_error .traceback ,
9581 )
9682
9783 @property
@@ -117,10 +103,12 @@ def current_context(self, value: Context):
117103 if self .config .output :
118104 logger .info (
119105 "ID %s - %s - Current Context -> %s" ,
120- self .workflow_id , self .task_id , str (value .storage )
106+ self .workflow_id ,
107+ self .task_id ,
108+ str (value .storage ),
121109 )
122110
123- self .config ._log_transfer ( )
111+ copy_file ( source = settings . LOG_PATH , destination = self .config .log_path )
124112
125113 @property
126114 def initial_context (self ):
@@ -135,10 +123,12 @@ def initial_context(self, value: Context):
135123 if self .config .output :
136124 logger .info (
137125 "ID %s - %s - Initial Context -> %s" ,
138- self .workflow_id , self .task_id , str (value )
126+ self .workflow_id ,
127+ self .task_id ,
128+ str (value ),
139129 )
140130
141- self .config ._log_transfer ( )
131+ copy_file ( source = settings . LOG_PATH , destination = self .config .log_path )
142132
143133 @property
144134 def config (self ):
@@ -167,11 +157,7 @@ def __init__(self, error: Exception = None) -> None:
167157
168158class TaskBuilder :
169159
170- def __init__ (
171- self ,
172- config : Config ,
173- workflow_id : UUID = None
174- ) -> None :
160+ def __init__ (self , config : Config , workflow_id : UUID = None ) -> None :
175161 self .queu : List [Task ] = []
176162 self .workflow_id = workflow_id
177163 self .config = config
@@ -180,7 +166,7 @@ def add(
180166 self ,
181167 step : Callable ,
182168 callback : Callable = basic_callback ,
183- initial_context : Any = None
169+ initial_context : Any = None ,
184170 ) -> None :
185171 step = Module (value = step )
186172
@@ -194,7 +180,7 @@ def add(
194180 callback = Module (value = callback ),
195181 initial_context = initial_context ,
196182 workflow_id = self .workflow_id ,
197- config = self .config
183+ config = self .config ,
198184 )
199185 )
200186
0 commit comments