mass mailing users from admin

Author: XataBq

mailing/constants.py

@@ -7,3 +7,6 @@ def get_default_mailing_schema() -> dict[str, dict[str, str]]:
         "text": {"title": "Основной текст письма"},
         "button_text": {"title": "Текст кнопки", "default": "Кнопка"},
     }
+
+
+QUANTITY_MAILING_USERS_IN_GROUP = 100

mailing/utils.py

@@ -1,4 +1,5 @@
 from typing import Dict, List, Union
+from .constants import QUANTITY_MAILING_USERS_IN_GROUP
 
 import django.db.models
 from django.contrib.auth import get_user_model
@@ -9,28 +10,40 @@
 User = get_user_model()
 
 
+def new_connection(old_connection):
+    old_connection.close()
+    connection = mail.get_connection()
+    return connection
+
+
+def get_users_groups(users_list):
+    users_groups_list = [users_list[user:user + QUANTITY_MAILING_USERS_IN_GROUP] for user in
+                         range(0, len(users_list), QUANTITY_MAILING_USERS_IN_GROUP)]
+    return users_groups_list
+
+
 def send_mail(
-    user: User,
-    subject: str,
-    template_string: str,
-    template_context: Union[
-        Dict,
-        List,
-    ] = None,
-    connection=None,
+        user: User,
+        subject: str,
+        template_string: str,
+        template_context: Union[
+            Dict,
+            List,
+        ] = None,
+        connection=None,
 ):
     return send_mass_mail([user], subject, template_string, template_context, connection)
 
 
 def send_mass_mail(
-    users: django.db.models.QuerySet | List[User],
-    subject: str,
-    template_string: str,
-    template_context: Union[
-        Dict,
-        List,
-    ] = None,
-    connection=None,
+        users: django.db.models.QuerySet | List[User],
+        subject: str,
+        template_string: str,
+        template_context: Union[
+            Dict,
+            List,
+        ] = None,
+        connection=None,
 ) -> None:
     """
     Begin mailing to specified users, sending rendered template with template_text arg.

mailing/views.py

@@ -2,10 +2,12 @@
 from django.http import JsonResponse
 from django.shortcuts import render
 from rest_framework.views import APIView
+from django.core import mail
 
 from users.models import CustomUser
 from .utils import send_mass_mail
 from .models import MailingSchema
+from .utils import get_users_groups, new_connection
 
 
 class SendMailView(APIView):
@@ -20,7 +22,11 @@ def post(self, request):
             if key_in_post in request.POST:
                 context[variable_name] = request.POST[key_in_post]
         users_to_send = CustomUser.objects.filter(pk__in=users)
-        send_mass_mail(users_to_send, subject, mail_schema.template, context)
+        users_groups = get_users_groups(users_to_send)
+        connection = mail.get_connection()
+        for group in users_groups:
+            send_mass_mail(users_to_send, subject, mail_schema.template, context, connection)
+            connection = new_connection(connection)
         return JsonResponse({"detail": "ok"})
 
 
@@ -41,11 +47,11 @@ class MailingTemplateRender:
 
     @classmethod
     def render_template(
-        cls,
-        request,
-        schema_id: int | None = None,
-        picked_users: list[CustomUser] | django.db.models.QuerySet = None,
-        unpicked_users: list[CustomUser] | django.db.models.QuerySet = None,
+            cls,
+            request,
+            schema_id: int | None = None,
+            picked_users: list[CustomUser] | django.db.models.QuerySet = None,
+            unpicked_users: list[CustomUser] | django.db.models.QuerySet = None,
     ):
         return render(
             request,
@@ -59,10 +65,10 @@ def render_template(
 
     @classmethod
     def _get_context(
-        cls,
-        schema_id: int | None = None,
-        picked_users: list[CustomUser] | django.db.models.QuerySet = None,
-        unpicked_users: list[CustomUser] | django.db.models.QuerySet = None,
+            cls,
+            schema_id: int | None = None,
+            picked_users: list[CustomUser] | django.db.models.QuerySet = None,
+            unpicked_users: list[CustomUser] | django.db.models.QuerySet = None,
     ):
         context = cls._get_schema_context(schema_id)
         context += cls._get_users_context(picked_users, unpicked_users)
@@ -83,9 +89,9 @@ def _get_schema_context(cls, schema_id: int | None = None):
 
     @classmethod
     def _get_users_context(
-        cls,
-        picked_users: list[CustomUser] | django.db.models.QuerySet = None,
-        unpicked_users: list[CustomUser] | django.db.models.QuerySet = None,
+            cls,
+            picked_users: list[CustomUser] | django.db.models.QuerySet = None,
+            unpicked_users: list[CustomUser] | django.db.models.QuerySet = None,
     ):
         if picked_users is None:
             picked_users = []

templates/admin/change_list.html

@@ -77,7 +77,7 @@
                 <br>
                 <script>
                     function mailing() {
-                        window.open("{% url 'admin:all_users_mailing' %}", '_blank').focus()
+                        window.open("{% url 'admin:user_mass_mail' %}", '_blank').focus()
                     }
                     window.onload = function() {
                     var currentURL = window.location.href;

users/admin.py

@@ -163,9 +163,18 @@ def get_urls(self):
                 self.admin_site.admin_view(self.force_verify),
                 name="force_verify",
             ),
+            path(
+                "mailing/",
+                self.admin_site.admin_view(self.mass_mail),
+                name="user_mass_mail",
+            ),
         ]
         return custom_urls + default_urls
 
+    def mass_mail(self, request):
+        users = CustomUser.objects.all()
+        return MailingTemplateRender().render_template(request, None, users, None)
+
     def mailing(self, request, user_object):
         user = CustomUser.objects.get(pk=user_object)
         users = [user]