From 40fcc5bcb9ae2afe506ffb591a6fc96b38def140 Mon Sep 17 00:00:00 2001
From: CaptainB <bin@fit2cloud.com>
Date: Wed, 25 Jun 2025 17:36:07 +0800
Subject: [PATCH] refactor: replace logging calls with maxkb_logger for
 consistency

---
 apps/ops/celery/signal_handler.py | 3 ++-
 apps/ops/celery/utils.py          | 3 ++-
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/apps/ops/celery/signal_handler.py b/apps/ops/celery/signal_handler.py
index bfe02241b..91cba3ed0 100644
--- a/apps/ops/celery/signal_handler.py
+++ b/apps/ops/celery/signal_handler.py
@@ -10,6 +10,7 @@ from celery.signals import (
 from django.core.cache import cache
 from django_celery_beat.models import PeriodicTask
 
+from common.utils.logger import maxkb_logger
 from .decorator import get_after_app_ready_tasks, get_after_app_shutdown_clean_tasks
 from .logger import CeleryThreadTaskFileHandler
 
@@ -65,7 +66,7 @@ def add_celery_logger_handler(sender=None, logger=None, loglevel=None, format=No
 
 @task_revoked.connect
 def on_task_revoked(request, terminated, signum, expired, **kwargs):
-    logging.getLogger('max_kb').info('task_revoked', terminated)
+    maxkb_logger.info('task_revoked', terminated)
 
 
 @task_prerun.connect
diff --git a/apps/ops/celery/utils.py b/apps/ops/celery/utils.py
index 72c2796d3..7a833dfb1 100644
--- a/apps/ops/celery/utils.py
+++ b/apps/ops/celery/utils.py
@@ -9,6 +9,7 @@ from django_celery_beat.models import (
     PeriodicTasks
 )
 
+from common.utils.logger import maxkb_logger
 from maxkb.const import PROJECT_DIR
 
 logger = logging.getLogger(__file__)
@@ -62,7 +63,7 @@ def get_celery_status():
     active_queue_worker = set([n.split('@')[0] for n in active_nodes if n])
     # Celery Worker 数量: 2
     if len(active_queue_worker) < 2:
-        logging.getLogger('max_kb').info("Not all celery worker worked")
+        maxkb_logger.info("Not all celery worker worked")
         return False
     else:
         return True