Fixing AttributeError in UnixDomainSocketConnection (redis#1903)

dvora-h · Feb 2, 2022 · 5222d6a · 5222d6a
1 parent 7f9c1e8
commit 5222d6a
Showing 1 changed file with 12 additions and 5 deletions.
diff --git a/redis/connection.py b/redis/connection.py
@@ -677,12 +677,19 @@ def _error_message(self, exception):
         # args for socket.error can either be (errno, "message")
         # or just "message"
         if len(exception.args) == 1:
-            return f"Error connecting to {self.host}:{self.port}. {exception.args[0]}."
+            try:
+                return f"Error connecting to {self.host}:{self.port}. \
+                        {exception.args[0]}."
+            except AttributeError:
+                return f"Connection Error: {exception.args[0]}"
         else:
-            return (
-                f"Error {exception.args[0]} connecting to "
-                f"{self.host}:{self.port}. {exception.args[1]}."
-            )
+            try:
+                return (
+                    f"Error {exception.args[0]} connecting to "
+                    f"{self.host}:{self.port}. {exception.args[1]}."
+                )
+            except AttributeError:
+                return f"Connection Error: {exception.args[0]}"
 
     def on_connect(self):
         "Initialize the connection, authenticate and select a database"