feat(resources): add email notifications for contact and demo requests

- Add email notification functionality for tool demo requests to notify the tool owner.
- Add email notification for contact to admin

Author: sudip-khanal

apps/resources/graphql/inputs.py

@@ -17,6 +17,6 @@ class RequestDemoInput:
     email: str
     content: str
     national_society: str
-    tool: int
+    tool: strawberry.ID
     captcha_hashkey: str
     captcha_code: str

apps/resources/serializers.py

@@ -1,11 +1,14 @@
 import typing
 
 from captcha.serializers import CaptchaModelSerializer
+from django.db import transaction
 from rest_framework import serializers
 
 from apps.resources.models import ContactRequest, RequestDemo
 from apps.tool_picker.models import Tool
 
+from .tasks import send_contact_request_email, send_demo_request_email
+
 
 class ContactRequestSerializer(CaptchaModelSerializer):
     captcha_code = serializers.CharField(write_only=True)
@@ -26,7 +29,11 @@ class Meta:
     def create(self, validated_data: dict[str, typing.Any]):
         validated_data.pop("captcha_code", None)
         validated_data.pop("captcha_hashkey", None)
-        return super().create(validated_data)
+        instance = super().create(validated_data)
+        transaction.on_commit(
+            lambda: send_contact_request_email.delay(instance.id),  # type: ignore[reportIncompatibleVariableOverride]
+        )
+        return instance
 
 
 class RequestDemoSerializer(CaptchaModelSerializer):
@@ -53,4 +60,8 @@ class Meta:
     def create(self, validated_data: dict[str, typing.Any]):
         validated_data.pop("captcha_code", None)
         validated_data.pop("captcha_hashkey", None)
-        return super().create(validated_data)
+        instance = super().create(validated_data)
+        transaction.on_commit(
+            lambda: send_demo_request_email.delay(instance.id),  # type: ignore[reportIncompatibleVariableOverride]
+        )
+        return instance

apps/resources/tasks.py

@@ -0,0 +1,57 @@
+import logging
+
+from celery import shared_task
+from django.conf import settings
+from django.template.loader import render_to_string
+
+from apps.resources.models import ContactRequest, RequestDemo
+from apps.resources.utils import get_contact_request_email_context, get_demo_request_email_context
+from utils.email.service import send_email
+
+logger = logging.getLogger(__name__)
+
+
+@shared_task
+def send_contact_request_email(contact_id: int):
+    instance = ContactRequest.objects.filter(id=contact_id).first()
+    if not instance:
+        return None
+
+    recipient = settings.EMAIL_TO
+    email_subject = "You got a new feedback!"
+    email_context = get_contact_request_email_context(instance)
+    html_body = render_to_string("email/contact_request.html", email_context)
+
+    send_email(
+        subject=email_subject,
+        to_email=[recipient],
+        html=html_body,
+    )
+    return True
+
+
+@shared_task
+def send_demo_request_email(request_demo_id: int):
+    instance = (
+        RequestDemo.objects.select_related("tool").prefetch_related("tool__tool_owners").filter(id=request_demo_id).first()
+    )
+    if not instance:
+        return None
+
+    tool_owners = instance.tool.tool_owners.all()
+
+    if not tool_owners.exists():
+        logger.info("skipping cause there are no tool owners associated with this tool")
+        return False
+
+    recipients = list(tool_owners.values_list("email", flat=True))
+    email_subject = "You got a new request for demo!"
+    email_context = get_demo_request_email_context(instance)
+    html = render_to_string("email/request_demo.html", email_context)
+    send_email(
+        subject=email_subject,
+        to_email=recipients,
+        html=html,
+        cc_email=[settings.EMAIL_TO],
+    )
+    return True

apps/resources/tests/mutation_test.py

@@ -1,4 +1,5 @@
 import typing
+from unittest.mock import call, patch
 
 from captcha.models import CaptchaStore
 
@@ -44,14 +45,16 @@ def setUpClass(cls):
         cls.user = UserFactory.create(email="test@gmail.com")
 
     def _create_contact_request_mutation(self, data: dict[str, str], **kwargs: typing.Any):
-        return self.query_check(
-            query=self.Mutation.CREATE_CONTACT_REQUEST,
-            variables={
-                "data": data,
-            },
-        )
-
-    def test_create_contact_request(self):
+        with self.captureOnCommitCallbacks(execute=True):
+            return self.query_check(
+                query=self.Mutation.CREATE_CONTACT_REQUEST,
+                variables={
+                    "data": data,
+                },
+            )
+
+    @patch("apps.resources.serializers.send_contact_request_email.delay")
+    def test_create_contact_request(self, mock_requests):  # type: ignore[reportMissingParameterType]
         # Generates CAPTCHA key and value
         captcha_key = CaptchaStore.generate_key()
         captcha_obj = CaptchaStore.objects.get(hashkey=captcha_key)
@@ -81,6 +84,9 @@ def test_create_contact_request(self):
             "nationalSociety": contact_request.national_society,
         }
 
+        mock_requests.assert_called_once()
+        mock_requests.assert_has_calls([call(contact_request.pk)])
+
     def test_create_contact_request_without_captcha(self):
         contact_request_data = {
             "name": "John",
@@ -157,14 +163,16 @@ def setUpClass(cls):
         cls.tool = ToolFactory.create()
 
     def _create_tool_demo_request_mutation(self, data: dict[str, str | int], **kwargs: typing.Any):
-        return self.query_check(
-            query=self.Mutation.CREATE_DEMO_REQUEST,
-            variables={
-                "data": data,
-            },
-        )
-
-    def test_create_tool_demo_request(self):
+        with self.captureOnCommitCallbacks(execute=True):
+            return self.query_check(
+                query=self.Mutation.CREATE_DEMO_REQUEST,
+                variables={
+                    "data": data,
+                },
+            )
+
+    @patch("apps.resources.serializers.send_demo_request_email.delay")
+    def test_create_tool_demo_request(self, mock_requests):  # type: ignore[reportMissingParameterType]
         captcha_key = CaptchaStore.generate_key()
         captcha_obj = CaptchaStore.objects.get(hashkey=captcha_key)
         captcha_code = captcha_obj.response
@@ -196,6 +204,8 @@ def test_create_tool_demo_request(self):
                 "pk": self.gID(demo_request.tool.pk),
             },
         }
+        mock_requests.assert_called_once()
+        mock_requests.assert_has_calls([call(demo_request.pk)])
 
     def test_create_tool_demo_request_without_captcha(self):
         tool_demo_request_data = {

apps/resources/utils.py

@@ -0,0 +1,28 @@
+from apps.resources.models import ContactRequest, RequestDemo
+
+
+def get_contact_request_email_context(instance: ContactRequest):
+    from .serializers import ContactRequestSerializer
+
+    data = dict(ContactRequestSerializer(instance).data)
+
+    return {
+        "name": data["name"],
+        "email": data["email"],
+        "national_society": data["national_society"],
+        "content": data["content"],
+    }
+
+
+def get_demo_request_email_context(instance: RequestDemo):
+    from .serializers import RequestDemoSerializer
+
+    data = dict(RequestDemoSerializer(instance).data)
+    tool_name = instance.tool.name
+    return {
+        "name": data["name"],
+        "email": data["email"],
+        "national_society": data["national_society"],
+        "content": data["content"],
+        "tool": tool_name,
+    }

schema.graphql

@@ -285,7 +285,7 @@ input RequestDemoInput {
   email: String!
   content: String!
   nationalSociety: String!
-  tool: Int!
+  tool: ID!
   captchaHashkey: String!
   captchaCode: String!
 }