Added context processor support for templates

Author: aaronn

drfpasswordless/__init__.py

@@ -1,7 +1,7 @@
 # -*- coding: utf-8 -*-
 
 __title__ = 'drfpasswordless'
-__version__ = '1.0.6'
+__version__ = '1.0.7'
 __author__ = 'Aaron Ng'
 __license__ = 'MIT'
 __copyright__ = 'Copyright 2017 Aaron Ng'

drfpasswordless/settings.py

@@ -49,12 +49,16 @@
     'PASSWORDLESS_REGISTER_NEW_USERS': True,
 
     # Suppresses actual SMS for testing
-    'PASSWORDLESS_TEST_SUPPRESSION': False
+    'PASSWORDLESS_TEST_SUPPRESSION': False,
+
+    # Context Processors for Email Template
+    'PASSWORDLESS_CONTEXT_PROCESSORS': [],
 }
 
 # List of settings that may be in string import notation.
 IMPORT_STRINGS = (
     'PASSWORDLESS_EMAIL_TOKEN_HTML_TEMPLATE',
+    'PASSWORDLESS_CONTEXT_PROCESSORS',
 )
 
 api_settings = APISettings(USER_SETTINGS, DEFAULTS, IMPORT_STRINGS)

drfpasswordless/utils.py

@@ -99,6 +99,15 @@ def verify_user_alias(user, token):
     return user
 
 
+def inject_template_context(context):
+    """
+    Injects additional context into email template.
+    """
+    for processor in api_settings.PASSWORDLESS_CONTEXT_PROCESSORS:
+        context.update(processor())
+    return context
+
+
 def send_email_with_callback_token(self, user, email_token):
     """
     Sends a SMS to user.mobile.
@@ -110,9 +119,12 @@ def send_email_with_callback_token(self, user, email_token):
         if api_settings.PASSWORDLESS_EMAIL_NOREPLY_ADDRESS:
             # Make sure we have a sending address before sending.
 
+            # Inject context if user specifies.
+            context = inject_template_context({'callback_token': email_token.key, })
+
             html_message = loader.render_to_string(
                 api_settings.PASSWORDLESS_EMAIL_TOKEN_HTML_TEMPLATE_NAME,
-                {'callback_token': email_token.key, }
+                context,
             )
             send_mail(
                 api_settings.PASSWORDLESS_EMAIL_SUBJECT,