101 lines
2.6 KiB
Python
101 lines
2.6 KiB
Python
# -*- coding: utf-8 -*-
|
|
"""
|
|
|
|
The :program:`celery beat` command.
|
|
|
|
.. program:: celery beat
|
|
|
|
.. seealso::
|
|
|
|
See :ref:`preload-options` and :ref:`daemon-options`.
|
|
|
|
.. cmdoption:: --detach
|
|
|
|
Detach and run in the background as a daemon.
|
|
|
|
.. cmdoption:: -s, --schedule
|
|
|
|
Path to the schedule database. Defaults to `celerybeat-schedule`.
|
|
The extension '.db' may be appended to the filename.
|
|
Default is {default}.
|
|
|
|
.. cmdoption:: -S, --scheduler
|
|
|
|
Scheduler class to use.
|
|
Default is :class:`celery.beat.PersistentScheduler`.
|
|
|
|
.. cmdoption:: --max-interval
|
|
|
|
Max seconds to sleep between schedule iterations.
|
|
|
|
.. cmdoption:: -f, --logfile
|
|
|
|
Path to log file. If no logfile is specified, `stderr` is used.
|
|
|
|
.. cmdoption:: -l, --loglevel
|
|
|
|
Logging level, choose between `DEBUG`, `INFO`, `WARNING`,
|
|
`ERROR`, `CRITICAL`, or `FATAL`.
|
|
|
|
"""
|
|
from __future__ import absolute_import
|
|
|
|
from functools import partial
|
|
|
|
from celery.platforms import detached, maybe_drop_privileges
|
|
|
|
from celery.bin.base import Command, Option, daemon_options
|
|
|
|
__all__ = ['beat']
|
|
|
|
|
|
class beat(Command):
|
|
"""Start the beat periodic task scheduler.
|
|
|
|
Examples::
|
|
|
|
celery beat -l info
|
|
celery beat -s /var/run/celery/beat-schedule --detach
|
|
celery beat -S djcelery.schedulers.DatabaseScheduler
|
|
|
|
"""
|
|
doc = __doc__
|
|
enable_config_from_cmdline = True
|
|
supports_args = False
|
|
|
|
def run(self, detach=False, logfile=None, pidfile=None, uid=None,
|
|
gid=None, umask=None, working_directory=None, **kwargs):
|
|
if not detach:
|
|
maybe_drop_privileges(uid=uid, gid=gid)
|
|
workdir = working_directory
|
|
kwargs.pop('app', None)
|
|
beat = partial(self.app.Beat,
|
|
logfile=logfile, pidfile=pidfile, **kwargs)
|
|
|
|
if detach:
|
|
with detached(logfile, pidfile, uid, gid, umask, workdir):
|
|
return beat().run()
|
|
else:
|
|
return beat().run()
|
|
|
|
def get_options(self):
|
|
c = self.app.conf
|
|
|
|
return (
|
|
(Option('--detach', action='store_true'),
|
|
Option('-s', '--schedule',
|
|
default=c.CELERYBEAT_SCHEDULE_FILENAME),
|
|
Option('--max-interval', type='float'),
|
|
Option('-S', '--scheduler', dest='scheduler_cls'),
|
|
Option('-l', '--loglevel', default=c.CELERYBEAT_LOG_LEVEL))
|
|
+ daemon_options(default_pidfile='celerybeat.pid')
|
|
+ tuple(self.app.user_options['beat'])
|
|
)
|
|
|
|
|
|
def main(app=None):
|
|
beat(app=app).execute_from_commandline()
|
|
|
|
if __name__ == '__main__': # pragma: no cover
|
|
main()
|