Refactor error handling to show clean authentication messages and hide technical details

Author: root

vertica_python/vertica/connection.py

@@ -44,6 +44,7 @@
 import ssl
 import uuid
 import warnings
+import re
 from collections import deque
 from struct import unpack
 
@@ -963,13 +964,21 @@ def send_startup(totp_value=None):
                             self.backend_key = message.key
                         elif isinstance(message, messages.ErrorResponse):
                             error_msg = message.error_message()
-                            if "Invalid TOTP" in error_msg:
+
+                            # Extract only the "Message: ..." part
+                            match = re.search(r'Message: (.+?)(?:, Sqlstate|$)', error_msg, re.DOTALL)
+                            short_msg = match.group(1).strip() if match else error_msg.strip()
+
+                            if "Invalid TOTP" in short_msg:
+                                print("❌ Authentication failed: Invalid TOTP token.")
                                 self._logger.error("❌ Authentication failed: Invalid TOTP token.")
                                 self.close_socket()
                                 raise errors.ConnectionError("Authentication failed: Invalid TOTP token.")
+
                             # Generic error fallback
-                            self._logger.error(error_msg)
-                            raise errors.ConnectionError("❌ Authentication failed due to server error.")
+                            print(f"❌ Authentication failed: {short_msg}")
+                            self._logger.error(short_msg)
+                            raise errors.ConnectionError(f"❌ Authentication failed: {short_msg}")
                         else:
                             self._logger.warning(f"⚠️  Unexpected message type: {type(message).__name__}")