feat:限制必须api-key或登录6
parent
e6efa1a074
commit
1db5c59925
|
|
@ -0,0 +1,31 @@
|
|||
import os
|
||||
import subprocess
|
||||
|
||||
from .celery_base import CeleryBaseService
|
||||
from django.conf import settings
|
||||
|
||||
__all__ = ['CeleryModelService']
|
||||
|
||||
|
||||
class CeleryModelService(CeleryBaseService):
|
||||
|
||||
def __init__(self, **kwargs):
|
||||
kwargs['queue'] = 'model'
|
||||
super().__init__(**kwargs)
|
||||
|
||||
def open_subprocess(self):
|
||||
env = os.environ.copy()
|
||||
env['LC_ALL'] = 'C.UTF-8'
|
||||
env['PYTHONOPTIMIZE'] = '1'
|
||||
env['ANSIBLE_FORCE_COLOR'] = 'True'
|
||||
env['PYTHONPATH'] = settings.APPS_DIR
|
||||
env['SERVER_NAME'] = 'celery'
|
||||
if os.getuid() == 0:
|
||||
env.setdefault('C_FORCE_ROOT', '1')
|
||||
kwargs = {
|
||||
'cwd': self.cwd,
|
||||
'stderr': self.log_file,
|
||||
'stdout': self.log_file,
|
||||
'env': env
|
||||
}
|
||||
self._process = subprocess.Popen(self.cmd, **kwargs)
|
||||
Loading…
Reference in New Issue