mirror of https://github.com/jumpserver/jumpserver
21 lines
575 B
Python
21 lines
575 B
Python
|
from django.core.management.base import BaseCommand, CommandError
|
||
|
|
||
|
|
||
|
class Command(BaseCommand):
|
||
|
help = 'Ops manage commands'
|
||
|
|
||
|
def add_arguments(self, parser):
|
||
|
parser.add_argument('check_celery', nargs='?', help='Check celery health')
|
||
|
|
||
|
def handle(self, *args, **options):
|
||
|
from ops.celery.utils import get_celery_status, get_beat_status
|
||
|
|
||
|
ok = get_celery_status()
|
||
|
if not ok:
|
||
|
raise CommandError('Celery worker unhealthy')
|
||
|
|
||
|
ok = get_beat_status()
|
||
|
if not ok:
|
||
|
raise CommandError('Beat unhealthy')
|
||
|
|