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( def create_task(
func: Callable[[None], None], func: [Callable[[None], None], Task, PeriodicTaskMeta],
*, *,
after_ms: int = 0, after_ms: int = 0,
period_ms: int = 0, period_ms: int = 0,
) -> [Task, PeriodicTaskMeta]: ) -> [Task, PeriodicTaskMeta]:
if period_ms: if isinstance(func, (Task, PeriodicTaskMeta)):
t = r = func
elif period_ms:
r = PeriodicTaskMeta(func, period_ms) r = PeriodicTaskMeta(func, period_ms)
t = r._task t = r._task
else: else:
t = r = Task(func) t = r = Task(func)
if after_ms: if after_ms > 0:
after_ms = ticks_add(ticks_ms(), after_ms) _task_queue.push_sorted(t, ticks_add(ticks_ms(), after_ms))
_task_queue.push_sorted(t, after_ms) elif after_ms == 0:
else:
_task_queue.push_head(t) _task_queue.push_head(t)
return r return r