Development of an internal social media platform with personalised dashboards for students
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.

events.py 4.0KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139
  1. # -*- coding: utf-8 -*-
  2. """
  3. The :program:`celery events` command.
  4. .. program:: celery events
  5. .. seealso::
  6. See :ref:`preload-options` and :ref:`daemon-options`.
  7. .. cmdoption:: -d, --dump
  8. Dump events to stdout.
  9. .. cmdoption:: -c, --camera
  10. Take snapshots of events using this camera.
  11. .. cmdoption:: --detach
  12. Camera: Detach and run in the background as a daemon.
  13. .. cmdoption:: -F, --freq, --frequency
  14. Camera: Shutter frequency. Default is every 1.0 seconds.
  15. .. cmdoption:: -r, --maxrate
  16. Camera: Optional shutter rate limit (e.g. 10/m).
  17. .. cmdoption:: -l, --loglevel
  18. Logging level, choose between `DEBUG`, `INFO`, `WARNING`,
  19. `ERROR`, `CRITICAL`, or `FATAL`. Default is INFO.
  20. """
  21. from __future__ import absolute_import, unicode_literals
  22. import sys
  23. from functools import partial
  24. from celery.platforms import detached, set_process_title, strargv
  25. from celery.bin.base import Command, Option, daemon_options
  26. __all__ = ['events']
  27. class events(Command):
  28. """Event-stream utilities.
  29. Commands::
  30. celery events --app=proj
  31. start graphical monitor (requires curses)
  32. celery events -d --app=proj
  33. dump events to screen.
  34. celery events -b amqp://
  35. celery events -c <camera> [options]
  36. run snapshot camera.
  37. Examples::
  38. celery events
  39. celery events -d
  40. celery events -c mod.attr -F 1.0 --detach --maxrate=100/m -l info
  41. """
  42. doc = __doc__
  43. supports_args = False
  44. def run(self, dump=False, camera=None, frequency=1.0, maxrate=None,
  45. loglevel='INFO', logfile=None, prog_name='celery events',
  46. pidfile=None, uid=None, gid=None, umask=None,
  47. working_directory=None, detach=False, **kwargs):
  48. self.prog_name = prog_name
  49. if dump:
  50. return self.run_evdump()
  51. if camera:
  52. return self.run_evcam(camera, freq=frequency, maxrate=maxrate,
  53. loglevel=loglevel, logfile=logfile,
  54. pidfile=pidfile, uid=uid, gid=gid,
  55. umask=umask,
  56. working_directory=working_directory,
  57. detach=detach)
  58. return self.run_evtop()
  59. def run_evdump(self):
  60. from celery.events.dumper import evdump
  61. self.set_process_status('dump')
  62. return evdump(app=self.app)
  63. def run_evtop(self):
  64. from celery.events.cursesmon import evtop
  65. self.set_process_status('top')
  66. return evtop(app=self.app)
  67. def run_evcam(self, camera, logfile=None, pidfile=None, uid=None,
  68. gid=None, umask=None, working_directory=None,
  69. detach=False, **kwargs):
  70. from celery.events.snapshot import evcam
  71. workdir = working_directory
  72. self.set_process_status('cam')
  73. kwargs['app'] = self.app
  74. cam = partial(evcam, camera,
  75. logfile=logfile, pidfile=pidfile, **kwargs)
  76. if detach:
  77. with detached(logfile, pidfile, uid, gid, umask, workdir):
  78. return cam()
  79. else:
  80. return cam()
  81. def set_process_status(self, prog, info=''):
  82. prog = '{0}:{1}'.format(self.prog_name, prog)
  83. info = '{0} {1}'.format(info, strargv(sys.argv))
  84. return set_process_title(prog, info=info)
  85. def get_options(self):
  86. return (
  87. (Option('-d', '--dump', action='store_true'),
  88. Option('-c', '--camera'),
  89. Option('--detach', action='store_true'),
  90. Option('-F', '--frequency', '--freq',
  91. type='float', default=1.0),
  92. Option('-r', '--maxrate'),
  93. Option('-l', '--loglevel', default='INFO')) +
  94. daemon_options(default_pidfile='celeryev.pid') +
  95. tuple(self.app.user_options['events'])
  96. )
  97. def main():
  98. ev = events()
  99. ev.execute_from_commandline()
  100. if __name__ == '__main__': # pragma: no cover
  101. main()