mirror of
https://github.com/django-q2/django-q2.git
synced 2026-10-03 07:38:12 +08:00
29 lines
805 B
Python
29 lines
805 B
Python
from optparse import make_option
|
|
|
|
from django.core.management.base import BaseCommand
|
|
from django.utils.translation import ugettext as _
|
|
|
|
from django_q.cluster import Cluster
|
|
|
|
|
|
class Command(BaseCommand):
|
|
# Translators: help text for qcluster management command
|
|
help = _("Starts a Django Q Cluster.")
|
|
|
|
option_list = BaseCommand.option_list + (
|
|
make_option('--run-once',
|
|
action='store_true',
|
|
dest='run_once',
|
|
default=False,
|
|
help='Run once and then stop.'),
|
|
)
|
|
|
|
def add_arguments(self, parser):
|
|
parser.add_argument('poll_id', nargs='+', type=int)
|
|
|
|
def handle(self, *args, **options):
|
|
q = Cluster()
|
|
q.start()
|
|
if options.get('run_once', False):
|
|
q.stop()
|