1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
|
from django.core.management.base import BaseCommand
from ... import get_queue
class Command(BaseCommand):
"""
Queue a function with the given arguments.
"""
help = __doc__
args = '<function arg arg ...>'
def add_arguments(self, parser):
parser.add_argument('--queue', '-q', dest='queue', default='default',
help='Specify the queue [default]')
parser.add_argument('--timeout', '-t', type=int, dest='timeout',
help='A timeout in seconds')
parser.add_argument('args', nargs='*')
def handle(self, *args, **options):
"""
Queues the function given with the first argument with the
parameters given with the rest of the argument list.
"""
queue = get_queue(options['queue'])
job = queue.enqueue_call(args[0], args=args[1:], timeout=options['timeout'])
if options['verbosity']:
print('Job %s created' % job.id)
|
