Update base.py

support new error msg
akoidan · Apr 11, 2024 · c699aa5 · c699aa5
1 parent 9e1d422
commit c699aa5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mysql_server_has_gone_away/base.py b/mysql_server_has_gone_away/base.py
@@ -17,7 +17,7 @@ def wrapper(self, query, args=None):
             except (OperationalError, InterfaceError) as e:
                 logger.warn("MySQL server has gone away. Rerunning query: %s", query)
                 if (
-                    '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()