blocking.py 924 B

123456789101112131415161718192021222324252627282930313233
  1. from __future__ import absolute_import
  2. from threading import Event
  3. from apscheduler.schedulers.base import BaseScheduler, STATE_STOPPED
  4. from apscheduler.util import TIMEOUT_MAX
  5. class BlockingScheduler(BaseScheduler):
  6. """
  7. A scheduler that runs in the foreground
  8. (:meth:`~apscheduler.schedulers.base.BaseScheduler.start` will block).
  9. """
  10. _event = None
  11. def start(self, *args, **kwargs):
  12. self._event = Event()
  13. super(BlockingScheduler, self).start(*args, **kwargs)
  14. self._main_loop()
  15. def shutdown(self, wait=True):
  16. super(BlockingScheduler, self).shutdown(wait)
  17. self._event.set()
  18. def _main_loop(self):
  19. wait_seconds = TIMEOUT_MAX
  20. while self.state != STATE_STOPPED:
  21. self._event.wait(wait_seconds)
  22. self._event.clear()
  23. wait_seconds = self._process_jobs()
  24. def wakeup(self):
  25. self._event.set()