Enhance UserAdmin save logic to auto-generate email from mobile or user ID; improve token creation logging and user existence check

djangopasswordlessknox/utils.py

@@ -44,6 +44,29 @@ def authenticate_by_token(callback_token):
 
 
 def create_callback_token_for_user(user, token_type):
+    logger.info(
+        f"Attempting to create token for user: {user}, user.pk: {user.pk if user else 'None'}"
+    )
+    if user and user.pk:
+        try:
+            # Explicitly check if the user exists in the database
+            User.objects.get(pk=user.pk)
+            logger.info(
+                f"User with pk {user.pk} confirmed to exist in DB before token creation."
+            )
+        except User.DoesNotExist:
+            logger.error(
+                f"CRITICAL: User with pk {user.pk} DOES NOT EXIST in DB right before token creation for {user}!"
+            )
+            return None
+            # You might want to return None or raise an exception here to prevent the IntegrityError
+            # For example: return None
+    else:
+        logger.error(
+            f"CRITICAL: Invalid user object passed to create_callback_token_for_user: {user}"
+        )
+        # return None
+
     token = None
     token_type = token_type.upper()