old_contents: from django.core.management.base import BaseCommand
  from django.conf import settings
  from celery import Celery
  
  
  class Command(BaseCommand):
      help = "Gracefully shutsdown a celery worker"
      args = 'hostname'
  
      def handle(self, hostname, *args, **options):
          celery = Celery()
          celery.config_from_object(settings)
          celery.control.broadcast('shutdown', destination=[hostname])
  
new_contents: from django.core.management.base import BaseCommand
  from django.conf import settings
  from celery import Celery
  
  from corehq.apps.hqadmin.utils import parse_celery_pings
  
  
  class Command(BaseCommand):
      help = "Gracefully shutsdown a celery worker"
      args = 'hostname'
  
      def handle(self, hostname, *args, **options):
          celery = Celery()
          celery.config_from_object(settings)
          celery.control.broadcast('shutdown', destination=[hostname])
          worker_responses = celery.control.ping(timeout=10, destination=[hostname])
          pings = parse_celery_pings(worker_responses)
          if hostname in pings:
              print 'Did not shutdown worker'
              exit(1)
          print 'Successfully initiated warm shutdown'
  
subject: Exit 1 based on whether or not it was shutdown correctly