Skip to content

Commit e1de84c

Browse files
committed
use list for unique_task2name
1 parent 6a1d377 commit e1de84c

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

custom_components/pyscript/function.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -164,7 +164,10 @@ async def task_unique(name, kill_me=False):
164164
pass
165165
if curr_task in cls.our_tasks:
166166
cls.unique_name2task[name] = curr_task
167-
cls.unique_task2name[curr_task] = name
167+
if curr_task not in cls.unique_task2name:
168+
cls.unique_task2name[curr_task] = []
169+
if name not in cls.unique_task2name[curr_task]:
170+
cls.unique_task2name[curr_task].append(name)
168171

169172
return task_unique
170173

@@ -296,7 +299,8 @@ async def run_coro(cls, coro):
296299
_LOGGER.error("run_coro: got exception %s", traceback.format_exc(-1))
297300
finally:
298301
if task in cls.unique_task2name:
299-
del cls.unique_name2task[cls.unique_task2name[task]]
302+
for name in cls.unique_task2name[task]:
303+
del cls.unique_name2task[name]
300304
del cls.unique_task2name[task]
301305
if task in cls.task2context:
302306
del cls.task2context[task]

0 commit comments

Comments
 (0)