Add email body_html template (django-otp#103)

djangoflow · Jan 2, 2023 · c3fa634 · c3fa634
1 parent ac02073
commit c3fa634
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/src/django_otp/plugins/otp_email/conf.py b/src/django_otp/plugins/otp_email/conf.py
@@ -14,6 +14,8 @@ class OTPEmailSettings:
         'OTP_EMAIL_SUBJECT': 'OTP token',
         'OTP_EMAIL_BODY_TEMPLATE': None,
         'OTP_EMAIL_BODY_TEMPLATE_PATH': 'otp/email/token.txt',
+        'OTP_EMAIL_BODY_HTML_TEMPLATE': None,
+        'OTP_EMAIL_BODY_HTML_TEMPLATE_PATH': None,
         'OTP_EMAIL_TOKEN_VALIDITY': 300,
         'OTP_EMAIL_THROTTLE_FACTOR': 1,
     }

diff --git a/src/django_otp/plugins/otp_email/models.py b/src/django_otp/plugins/otp_email/models.py
@@ -64,10 +64,18 @@ def generate_challenge(self, extra_context=None):
         else:
             body = get_template(settings.OTP_EMAIL_BODY_TEMPLATE_PATH).render(context)
 
+        if settings.OTP_EMAIL_BODY_HTML_TEMPLATE:
+            body_html = Template(settings.OTP_EMAIL_BODY_HTML_TEMPLATE).render(Context(context))
+        elif settings.OTP_EMAIL_BODY_HTML_TEMPLATE_PATH:
+            body_html = get_template(settings.OTP_EMAIL_BODY_HTML_TEMPLATE_PATH).render(context)
+        else:
+            body_html = None
+
         send_mail(settings.OTP_EMAIL_SUBJECT,
                   body,
                   settings.OTP_EMAIL_SENDER,
-                  [self.email or self.user.email])
+                  [self.email or self.user.email],
+                  html_message=body_html)
 
         message = "sent by email"