feat(scheduler): Add function to add scheduled tasks programmatically
This commit is contained in:
parent
691345cf40
commit
e1f919bd20
|
@ -17,19 +17,25 @@ class __Task:
|
|||
_scheduled_tasks: dict[__Task] = dict()
|
||||
|
||||
|
||||
def add_scheduled(id: str, function, replace=False, **kwargs):
|
||||
if id not in _scheduled_tasks or replace:
|
||||
_scheduled_tasks[id] = __Task(function, **kwargs)
|
||||
logger.info(f"Registered task: {id}")
|
||||
else:
|
||||
logger.debug(f"Skipping already registered task: {id}")
|
||||
|
||||
|
||||
def scheduled(id: str, replace=False, **kwargs):
|
||||
"""
|
||||
kwargs: days, hours, minutes
|
||||
"""
|
||||
|
||||
def real_decorator(function):
|
||||
if id not in _scheduled_tasks or replace:
|
||||
logger.info(f"Registered task: {id}")
|
||||
_scheduled_tasks[id] = __Task(function, **kwargs)
|
||||
else:
|
||||
logger.debug(f"Skipping already registered task: {id}")
|
||||
add_scheduled(id, function, replace, **kwargs)
|
||||
return function
|
||||
|
||||
if not isinstance(id, str):
|
||||
raise TypeError
|
||||
return real_decorator
|
||||
|
||||
|
||||
|
@ -75,6 +81,8 @@ class SchedulerPlugin(Plugin):
|
|||
)
|
||||
task.function()
|
||||
changed = True
|
||||
else:
|
||||
logger.debug(f"Skip task {id}, is scheduled for {last_run + task.interval}")
|
||||
if changed:
|
||||
# Remove not registered tasks
|
||||
for id in status.keys():
|
||||
|
|
Loading…
Reference in New Issue