47 lines
1.2 KiB
Python
47 lines
1.2 KiB
Python
|
# coding=utf-8
|
|||
|
|
"""
|
|||
|
|
@project: MaxKB
|
|||
|
|
@Author：虎
|
|||
|
|
@file： celery.py
|
|||
|
|
@date：2024/8/19 11:57
|
|||
|
|
@desc:
|
|||
|
|
"""
|
|||
|
|
import os
|
|||
|
|
import subprocess
|
|||
|
|
|
|||
|
|
from django.core.management.base import BaseCommand
|
|||
|
|
|
|||
|
|
from maxkb.const import BASE_DIR
|
|||
|
|
|
|||
|
|
|
|||
|
|
class Command(BaseCommand):
|
|||
|
|
help = 'celery'
|
|||
|
|
|
|||
|
|
def add_arguments(self, parser):
|
|||
|
|
parser.add_argument(
|
|||
|
|
'service', nargs='+', type=str, choices=("celery", "model"), help='Service',
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
def handle(self, *args, **options):
|
|||
|
|
service = options.get('service')
|
|||
|
|
os.environ.setdefault('CELERY_NAME', ','.join(service))
|
|||
|
|
server_hostname = os.environ.get("SERVER_HOSTNAME")
|
|||
|
|
if hasattr(os, 'getuid') and os.getuid() == 0:
|
|||
|
|
os.environ.setdefault('C_FORCE_ROOT', '1')
|
|||
|
|
if not server_hostname:
|
|||
|
|
server_hostname = '%h'
|
|||
|
|
cmd = [
|
|||
|
|
'celery',
|
|||
|
|
'-A', 'ops',
|
|||
|
|
'worker',
|
|||
|
|
'-P', 'threads',
|
|||
|
|
'-l', 'info',
|
|||
|
|
'-c', '10',
|
|||
|
|
'-Q', ','.join(service),
|
|||
|
|
'--heartbeat-interval', '10',
|
|||
|
|
'-n', f'{",".join(service)}@{server_hostname}',
|
|||
|
|
'--without-mingle',
|
|||
|
|
]
|
|||
|
|
kwargs = {'cwd': BASE_DIR}
|
|||
|
|
subprocess.run(cmd, **kwargs)
|