add settings for demo users email and mobile

Author: tiwariav

README.md

@@ -314,6 +314,12 @@ DEFAULTS = {
     # A dictionary of demo user's primary key mapped to their static pin
     'PASSWORDLESS_DEMO_USERS': {},
 
+    # A dictionary of demo user's email mapped to their static pin
+    'PASSWORDLESS_DEMO_USERS_EMAIL': {},
+
+    # A dictionary of demo user's mobile mapped to their static pin
+    'PASSWORDLESS_DEMO_USERS_MOBILE': {},
+
     # configurable function for sending email
     'PASSWORDLESS_EMAIL_CALLBACK': 'drfpasswordless.utils.send_email_with_callback_token',
     

drfpasswordless/services.py

@@ -1,17 +1,23 @@
 from django.utils.module_loading import import_string
+
 from drfpasswordless.settings import api_settings
-from drfpasswordless.utils import (
-    create_callback_token_for_user,
-)
+from drfpasswordless.utils import create_callback_token_for_user
 
 
 class TokenService(object):
     @staticmethod
     def send_token(user, alias_type, token_type, **message_payload):
-        token = create_callback_token_for_user(user, alias_type, token_type)
+        alias_type_u = alias_type.upper()
+        to_alias_field = getattr(
+            api_settings, f"PASSWORDLESS_USER_{alias_type_u}_FIELD_NAME"
+        )
+        to_alias = getattr(user, to_alias_field)
+        token = create_callback_token_for_user(user, alias_type, token_type, to_alias)
         send_action = None
 
-        if user.pk in api_settings.PASSWORDLESS_DEMO_USERS.keys():
+        if user.pk in api_settings.PASSWORDLESS_DEMO_USERS or to_alias in getattr(
+            api_settings, f"PASSWORDLESS_DEMO_USERS_{alias_type_u}"
+        ):
             return True
         if alias_type == 'email':
             send_action = import_string(api_settings.PASSWORDLESS_EMAIL_CALLBACK)

drfpasswordless/settings.py

@@ -85,6 +85,10 @@
 
     # A dictionary of demo user's primary key mapped to their static pin
     'PASSWORDLESS_DEMO_USERS': {},
+    # A dictionary of demo user's email/mobile mapped to their static pin
+    'PASSWORDLESS_DEMO_USERS_EMAIL': {},
+    'PASSWORDLESS_DEMO_USERS_MOBILE': {},
+
     'PASSWORDLESS_EMAIL_CALLBACK': 'drfpasswordless.utils.send_email_with_callback_token',
     'PASSWORDLESS_SMS_CALLBACK': 'drfpasswordless.utils.send_sms_with_callback_token',
 

drfpasswordless/signals.py

@@ -1,12 +1,11 @@
 import logging
 from django.contrib.auth import get_user_model
 from django.core.exceptions import ValidationError
-from django.dispatch import receiver
 from django.db.models import signals
-from drfpasswordless.models import CallbackToken
-from drfpasswordless.models import generate_numeric_token
-from drfpasswordless.settings import api_settings
+from django.dispatch import receiver
+from drfpasswordless.models import CallbackToken, generate_numeric_token
 from drfpasswordless.services import TokenService
+from drfpasswordless.settings import api_settings
 
 logger = logging.getLogger(__name__)
 
@@ -17,7 +16,9 @@ def invalidate_previous_tokens(sender, instance, created, **kwargs):
     Invalidates all previously issued tokens of that type when a new one is created, used, or anything like that.
     """
 
-    if instance.user.pk in api_settings.PASSWORDLESS_DEMO_USERS.keys():
+    if instance.user.pk in api_settings.PASSWORDLESS_DEMO_USERS or instance.to_alias in getattr(
+        api_settings, f"PASSWORDLESS_DEMO_USERS_{instance.to_alias_type}"
+    ):
         return
 
     if isinstance(instance, CallbackToken):

drfpasswordless/utils.py

@@ -35,34 +35,24 @@ def authenticate_by_token(callback_token):
     return None
 
 
-def create_callback_token_for_user(user, alias_type, token_type):
-    token = None
+def create_callback_token_for_user(user, alias_type, token_type, to_alias):
     alias_type_u = alias_type.upper()
-    to_alias_field = getattr(api_settings, f'PASSWORDLESS_USER_{alias_type_u}_FIELD_NAME')
-    if user.pk in api_settings.PASSWORDLESS_DEMO_USERS.keys():
-        token = CallbackToken.objects.filter(user=user).first()
-        if token:
-            return token
-        else:
-            return CallbackToken.objects.create(
-                user=user,
-                key=api_settings.PASSWORDLESS_DEMO_USERS[user.pk],
-                to_alias_type=alias_type_u,
-                to_alias=getattr(user, to_alias_field),
-                type=token_type
-            )
-    
-    token = CallbackToken.objects.create(user=user,
-                                            to_alias_type=alias_type_u,
-                                            to_alias=getattr(user, to_alias_field),
-                                            type=token_type)
-
-
-
-    if token is not None:
+    demo_key = api_settings.PASSWORDLESS_DEMO_USERS.get(user.pk) or getattr(
+        api_settings, f"PASSWORDLESS_DEMO_USERS_{alias_type_u}"
+    ).get(to_alias)
+    if demo_key:
+        token, _ = CallbackToken.objects.get_or_create(
+                                                    user=user,
+                                                    key=demo_key,
+                                                    to_alias_type=alias_type_u,
+                                                    to_alias=to_alias,
+                                                    type=token_type)
         return token
 
-    return None
+    return CallbackToken.objects.create(user=user,
+                                            to_alias_type=alias_type_u,
+                                            to_alias=to_alias,
+                                            type=token_type)
 
 
 def validate_token_age(callback_token):
@@ -74,7 +64,9 @@ def validate_token_age(callback_token):
         token = CallbackToken.objects.get(key=callback_token, is_active=True)
         seconds = (timezone.now() - token.created_at).total_seconds()
         token_expiry_time = api_settings.PASSWORDLESS_TOKEN_EXPIRE_TIME
-        if token.user.pk in api_settings.PASSWORDLESS_DEMO_USERS.keys():
+        if token.user.pk in api_settings.PASSWORDLESS_DEMO_USERS or token.to_alias in getattr(
+            api_settings, f"PASSWORDLESS_DEMO_USERS_{token.to_alias_type}"
+        ):
             return True
         if seconds <= token_expiry_time:
             return True