You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
34 lines
924 B
34 lines
924 B
6 years ago
|
from __future__ import absolute_import
|
||
|
|
|
||
|
|
from threading import Event
|
||
|
|
|
||
|
|
from apscheduler.schedulers.base import BaseScheduler, STATE_STOPPED
|
||
|
|
from apscheduler.util import TIMEOUT_MAX
|
||
|
|
|
||
|
|
|
||
|
|
class BlockingScheduler(BaseScheduler):
|
||
|
|
"""
|
||
|
|
A scheduler that runs in the foreground
|
||
|
|
(:meth:`~apscheduler.schedulers.base.BaseScheduler.start` will block).
|
||
|
|
"""
|
||
|
|
_event = None
|
||
|
|
|
||
|
|
def start(self, *args, **kwargs):
|
||
|
|
self._event = Event()
|
||
|
|
super(BlockingScheduler, self).start(*args, **kwargs)
|
||
|
|
self._main_loop()
|
||
|
|
|
||
|
|
def shutdown(self, wait=True):
|
||
|
|
super(BlockingScheduler, self).shutdown(wait)
|
||
|
|
self._event.set()
|
||
|
|
|
||
|
|
def _main_loop(self):
|
||
|
|
wait_seconds = TIMEOUT_MAX
|
||
|
|
while self.state != STATE_STOPPED:
|
||
|
|
self._event.wait(wait_seconds)
|
||
|
|
self._event.clear()
|
||
|
|
wait_seconds = self._process_jobs()
|
||
|
|
|
||
|
|
def wakeup(self):
|
||
|
|
self._event.set()
|