From 49378d1f13c69cdb4a265d60b84588ac4c64bb19 Mon Sep 17 00:00:00 2001
From: ibuler <ibuler@qq.com>
Date: Tue, 18 Mar 2025 16:32:59 +0800
Subject: [PATCH] perf: update ansible env set

---
 .../commands/services/services/celery_base.py    |  4 ++++
 apps/ops/ansible/runners/base.py                 | 16 ++++++++++++++--
 apps/ops/ansible/runners/native.py               |  3 ---
 jms                                              | 10 ----------
 4 files changed, 18 insertions(+), 15 deletions(-)

diff --git a/apps/common/management/commands/services/services/celery_base.py b/apps/common/management/commands/services/services/celery_base.py
index 57cf8ec4d..6bbed5a11 100644
--- a/apps/common/management/commands/services/services/celery_base.py
+++ b/apps/common/management/commands/services/services/celery_base.py
@@ -12,6 +12,10 @@ class CeleryBaseService(BaseService):
     @property
     def cmd(self):
         print('\n- Start Celery as Distributed Task Queue: {}'.format(self.queue.capitalize()))
+        os.environ.setdefault('PYTHONPATH', settings.APPS_DIR)
+        os.environ.setdefault('LC_ALL', 'en_US.UTF-8')
+        os.environ.setdefault('LANG', 'en_US.UTF-8')
+        os.environ.setdefault('PYTHONOPTIMIZE', '1')
 
         if os.getuid() == 0:
             os.environ.setdefault('C_FORCE_ROOT', '1')
diff --git a/apps/ops/ansible/runners/base.py b/apps/ops/ansible/runners/base.py
index 2cadbb368..067ad7635 100644
--- a/apps/ops/ansible/runners/base.py
+++ b/apps/ops/ansible/runners/base.py
@@ -1,11 +1,23 @@
-from ops.ansible.cleaner import WorkPostRunCleaner, cleanup_post_run
+import os
+
+from django.conf import settings
+
+from ops.ansible.cleaner import WorkPostRunCleaner
 
 
 class BaseRunner(WorkPostRunCleaner):
-
     def __init__(self, **kwargs):
         self.runner_params = kwargs
         self.clean_workspace = kwargs.pop("clean_workspace", True)
+        self.setup_env()
+
+    @staticmethod
+    def setup_env():
+        ansible_config_path = os.path.join(settings.APPS_DIR, 'libs', 'ansible', 'ansible.cfg')
+        ansible_modules_path = os.path.join(settings.APPS_DIR, 'libs', 'ansible', 'modules')
+        os.environ.setdefault('ANSIBLE_FORCE_COLOR', 'True')
+        os.environ.setdefault('ANSIBLE_CONFIG', ansible_config_path)
+        os.environ.setdefault('ANSIBLE_LIBRARY', ansible_modules_path)
 
     @classmethod
     def kill_precess(cls, pid):
diff --git a/apps/ops/ansible/runners/native.py b/apps/ops/ansible/runners/native.py
index 00f541ae8..9ff0ed5c7 100644
--- a/apps/ops/ansible/runners/native.py
+++ b/apps/ops/ansible/runners/native.py
@@ -8,9 +8,6 @@ __all__ = ['AnsibleNativeRunner']
 
 
 class AnsibleNativeRunner(BaseRunner):
-    def __init__(self, **kwargs):
-        super().__init__(**kwargs)
-
     @classmethod
     def kill_precess(cls, pid):
         return kill_ansible_ssh_process(pid)
diff --git a/jms b/jms
index 8021ff835..6797c6d59 100755
--- a/jms
+++ b/jms
@@ -20,16 +20,6 @@ sys.path.insert(0, APP_DIR)
 os.environ.setdefault("DJANGO_SETTINGS_MODULE", "jumpserver.settings")
 django.setup()
 
-ansible_config_path = os.path.join(APP_DIR, 'libs', 'ansible', 'ansible.cfg')
-ansible_modules_path = os.path.join(APP_DIR, 'libs', 'ansible', 'modules')
-os.environ.setdefault('LC_ALL', 'en_US.UTF-8')
-os.environ.setdefault('LANG', 'en_US.UTF-8')
-os.environ.setdefault('PYTHONOPTIMIZE', '1')
-os.environ.setdefault('ANSIBLE_FORCE_COLOR', 'True')
-os.environ.setdefault('ANSIBLE_CONFIG', ansible_config_path)
-os.environ.setdefault('ANSIBLE_LIBRARY', ansible_modules_path)
-os.environ.setdefault('PYTHONPATH', APP_DIR)
-
 logging.basicConfig(level=logging.DEBUG, format="%(asctime)s %(message)s", datefmt="%Y-%m-%d %H:%M:%S")
 
 try: