4
4
from typing import Generator
5
5
from typing import List
6
6
from typing import Optional
7
- from typing import Set
8
7
from typing import Union
9
8
10
9
import attr
@@ -136,7 +135,7 @@ class LiveExecution:
136
135
_n_entries_in_table = attr .ib (type = int )
137
136
_verbose = attr .ib (type = int )
138
137
_editor_url_scheme = attr .ib (type = str )
139
- _running_tasks = attr .ib (factory = set , type = Set [str ])
138
+ _running_tasks = attr .ib (factory = dict , type = Dict [str , MetaTask ])
140
139
_reports = attr .ib (factory = list , type = List [Dict [str , Any ]])
141
140
142
141
@hookimpl (hookwrapper = True )
@@ -208,8 +207,13 @@ def _generate_table(self, reduce_table: bool, sort_table: bool) -> Optional[Tabl
208
207
),
209
208
Text (report ["outcome" ].symbol , style = report ["outcome" ].style ),
210
209
)
211
- for running_task in self ._running_tasks :
212
- table .add_row (running_task , "running" )
210
+ for task in self ._running_tasks .values ():
211
+ table .add_row (
212
+ format_task_id (
213
+ task , editor_url_scheme = self ._editor_url_scheme , short_name = True
214
+ ),
215
+ "running" ,
216
+ )
213
217
else :
214
218
table = None
215
219
@@ -224,12 +228,12 @@ def _update_table(
224
228
225
229
def update_running_tasks (self , new_running_task : MetaTask ) -> None :
226
230
"""Add a new running task."""
227
- self ._running_tasks . add ( new_running_task .short_name )
231
+ self ._running_tasks [ new_running_task .name ] = new_running_task
228
232
self ._update_table ()
229
233
230
234
def update_reports (self , new_report : ExecutionReport ) -> None :
231
235
"""Update the status of a running task by adding its report."""
232
- self ._running_tasks .remove (new_report .task .short_name )
236
+ self ._running_tasks .pop (new_report .task .name )
233
237
self ._reports .append (
234
238
{
235
239
"name" : new_report .task .short_name ,
0 commit comments