mailing example in partner programs

Author: koshakkkq

mailing/views.py

@@ -25,16 +25,33 @@ def post(self, request):
 
 
 class MailingSchemaView(APIView):
+    def get(self, request):
+        return MailingTemplateRender().render_template(request)
+
+
+def template_fields(request, schema_id):
+    return JsonResponse(
+        dict(MailingTemplateRender().get_template_fields_context(schema_id))
+    )
+
+
+class MailingTemplateRender:
     template_name = "templates/mailing/mail_schema.html"
 
-    def get(self, request):
+    def render_template(
+        self,
+        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,
             self.template_name,
             self._get_context(
-                1,
-                None,
-                CustomUser.objects.all(),
+                schema_id,
+                picked_users,
+                unpicked_users,
             ),
         )
 
@@ -84,8 +101,11 @@ def _user_to_dict_for_template(self, user, picked):
         }
 
     def get_template_fields_context(self, schema_id):
-        schema = MailingSchema.objects.get(pk=schema_id).schema
         context = {"template_fields": []}
+        if schema_id is None:
+            return list(context.items())
+        schema = MailingSchema.objects.get(pk=schema_id).schema
+
         for key in schema:
             context["template_fields"].append(
                 {
@@ -95,7 +115,3 @@ def get_template_fields_context(self, schema_id):
                 }
             )
         return list(context.items())
-
-
-def template_fields(request, schema_id):
-    return JsonResponse(dict(MailingSchemaView().get_template_fields_context(schema_id)))

partner_programs/admin.py

@@ -4,7 +4,7 @@
 from django.http import HttpResponse
 from django.urls import path
 from django.utils import timezone
-
+from mailing.views import MailingTemplateRender
 from partner_programs.models import PartnerProgram, PartnerProgramUserProfile
 
 
@@ -42,10 +42,23 @@ def get_urls(self):
                 "export/<int:object_id>/",
                 self.admin_site.admin_view(self.get_export_file_view),
                 name="export_profiles",
-            )
+            ),
+            path(
+                "mailing/<int:partner_program>/",
+                self.admin_site.admin_view(self.mailing),
+                name="partner_programs_mailing",
+            ),
         ]
         return custom_urls + default_urls
 
+    def mailing(self, request, partner_program):
+        print("kekekekekek")
+        profiles = PartnerProgramUserProfile.objects.filter(
+            partner_program=partner_program
+        )
+        users = [profile.user for profile in profiles]
+        return MailingTemplateRender().render_template(request, None, users, None)
+
     def changeform_view(self, request, object_id=None, form_url="", extra_context=None):
         extra_context = {"object_id": int(object_id)}
         res = super(PartnerProgramAdmin, self).changeform_view(

templates/partner_programs/admin/programs_change_form.html

@@ -8,11 +8,19 @@
     <div class="submit-row">
         <input type="button" class="export-btn" value="Экспорт" onclick="export_profiles()"/>
     </div>
+
+
+    <div class="submit-row">
+        <input type="button" class="export-btn" value="Рассылка." onclick="mailing()"/>
+    </div>
 <script src="https://ajax.googleapis.com/ajax/libs/jquery/3.3.1/jquery.min.js"></script>
     <script>
     function export_profiles(){
         window.open("{% url 'admin:export_profiles' object_id %}", '_blank').focus()
     }
+    function mailing(){
+        window.open("{% url 'admin:partner_programs_mailing' object_id %}", '_blank').focus()
+    }
 </script>
 
 {% endblock %}