diff --git a/apps/utils/tasks.py b/apps/utils/tasks.py index 7dc6f2c9..3c69e88d 100644 --- a/apps/utils/tasks.py +++ b/apps/utils/tasks.py @@ -5,21 +5,39 @@ import logging from django.conf import settings from server.settings import get_sysconfig import importlib +import hashlib +from django.core.cache import cache # 实例化myLogger myLogger = logging.getLogger('log') +def generate_cache_key(subject): + # 使用邮件主题生成一个唯一的缓存键 + hash_subject = hashlib.md5(subject.encode()).hexdigest() + return f"error_email_{hash_subject}" @shared_task def send_mail_task(**args): config = get_sysconfig() - from django.core.mail import send_mail args['subject'] = '{}:{}_{}_{}'.format( settings.SYS_NAME, settings.SYS_VERSION, config['base']['base_name_short'], args.get('subject', '500')) args['from_email'] = args.get('from_email', settings.EMAIL_HOST_USER) args['recipient_list'] = args.get( 'recipient_list', [settings.EMAIL_HOST_USER]) - send_mail(**args) + cache_key = generate_cache_key(args['subject']) + email_tuple = cache.get(cache_key) + if email_tuple is None: + email_tuple = (0, True) + cache.set(cache_key, email_tuple, 60) + email_count, email_enable = email_tuple + if email_enable: + email_count += 1 + if email_count > 4: + email_enable = False + # 如果频率高于每分钟4封,则自动屏蔽半小时 + cache.set(cache_key, (email_count, email_enable), 1800) + from django.core.mail import send_mail + send_mail(**args) class CustomTask(Task): """