diff --git a/celery_app.py b/celery_app.py index 324974e..480504f 100644 --- a/celery_app.py +++ b/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') diff --git a/celery_config.py b/celery_config.py index e984dcd..91e0894 100644 --- a/celery_config.py +++ b/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", -} \ No newline at end of file +} + + +# 确保 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') diff --git a/tasks.py b/tasks.py index 8aef282..4c1e372 100644 --- a/tasks.py +++ b/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")