From f59f03adfd00897d1cfba172385177c9b52cc749 Mon Sep 17 00:00:00 2001
From: ibuler <ibuler@qq.com>
Date: Mon, 12 Mar 2018 10:20:23 +0800
Subject: [PATCH] =?UTF-8?q?[Update]=20=E6=94=AF=E6=8C=81=E4=BD=BF=E7=94=A8?=
 =?UTF-8?q?=E7=8E=AF=E5=A2=83=E5=8F=98=E9=87=8F=E5=85=B3=E9=97=AD=E5=AE=9A?=
 =?UTF-8?q?=E6=97=B6=E4=BB=BB=E5=8A=A1?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 apps/assets/tasks.py | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/apps/assets/tasks.py b/apps/assets/tasks.py
index 3f792a080..be9f75152 100644
--- a/apps/assets/tasks.py
+++ b/apps/assets/tasks.py
@@ -1,6 +1,7 @@
 # ~*~ coding: utf-8 ~*~
 import json
 import re
+import os
 
 from celery import shared_task
 from django.core.cache import cache
@@ -20,6 +21,7 @@ TIMEOUT = 60
 logger = get_logger(__file__)
 CACHE_MAX_TIME = 60*60*60
 disk_pattern = re.compile(r'^hd|sd|xvd|vd')
+PERIOD_TASK = os.environ.get("PERIOD_TASK", "on")
 
 
 @shared_task
@@ -118,6 +120,10 @@ def update_assets_hardware_info_period():
     Update asset hardware period task
     :return:
     """
+    if PERIOD_TASK != "on":
+        logger.debug("Period task disabled, update assets hardware info pass")
+        return
+
     from ops.utils import update_or_create_ansible_task
     task_name = _("Update assets hardware info period")
     hostname_list = [
@@ -190,6 +196,10 @@ def test_admin_user_connectability_period():
     """
     A period task that update the ansible task period
     """
+    if PERIOD_TASK != "on":
+        logger.debug("Period task disabled, test admin user connectability pass")
+        return
+
     admin_users = AdminUser.objects.all()
     for admin_user in admin_users:
         task_name = _("Test admin user connectability period: {}".format(admin_user.name))
@@ -287,6 +297,10 @@ def test_system_user_connectability_manual(system_user):
 @after_app_ready_start
 @after_app_shutdown_clean
 def test_system_user_connectability_period():
+    if PERIOD_TASK != "on":
+        logger.debug("Period task disabled, test system user connectability pass")
+        return
+
     system_users = SystemUser.objects.all()
     for system_user in system_users:
         task_name = _("test system user connectability period: {}".format(system_user))
-- 
GitLab