Update base.py

akoidan · Apr 11, 2024 · 4ec61ac · 4ec61ac
1 parent c699aa5
commit 4ec61ac
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/mysql_server_has_gone_away/base.py b/mysql_server_has_gone_away/base.py
@@ -17,7 +17,8 @@ def wrapper(self, query, args=None):
             except (OperationalError, InterfaceError) as e:
                 logger.warn("MySQL server has gone away. Rerunning query: %s", query)
                 if (
-                    'server has gone away' in str(e) or
+                    'MySQL server has gone away' in str(e) or
+                    'Server has gone away' in str(e) or
                     'Lost connection to MySQL server during query' in str(e)
                 ):
                     db_wrapper.connection.close()