22 lines
743 B
Python
22 lines
743 B
Python
import logging
|
|
import signal
|
|
|
|
from django.core.management.base import BaseCommand
|
|
|
|
from jobs.worker import JobWorkerRunner
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class Command(BaseCommand):
|
|
help = "Run the configured threaded job worker."
|
|
|
|
def handle(self, *args, **options):
|
|
logging.basicConfig(level=logging.INFO, format="%(asctime)s %(levelname)s %(name)s %(message)s")
|
|
runner = JobWorkerRunner()
|
|
signal.signal(signal.SIGINT, runner.request_stop)
|
|
signal.signal(signal.SIGTERM, runner.request_stop)
|
|
self.stdout.write(self.style.SUCCESS(f"Starting {runner.thread_count} job worker thread(s)."))
|
|
runner.run_forever()
|
|
self.stdout.write(self.style.SUCCESS("Job worker stopped."))
|