sporq/kmk_firmware
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Extend create_task to allow for re- and no-scheduling

Browse Source
This commit is contained in:
xs5871 2023-04-08 08:42:24 +00:00
parent b84cd0bdab
commit 62984d34f7
No known key found for this signature in database
GPG Key ID: 2A617DBC375C8352
1 changed files with 7 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
kmk/scheduler.py
View File

@ -32,21 +32,22 @@ class PeriodicTaskMeta:
def create_task(
func: Callable[[None], None],
func: [Callable[[None], None], Task, PeriodicTaskMeta],
*,
after_ms: int = 0,
period_ms: int = 0,
) -> [Task, PeriodicTaskMeta]:
if period_ms:
if isinstance(func, (Task, PeriodicTaskMeta)):
t = r = func
elif period_ms:
r = PeriodicTaskMeta(func, period_ms)
t = r._task
else:
t = r = Task(func)
if after_ms:
after_ms = ticks_add(ticks_ms(), after_ms)
_task_queue.push_sorted(t, after_ms)
else:
if after_ms > 0:
_task_queue.push_sorted(t, ticks_add(ticks_ms(), after_ms))
elif after_ms == 0:
_task_queue.push_head(t)
return r