diff --git a/apps/utils/tasks.py b/apps/utils/tasks.py index 7dc6f2c9..9e831e7f 100644 --- a/apps/utils/tasks.py +++ b/apps/utils/tasks.py @@ -5,21 +5,35 @@ import logging from django.conf import settings from server.settings import get_sysconfig import importlib +from django.core.cache import cache # 实例化myLogger myLogger = logging.getLogger('log') - @shared_task def send_mail_task(**args): config = get_sysconfig() - from django.core.mail import send_mail + subject = args.get('subject', '500') args['subject'] = '{}:{}_{}_{}'.format( - settings.SYS_NAME, settings.SYS_VERSION, config['base']['base_name_short'], args.get('subject', '500')) + settings.SYS_NAME, settings.SYS_VERSION, config['base']['base_name_short'], subject) 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 = f'error_mail_{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) + args['subject'] = args['subject'] + '_发送频繁' + from django.core.mail import send_mail + send_mail(**args) class CustomTask(Task): """