Merge pull request #421 from PROCOLLAB-github/feature/email-program-user-register

Feature/email program user register

Author: sh1nkey

partner_programs/views.py

@@ -18,6 +18,11 @@
     PartnerProgramNewUserSerializer,
     PartnerProgramUserSerializer,
 )
+from vacancy.mapping import (
+    MessageTypeEnum,
+    UserProgramRegisterParamsDict,
+)
+from vacancy.tasks import send_email
 
 User = get_user_model()
 
@@ -103,7 +108,7 @@ def post(self, request, *args, **kwargs):
                 "onboarding_stage": None,  # bypass onboarding
                 "verification_date": timezone.now(),  # bypass ClickUp verification
                 **{field_name: data.get(field_name, "") for field_name in user_fields},
-            }
+            },
         )
         if created:  # Only when registering a new user.
             user.set_password(password)
@@ -123,6 +128,16 @@ def post(self, request, *args, **kwargs):
                 data={"detail": "User has already registered in this program."},
                 status=status.HTTP_400_BAD_REQUEST,
             )
+
+        send_email.delay(
+            UserProgramRegisterParamsDict(
+                message_type=MessageTypeEnum.REGISTERED_PROGRAM_USER.value,
+                user_id=user.id,
+                program_name=program.name,
+                program_id=program.id,
+                schema_id=2,
+            )
+        )
         return Response(status=status.HTTP_201_CREATED)
 
     def get(self, request, *args, **kwargs):
@@ -134,7 +149,7 @@ class PartnerProgramRegister(generics.GenericAPIView):
     Register user to program and save additional program data
     """
 
-    queryset = PartnerProgram.objects.none()
+    queryset = PartnerProgram.objects.all()
     permission_classes = [IsAuthenticated]
     serializer_class = PartnerProgramUserSerializer
 
@@ -156,6 +171,16 @@ def post(self, request, *args, **kwargs):
             )
             added_user_profile.save()
 
+            send_email.delay(
+                UserProgramRegisterParamsDict(
+                    message_type=MessageTypeEnum.REGISTERED_PROGRAM_USER.value,
+                    user_id=user_to_add.id,
+                    program_name=program.name,
+                    program_id=program.id,
+                    schema_id=2,
+                )
+            )
+
             return Response(status=status.HTTP_201_CREATED)
         except PartnerProgram.DoesNotExist:
             return Response(status=status.HTTP_404_NOT_FOUND)

vacancy/mapping.py

@@ -1,7 +1,9 @@
 from enum import Enum
 from typing import Literal, TypeAlias, TypedDict, Annotated
 
-EmailTextTypes: TypeAlias = Literal["accepted", "outdated", "responded", "rejected"]
+EmailTextTypes: TypeAlias = Literal[
+    "accepted", "outdated", "responded", "rejected", "registered_program"
+]
 
 
 class MessageTypeEnum(Enum):
@@ -10,61 +12,81 @@ class MessageTypeEnum(Enum):
     RESPONDED = "responded"
     REJECTED = "rejected"
 
+    REGISTERED_PROGRAM_USER = "registered_program"
 
-class CeleryEmailParamsDict(TypedDict):
+
+class BaseEmailCeleryParamsDict(TypedDict):
     message_type: EmailTextTypes
     user_id: int
+    schema_id: int  # = 2
+
+
+class CeleryEmailParamsDict(BaseEmailCeleryParamsDict):
     project_name: str
     project_id: int
     vacancy_role: str
-    schema_id: int  # = 2
+
+
+class UserProgramRegisterParamsDict(BaseEmailCeleryParamsDict):
+    program_name: str
+    program_id: int
 
 
 message_type_to_button_text: dict[Annotated[str, EmailTextTypes], str] = {
     MessageTypeEnum.ACCEPTED.value: "Посмотреть на проект",
     MessageTypeEnum.RESPONDED.value: "Посмотреть на отклики",
     MessageTypeEnum.OUTDATED.value: "Посмотреть на вакансию",
     MessageTypeEnum.REJECTED.value: "Посмотреть на проект",
+    MessageTypeEnum.REGISTERED_PROGRAM_USER.value: "Посмотреть на программу",
 }
 
 message_type_to_title: dict[Annotated[str, EmailTextTypes], str] = {
     MessageTypeEnum.ACCEPTED.value: "На ваш отклик ответили",
     MessageTypeEnum.RESPONDED.value: "На вашу вакансию откликнулись",
     MessageTypeEnum.OUTDATED.value: "У вашей вакансии истёк срок годности",
     MessageTypeEnum.REJECTED.value: "На ваш отклик ответили отказом",
+    MessageTypeEnum.REGISTERED_PROGRAM_USER.value: "Вы зарегистрировались на программу",
 }
 
 
-def get_link(data: CeleryEmailParamsDict):
+def get_link(data: CeleryEmailParamsDict | UserProgramRegisterParamsDict):
     match data["message_type"]:
-        case "accepted":
+        case MessageTypeEnum.ACCEPTED.value:
             return f"https://app.procollab.ru/office/projects/{data['project_id']}"
-        case "responded":
+        case MessageTypeEnum.RESPONDED.value:
             return (
                 f"https://app.procollab.ru/office/projects/{data['project_id']}/responses"
             )
-        case "outdated":
+        case MessageTypeEnum.OUTDATED.value:
             return f"https://app.procollab.ru/office/projects/{data['project_id']}/edit"
-        case "rejected":
+        case MessageTypeEnum.REJECTED.value:
             return f"https://app.procollab.ru/office/projects/{data['project_id']}"
+        case MessageTypeEnum.REGISTERED_PROGRAM_USER.value:
+            return f"https://app.procollab.ru/office/program/{data['program_id']}"
 
 
-def create_text_for_email(data: CeleryEmailParamsDict) -> str:
+def create_text_for_email(
+    data: CeleryEmailParamsDict | UserProgramRegisterParamsDict,
+) -> str:
     match data["message_type"]:
-        case "accepted":
+        case MessageTypeEnum.ACCEPTED.value:
             return f"""
         Ваш отклик на роль {data["vacancy_role"]} в проекте "{data["project_name"]}" не остался незамеченным.
         Вас готовы принять в команду!
         """
-        case "responded":
+        case MessageTypeEnum.RESPONDED.value:
             return f"На вакансию {data['vacancy_role']} для проекта '{data['project_name']}' оставили отклик."
-        case "outdated":
+        case MessageTypeEnum.OUTDATED.value:
             return f"""
             У вакансии {data["vacancy_role"]} для проекта '{data["project_name"]}' истёк срок годности.\n
             Она более не будет показываться в поиске.
     """
-        case "rejected":
+        case MessageTypeEnum.REJECTED.value:
             return f"""
             К сожалению, на ваш отклик на вакансию {data['vacancy_role']}
             для проекта '{data['project_name']}' ответили отказом.
     """
+        case MessageTypeEnum.REGISTERED_PROGRAM_USER.value:
+            return f"""
+            Вы успешно зарегистрировались на программу {data['program_name']}
+    """