Merge pull request #602 from PROCOLLAB-github/feature/auto_sending_email

Toksi86 · web-flow · commit 7c89b73491af · 2026-02-05T13:08:14.000+05:00
Добавлены новые сценарии рассылок
diff --git a/mailing/scenarios.py b/mailing/scenarios.py
@@ -11,11 +11,14 @@
 
 class TriggerType(Enum):
     PROGRAM_SUBMISSION_DEADLINE = "program_submission_deadline"
+    PROGRAM_REGISTRATION_DATE = "program_registration_date"
 
 
 class RecipientRule(Enum):
     ALL_PARTICIPANTS = "all_participants"
     NO_PROJECT_IN_PROGRAM = "no_project_in_program"
+    NO_PROJECT_IN_PROGRAM_REGISTERED_ON_DATE = "no_project_in_program_registered_on_date"
+    PROJECT_NOT_SUBMITTED = "project_not_submitted"
 
 
 ContextBuilder = Callable[[PartnerProgram, CustomUser, date], dict]
@@ -34,14 +37,13 @@ class Scenario:
 
 def _build_submission_deadline_context(offset_days: int) -> ContextBuilder:
     def _builder(program: PartnerProgram, user: CustomUser, deadline_date: date) -> dict:
-        deadline_str = deadline_date.strftime("%d.%m.%Y")
         return {
-            "preview_text": f"До окончания подачи проектов осталось {offset_days} дней",
-            "title": "Пора подать проект",
+            "preview_text": "Кейс-чемпионат уже стартовал",
+            "title": "Время начинать!",
             "text": (
-                f"До окончания подачи проектов в программе «{program.name}» "
-                f"осталось {offset_days} дней. "
-                f"Пожалуйста, подайте проект и сформируйте команду до {deadline_str}."
+                "Кейс-чемпионат уже стартовал. Скорее заходите на платформу, "
+                "создавайте проект и подключайте команду к работе.\n\n"
+                "Вас ждет много интересного ⚡"
             ),
             "button_text": "Подать проект",
             "button_link": f"{FRONTEND_BASE_URL}/office/program/{program.id}",
@@ -50,14 +52,94 @@ def _builder(program: PartnerProgram, user: CustomUser, deadline_date: date) ->
     return _builder
 
 
+def _build_registration_plus_5_context() -> ContextBuilder:
+    def _builder(program: PartnerProgram, user: CustomUser, _ref_date: date) -> dict:
+        return {
+            "preview_text": "Сделайте первый шаг в программе",
+            "title": "Сделать первый шаг",
+            "text": (
+                "Когда непонятно с чего начать — стоит начать с самого простого. "
+                "На раз-два-три: зайти на платформу — создать проект — "
+                "пригласить команду.\n\n"
+                "И вот, первый шаг уже сделан"
+            ),
+        }
+
+    return _builder
+
+
+def _build_project_not_submitted_context(title: str, text: str) -> ContextBuilder:
+    def _builder(program: PartnerProgram, user: CustomUser, _ref_date: date) -> dict:
+        return {
+            "preview_text": title,
+            "title": title,
+            "text": text,
+        }
+
+    return _builder
+
+
 SCENARIOS: tuple[Scenario, ...] = (
     Scenario(
         code="program_submission_deadline_minus_10_no_project",
         trigger=TriggerType.PROGRAM_SUBMISSION_DEADLINE,
         offset_days=10,
         template_name="email/generic-template-0.html",
-        subject="Procollab | Подача проекта",
+        subject="Время начинать!",
         recipient_rule=RecipientRule.NO_PROJECT_IN_PROGRAM,
         context_builder=_build_submission_deadline_context(10),
     ),
+    Scenario(
+        code="program_registration_plus_5_no_project",
+        trigger=TriggerType.PROGRAM_REGISTRATION_DATE,
+        offset_days=5,
+        template_name="email/generic-template-0.html",
+        subject="Сделать первый шаг",
+        recipient_rule=RecipientRule.NO_PROJECT_IN_PROGRAM_REGISTERED_ON_DATE,
+        context_builder=_build_registration_plus_5_context(),
+    ),
+    Scenario(
+        code="program_submission_deadline_minus_9_project_not_submitted",
+        trigger=TriggerType.PROGRAM_SUBMISSION_DEADLINE,
+        offset_days=9,
+        template_name="email/generic-template-0.html",
+        subject="Кейс-задания опубликованы",
+        recipient_rule=RecipientRule.PROJECT_NOT_SUBMITTED,
+        context_builder=_build_project_not_submitted_context(
+            "Кейс-задания опубликованы",
+            "Заходите на платформу, чтобы познакомиться с кейсами первого этапа "
+            "кейс-чемпионата. Кейсы загружены в материалы закрытой группы.\n\n"
+            "Приступайте к работе уже сегодня, чтобы успеть подготовить итоговое "
+            "решение в срок ⚡",
+        ),
+    ),
+    Scenario(
+        code="program_submission_deadline_minus_3_project_not_submitted",
+        trigger=TriggerType.PROGRAM_SUBMISSION_DEADLINE,
+        offset_days=3,
+        template_name="email/generic-template-0.html",
+        subject="До сдачи итогового решения осталось 3 дня",
+        recipient_rule=RecipientRule.PROJECT_NOT_SUBMITTED,
+        context_builder=_build_project_not_submitted_context(
+            "До сдачи итогового решения осталось 3 дня",
+            "Работа в самом разгаре, и мы запускаем обратный отсчет. "
+            "Осталось всего 3 дня, чтобы доработать проект, оформить презентацию "
+            "и загрузить итоговое решение на платформу.",
+        ),
+    ),
+    Scenario(
+        code="program_submission_deadline_minus_1_project_not_submitted",
+        trigger=TriggerType.PROGRAM_SUBMISSION_DEADLINE,
+        offset_days=1,
+        template_name="email/generic-template-0.html",
+        subject="1 день до сдачи итогового решения",
+        recipient_rule=RecipientRule.PROJECT_NOT_SUBMITTED,
+        context_builder=_build_project_not_submitted_context(
+            "1 день до сдачи итогового решения",
+            "День X совсем скоро. Осталось только внести последние штрихи и "
+            "загрузить итоговое решение на платформу.\n\n"
+            "По любым техническим вопросам всегда на связи @procollab_support\n\n"
+            "Удачи!",
+        ),
+    ),
 )
diff --git a/mailing/tasks.py b/mailing/tasks.py
@@ -9,7 +9,10 @@
 from mailing.utils import send_mass_mail_from_template
 from partner_programs.selectors import (
     program_participants,
+    program_participants_with_unsubmitted_project,
+    program_participants_without_project_registered_on,
     program_participants_without_project,
+    programs_with_registrations_on,
     programs_with_submission_deadline_on,
 )
 from procollab.celery import app
@@ -21,16 +24,24 @@ def _get_programs_for_scenario(scenario, target_date):
     match scenario.trigger:
         case TriggerType.PROGRAM_SUBMISSION_DEADLINE:
             return programs_with_submission_deadline_on(target_date)
+        case TriggerType.PROGRAM_REGISTRATION_DATE:
+            return programs_with_registrations_on(target_date)
         case _:
             raise ValueError(f"Unsupported trigger: {scenario.trigger}")
 
 
-def _get_recipients(scenario, program_id: int):
+def _get_recipients(scenario, program_id: int, target_date):
     match scenario.recipient_rule:
         case RecipientRule.ALL_PARTICIPANTS:
             return program_participants(program_id)
         case RecipientRule.NO_PROJECT_IN_PROGRAM:
             return program_participants_without_project(program_id)
+        case RecipientRule.NO_PROJECT_IN_PROGRAM_REGISTERED_ON_DATE:
+            return program_participants_without_project_registered_on(
+                program_id, target_date
+            )
+        case RecipientRule.PROJECT_NOT_SUBMITTED:
+            return program_participants_with_unsubmitted_project(program_id)
         case _:
             raise ValueError(f"Unsupported recipient rule: {scenario.recipient_rule}")
 
@@ -40,8 +51,8 @@ def _deadline_date(program):
     return timezone.localtime(deadline).date()
 
 
-def _send_scenario_for_program(scenario, program, scheduled_for):
-    recipients = _get_recipients(scenario, program.id)
+def _send_scenario_for_program(scenario, program, scheduled_for, target_date):
+    recipients = _get_recipients(scenario, program.id, target_date)
     if not recipients.exists():
         return 0
 
@@ -88,10 +99,14 @@ def _send_scenario_for_program(scenario, program, scheduled_for):
     ]
     MailingScenarioLog.objects.bulk_create(logs, ignore_conflicts=True)
 
-    deadline_date = _deadline_date(program)
+    reference_date = (
+        _deadline_date(program)
+        if scenario.trigger == TriggerType.PROGRAM_SUBMISSION_DEADLINE
+        else target_date
+    )
 
     def context_builder(user):
-        return scenario.context_builder(program, user, deadline_date)
+        return scenario.context_builder(program, user, reference_date)
 
     sent_count = 0
     failed_count = 0
@@ -237,9 +252,14 @@ def run_program_mailings() -> int:
     today = timezone.localdate()
     total_sent = 0
     for scenario in SCENARIOS:
-        target_date = today + timedelta(days=scenario.offset_days)
+        if scenario.trigger == TriggerType.PROGRAM_SUBMISSION_DEADLINE:
+            target_date = today + timedelta(days=scenario.offset_days)
+        else:
+            target_date = today - timedelta(days=scenario.offset_days)
         programs = _get_programs_for_scenario(scenario, target_date)
         for program in programs:
-            total_sent += _send_scenario_for_program(scenario, program, today)
+            total_sent += _send_scenario_for_program(
+                scenario, program, today, target_date
+            )
     logger.info("Program mailings sent: %s", total_sent)
     return total_sent
diff --git a/partner_programs/selectors.py b/partner_programs/selectors.py
@@ -21,6 +21,12 @@ def programs_with_submission_deadline_on(target_date):
     )
 
 
+def programs_with_registrations_on(target_date):
+    return PartnerProgram.objects.filter(
+        partner_program_profiles__datetime_created__date=target_date
+    ).distinct()
+
+
 def _participant_profiles(program_id: int):
     return PartnerProgramUserProfile.objects.filter(
         partner_program_id=program_id, user__isnull=False
@@ -52,3 +58,43 @@ def program_participants_without_project(program_id: int):
         .values_list("user_id", flat=True)
     )
     return User.objects.filter(id__in=eligible_ids).distinct()
+
+
+def program_participants_without_project_registered_on(program_id: int, target_date):
+    profiles = _participant_profiles(program_id).filter(
+        datetime_created__date=target_date
+    )
+    leader_exists = Exists(
+        PartnerProgramProject.objects.filter(
+            partner_program_id=program_id,
+            project__leader_id=OuterRef("user_id"),
+        )
+    )
+    collab_exists = Exists(
+        Collaborator.objects.filter(
+            user_id=OuterRef("user_id"),
+            project__program_links__partner_program_id=program_id,
+        )
+    )
+    eligible_ids = (
+        profiles.annotate(is_leader=leader_exists, is_collab=collab_exists)
+        .filter(is_leader=False, is_collab=False)
+        .values_list("user_id", flat=True)
+    )
+    return User.objects.filter(id__in=eligible_ids).distinct()
+
+
+def program_participants_with_unsubmitted_project(program_id: int):
+    participant_ids = _participant_profiles(program_id).values_list(
+        "user_id", flat=True
+    )
+    leader_ids = PartnerProgramProject.objects.filter(
+        partner_program_id=program_id, submitted=False
+    ).values_list("project__leader_id", flat=True)
+    collab_ids = Collaborator.objects.filter(
+        project__program_links__partner_program_id=program_id,
+        project__program_links__submitted=False,
+    ).values_list("user_id", flat=True)
+    return User.objects.filter(id__in=participant_ids).filter(
+        Q(id__in=leader_ids) | Q(id__in=collab_ids)
+    ).distinct()
diff --git a/procollab/celery.py b/procollab/celery.py
@@ -20,7 +20,7 @@
     },
     "program_scenarios_mailings": {
         "task": "mailing.tasks.run_program_mailings",
-        "schedule": crontab(minute=0, hour=12),
+        "schedule": crontab(minute=0, hour=10),
     },
     "publish_finished_program_projects": {
         "task": "partner_programs.tasks.publish_finished_program_projects_task",
diff --git a/templates/email/generic-template-0.html b/templates/email/generic-template-0.html
@@ -261,7 +261,7 @@
                   
       <div
          style="font-family:sans-serif;font-size:20px;font-weight:400;line-height:150%;text-align:left;color:#8c888a;"
-      >{{ text }}</div>
+      >{{ text|linebreaksbr }}</div>
     
                 </td>
               </tr>
@@ -316,6 +316,7 @@
       >
         <tbody>
           
+              {% if button_text and button_link %}
               <tr>
                 <td
                    align="center" vertical-align="middle" style="font-size:0px;padding:0;word-break:break-word;"
@@ -341,6 +342,7 @@
     
                 </td>
               </tr>
+              {% endif %}
             
         </tbody>
       </table>
@@ -453,4 +455,4 @@
     
   </body>
 </html>
-  
+  
