qt.py 1.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142
  1. from __future__ import absolute_import
  2. from apscheduler.schedulers.base import BaseScheduler
  3. try:
  4. from PyQt5.QtCore import QObject, QTimer
  5. except ImportError: # pragma: nocover
  6. try:
  7. from PyQt4.QtCore import QObject, QTimer
  8. except ImportError:
  9. try:
  10. from PySide.QtCore import QObject, QTimer # flake8: noqa
  11. except ImportError:
  12. raise ImportError('QtScheduler requires either PyQt5, PyQt4 or PySide installed')
  13. class QtScheduler(BaseScheduler):
  14. """A scheduler that runs in a Qt event loop."""
  15. _timer = None
  16. def shutdown(self, *args, **kwargs):
  17. super(QtScheduler, self).shutdown(*args, **kwargs)
  18. self._stop_timer()
  19. def _start_timer(self, wait_seconds):
  20. self._stop_timer()
  21. if wait_seconds is not None:
  22. self._timer = QTimer.singleShot(wait_seconds * 1000, self._process_jobs)
  23. def _stop_timer(self):
  24. if self._timer:
  25. if self._timer.isActive():
  26. self._timer.stop()
  27. del self._timer
  28. def wakeup(self):
  29. self._start_timer(0)
  30. def _process_jobs(self):
  31. wait_seconds = super(QtScheduler, self)._process_jobs()
  32. self._start_timer(wait_seconds)