8354d07c-7eab-4972-926d-a2104e534166

from django.apps import apps
from django.conf import settings
from django.contrib.auth.models import User
from django.dispatch import receiver

from allauth.account.signals import user_signed_up, user_logged_in
from django.http.request import HttpRequest
from mypy_boto3_ses.client import SESClient
from mypy_boto3_ses.type_defs import ContentTypeDef
from waffle import flag_is_active

from emails.apps import EmailsConfig
from emails.utils import get_welcome_email, incr_if_enabled

from .ftl_bundles import main as ftl_bundle


@receiver(user_signed_up)
def record_user_signed_up(request, user, **kwargs):
    incr_if_enabled("user_signed_up", 1)
    # the user_signed_up signal doesn't have access to the response object
    # so we have to set a user_created session var for user_logged_in receiver
    request.session["user_created"] = True
    request.session.modified = True


def _ses_message_props(data: str) -> ContentTypeDef:
    return {"Charset": "UTF-8", "Data": data}


@receiver(user_signed_up)
def send_first_email(request: HttpRequest, user: User, **kwargs):
    if not flag_is_active(request, "welcome_email"):
        return
    app_config = apps.get_app_config("emails")
    assert isinstance(app_config, EmailsConfig)
    ses_client = app_config.ses_client
    assert ses_client
    assert settings.RELAY_FROM_ADDRESS
    ses_client.send_email(
        Destination={
            "ToAddresses": [user.email],
        },
        Source=settings.RELAY_FROM_ADDRESS,
        Message={
            "Subject": _ses_message_props(
                ftl_bundle.format("first-time-user-email-welcome")
            ),
            "Body": {
                "Html": _ses_message_props(get_welcome_email(request, user, "html")),
                "Text": _ses_message_props(get_welcome_email(request, user, "txt")),
            },
        },
    )


@receiver(user_logged_in)
def record_user_logged_in(request, user, **kwargs):
    incr_if_enabled("user_logged_in", 1)
    response = kwargs.get("response")
    event = "user_logged_in"
    # the user_signed_up signal doesn't have access to the response object
    # so we have to check for user_created session var from user_signed_up
    if request.session.get("user_created", False):
        event = "user_signed_up"
    if response:
        response.set_cookie(f"server_ga_event:{event}", event, max_age=5)

1	from django.apps import apps	1✔
2	from django.conf import settings	1✔
3	from django.contrib.auth.models import User	1✔
4	from django.dispatch import receiver	1✔
5
6	from allauth.account.signals import user_signed_up, user_logged_in	1✔
7	from django.http.request import HttpRequest	1✔
8	from mypy_boto3_ses.client import SESClient	1✔
9	from mypy_boto3_ses.type_defs import ContentTypeDef	1✔
10	from waffle import flag_is_active	1✔
11
12	from emails.apps import EmailsConfig	1✔
13	from emails.utils import get_welcome_email, incr_if_enabled	1✔
14
15	from .ftl_bundles import main as ftl_bundle	1✔
16
17
18	@receiver(user_signed_up)	1✔
19	def record_user_signed_up(request, user, **kwargs):	1✔
20	incr_if_enabled("user_signed_up", 1)	1✔
21	# the user_signed_up signal doesn't have access to the response object
22	# so we have to set a user_created session var for user_logged_in receiver
23	request.session["user_created"] = True	1✔
24	request.session.modified = True	1✔
25
26
27	def _ses_message_props(data: str) -> ContentTypeDef:	1✔
28	return {"Charset": "UTF-8", "Data": data}	1✔
29
30
31	@receiver(user_signed_up)	1✔
32	def send_first_email(request: HttpRequest, user: User, **kwargs):	1✔
33	if not flag_is_active(request, "welcome_email"):	1✔
34	return	1✔
35	app_config = apps.get_app_config("emails")	1✔
36	assert isinstance(app_config, EmailsConfig)	1✔
37	ses_client = app_config.ses_client	1✔
38	assert ses_client	1✔
39	assert settings.RELAY_FROM_ADDRESS	1✔
40	ses_client.send_email(	1✔
41	Destination={
42	"ToAddresses": [user.email],
43	},
44	Source=settings.RELAY_FROM_ADDRESS,
45	Message={
46	"Subject": _ses_message_props(
47	ftl_bundle.format("first-time-user-email-welcome")
48	),
49	"Body": {
50	"Html": _ses_message_props(get_welcome_email(request, user, "html")),
51	"Text": _ses_message_props(get_welcome_email(request, user, "txt")),
52	},
53	},
54	)
55
56
57	@receiver(user_logged_in)	1✔
58	def record_user_logged_in(request, user, **kwargs):	1✔
59	incr_if_enabled("user_logged_in", 1)	×
60	response = kwargs.get("response")	×
61	event = "user_logged_in"	×
62	# the user_signed_up signal doesn't have access to the response object
63	# so we have to check for user_created session var from user_signed_up
64	if request.session.get("user_created", False):	×
65	event = "user_signed_up"	×
66	if response:	×
67	response.set_cookie(f"server_ga_event:{event}", event, max_age=5)	×

mozilla / fx-private-relay / 8354d07c-7eab-4972-926d-a2104e534166

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous