调整任务日志

celery_app.py

@@ -14,7 +14,11 @@ from services.biz_service import BizService
 from urllib.parse import quote
 import asyncio,os,random,sys 
 #from tasks import add_friends_task
-from common.log import logger
+#from common.log import logger
+
+import logging
+
+logger = logging.getLogger('redbeat')
 
 
 

celery_config.py

@@ -3,6 +3,7 @@ from celery import Celery
 
 from config import load_config,conf
 from urllib.parse import quote
+import logging,os
 
 
 load_config()
@@ -52,4 +53,19 @@ kafka_config = {
 }
 gewe_config = {
     'api_url': "http://api.geweapi.com/gewe",
-}
+}
+
+
+# 确保 logs 文件夹存在
+log_dir = 'logs'
+if not os.path.exists(log_dir):
+    os.makedirs(log_dir)
+
+# 配置日志记录
+logging.basicConfig(
+    level=logging.INFO,
+    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
+    filename=os.path.join(log_dir, 'redbeat.log'),  # 日志文件路径
+    filemode='a'  # 追加模式
+)
+logger = logging.getLogger('redbeat')

tasks.py

@@ -9,12 +9,15 @@ from datetime import timedelta
 from services.redis_service import RedisService
 from services.kafka_service import KafkaService
 from services.gewe_service import GeWeService
-from common.log import logger
+# from common.log import logger
 from common.utils import *
 import asyncio,random
 from model.models import AddGroupContactsHistory
+import logging
 
+import logging
 
+logger = logging.getLogger('redbeat')
 
 
 @celery_app.task(name='tasks.add_task', bind=True, acks_late=True)
@@ -765,7 +768,7 @@ def add_friends_task(self,redis_config):
     async def task():
         redis_service = RedisService()
         await redis_service.init(**redis_config)
-        today_str = datetime.datetime.now().strftime("%Y%m%d")
+        today_str = datetime.now().strftime("%Y%m%d")
         redis_key = REDIS_KEY_PATTERN.format(date=today_str)
 
         # 获取当前总添加数量
@@ -842,7 +845,7 @@ def add_friends_task(self,redis_config):
 
 @celery_app.task(name='tasks.random_scheduled_task', bind=True, acks_late=True)
 def random_scheduled_task(self,):
-    print(f"Task executed at {datetime.datetime.now()}")
+    print(f"Task executed at {datetime.now()}")
     # 随机生成下次执行时间（例如：10-60秒内的随机时间）
     next_run_in = random.randint(10, 60)
     print(f"Next execution will be in {next_run_in} seconds")