Correct functions for creating new users, confirming email, resetting password and forgot password.

common/utils/security_utils.py

@@ -1,6 +1,7 @@
 from flask import current_app, render_template
 from flask_mailman import EmailMessage
 from itsdangerous import URLSafeTimedSerializer
+import socket
 
 from common.utils.nginx_utils import prefixed_url_for
 
@@ -35,15 +36,66 @@ def generate_confirmation_token(email):
 
 def send_confirmation_email(user):
     current_app.logger.debug(f'Sending confirmation email to {user.email}')
+
+    if not test_smtp_connection():
+        raise Exception("Failed to connect to SMTP server")
+
     token = generate_confirmation_token(user.email)
     confirm_url = prefixed_url_for('security_bp.confirm_email', token=token, _external=True)
     current_app.logger.debug(f'Confirmation URL: {confirm_url}')
+
     html = render_template('email/activate.html', confirm_url=confirm_url)
-    send_email(user.email, "Confirm your email", html)
+    subject = "Please confirm your email"
+
+    try:
+        send_email(user.email, "Confirm your email", html)
+        current_app.logger.info(f'Confirmation email sent to {user.email}')
+    except Exception as e:
+        current_app.logger.error(f'Failed to send confirmation email to {user.email}. Error: {str(e)}')
+        raise
 
 
 def send_reset_email(user):
+    current_app.logger.debug(f'Sending reset email to {user.email}')
     token = generate_reset_token(user.email)
     reset_url = prefixed_url_for('security_bp.reset_password', token=token, _external=True)
+    current_app.logger.debug(f'Reset URL: {reset_url}')
+
     html = render_template('email/reset_password.html', reset_url=reset_url)
-    send_email(user.email, "Reset Your Password", html)
+    subject = "Reset Your Password"
+
+    try:
+        send_email(user.email, "Reset Your Password", html)
+        current_app.logger.info(f'Reset email sent to {user.email}')
+    except Exception as e:
+        current_app.logger.error(f'Failed to send reset email to {user.email}. Error: {str(e)}')
+        raise
+
+
+def test_smtp_connection():
+    try:
+        current_app.logger.info(f"Attempting to resolve google.com...")
+        google_ip = socket.gethostbyname('google.com')
+        current_app.logger.info(f"Successfully resolved google.com to {google_ip}")
+    except Exception as e:
+        current_app.logger.error(f"Failed to resolve google.com: {str(e)}")
+
+    try:
+        smtp_server = current_app.config['MAIL_SERVER']
+        current_app.logger.info(f"Attempting to resolve {smtp_server}...")
+        smtp_ip = socket.gethostbyname(smtp_server)
+        current_app.logger.info(f"Successfully resolved {smtp_server} to {smtp_ip}")
+    except Exception as e:
+        current_app.logger.error(f"Failed to resolve {smtp_server}: {str(e)}")
+
+    try:
+        smtp_server = current_app.config['MAIL_SERVER']
+        smtp_port = current_app.config['MAIL_PORT']
+        sock = socket.create_connection((smtp_server, smtp_port), timeout=10)
+        sock.close()
+        current_app.logger.info(f"Successfully connected to SMTP server {smtp_server}:{smtp_port}")
+        return True
+    except Exception as e:
+        current_app.logger.error(f"Failed to connect to SMTP server: {str(e)}")
+        return False
+