from optparse import make_option from os import path from django.core.management.base import BaseCommand from orchestra.utils.paths import get_site_root, get_orchestra_root from orchestra.utils.system import run, check_root class Command(BaseCommand): def __init__(self, *args, **kwargs): super(Command, self).__init__(*args, **kwargs) self.option_list = BaseCommand.option_list + ( make_option('--username', dest='username', default='orchestra', help='Specifies the system user that would run celeryd.'), make_option('--processes', dest='processes', default=2, help='Number of celeryd processes.'), make_option('--noinput', action='store_false', dest='interactive', default=True, help='Tells Django to NOT prompt the user for input of any kind. ' 'You must use --username with --noinput, and must contain the ' 'cleleryd process owner, which is the user how will perform tincd updates'), ) option_list = BaseCommand.option_list help = 'Configure Celeryd to run with your orchestra instance.' @check_root def handle(self, *args, **options): context = { 'site_root': get_site_root(), 'username': options.get('username'), 'bin_path': path.join(get_orchestra_root(), 'bin'), 'processes': options.get('processes'), } celery_config = ( '# Name of nodes to start, here we have a single node\n' 'CELERYD_NODES="w1"\n' '\n' '# Where to chdir at start.\n' 'CELERYD_CHDIR="%(site_root)s"\n' '\n' '# How to call "manage.py celeryd_multi"\n' 'CELERYD_MULTI="$CELERYD_CHDIR/manage.py celeryd_multi"\n' '\n' '# Extra arguments to celeryd\n' 'CELERYD_OPTS="-P:w1 processes -c:w1 %(processes)s -Q:w1 celery"\n' '\n' '# Name of the celery config module.\n' 'CELERY_CONFIG_MODULE="celeryconfig"\n' '\n' '# %%n will be replaced with the nodename.\n' 'CELERYD_LOG_FILE="/var/log/celery/%%n.log"\n' 'CELERYD_PID_FILE="/var/run/celery/%%n.pid"\n' 'CELERY_CREATE_DIRS=1\n' '\n' '# Full path to the celeryd logfile.\n' 'CELERYEV_LOG_FILE="/var/log/celery/celeryev.log"\n' 'CELERYEV_PID_FILE="/var/run/celery/celeryev.pid"\n' '\n' '# Workers should run as an unprivileged user.\n' 'CELERYD_USER="%(username)s"\n' 'CELERYD_GROUP="$CELERYD_USER"\n' '\n' '# Persistent revokes\n' 'CELERYD_STATE_DB="$CELERYD_CHDIR/persistent_revokes"\n' '\n' '# Celeryev\n' 'CELERYEV="$CELERYD_CHDIR/manage.py"\n' 'CELERYEV_CAM="djcelery.snapshot.Camera"\n' 'CELERYEV_USER="$CELERYD_USER"\n' 'CELERYEV_GROUP="$CELERYD_USER"\n' 'CELERYEV_OPTS="celerycam"\n' '\n' '# Celerybeat\n' 'CELERYBEAT="${CELERYD_CHDIR}/manage.py celerybeat"\n' 'CELERYBEAT_USER="$CELERYD_USER"\n' 'CELERYBEAT_GROUP="$CELERYD_USER"\n' 'CELERYBEAT_CHDIR="$CELERYD_CHDIR"\n' 'CELERYBEAT_OPTS="--schedule=/var/run/celerybeat-schedule"\n' % context ) run("echo '%s' > /etc/default/celeryd" % celery_config) # https://raw.github.com/celery/celery/master/extra/generic-init.d/ for script in ['celeryevcam', 'celeryd', 'celerybeat']: context['script'] = script run('cp %(bin_path)s/%(script)s /etc/init.d/%(script)s' % context) run('chmod +x /etc/init.d/%(script)s' % context) run('update-rc.d %(script)s defaults' % context) rotate = ( '/var/log/celery/*.log {\n' ' weekly\n' ' missingok\n' ' rotate 10\n' ' compress\n' ' delaycompress\n' ' notifempty\n' ' copytruncate\n' '}' ) run("echo '%s' > /etc/logrotate.d/celeryd" % rotate)