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."))