feat: 密码计划邮件提醒

apps/common/tasks.py

@@ -1,4 +1,6 @@
-from django.core.mail import send_mail
+import os
+
+from django.core.mail import send_mail, EmailMultiAlternatives
 from django.conf import settings
 from celery import shared_task
 
@@ -24,11 +26,31 @@ def send_mail_async(*args, **kwargs):
     if len(args) == 3:
         args = list(args)
         args[0] = (settings.EMAIL_SUBJECT_PREFIX or '') + args[0]
-        email_from = settings.EMAIL_FROM or settings.EMAIL_HOST_USER
-        args.insert(2, email_from)
+        from_email = settings.EMAIL_FROM or settings.EMAIL_HOST_USER
+        args.insert(2, from_email)
         args = tuple(args)
 
     try:
         return send_mail(*args, **kwargs)
     except Exception as e:
         logger.error("Sending mail error: {}".format(e))
+
+
+@shared_task
+def send_mail_attachment_async(subject, message, recipient_list, attachment_list=None):
+    if attachment_list is None:
+        attachment_list = []
+    from_email = settings.EMAIL_FROM or settings.EMAIL_HOST_USER
+    email = EmailMultiAlternatives(
+        subject=subject,
+        body=message,
+        from_email=from_email,
+        to=recipient_list
+    )
+    for attachment in attachment_list:
+        email.attach_file(attachment)
+        os.remove(attachment)
+    try:
+        return email.send()
+    except Exception as e:
+        logger.error("Sending mail attachment error: {}".format(e))

apps/common/utils/file.py

@@ -0,0 +1,19 @@
+import os
+import csv
+import pyzipper
+
+
+def create_csv_file(filename, headers, rows, ):
+    with open(filename, 'w', encoding='utf-8-sig')as f:
+        w = csv.writer(f)
+        w.writerow(headers)
+        w.writerows(rows)
+
+
+def encrypt_and_compress_zip_file(filename, secret_password, encrypted_filename):
+    with pyzipper.AESZipFile(
+            filename, 'w', compression=pyzipper.ZIP_LZMA, encryption=pyzipper.WZ_AES
+    ) as zf:
+        zf.setpassword(secret_password)
+        with open(encrypted_filename, 'rb') as f:
+            zf.writestr(os.path.basename(encrypted_filename), f.read())