feat: 邮件报错频率过高自动屏蔽

apps/utils/tasks.py

@@ -5,20 +5,38 @@ 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])
+    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):