Fix error handling of mysql_real_connect (#317)

Fixes #316
PyMySQL · Dec 17, 2018 · ea80e8f · ea80e8f
1 parent ecf6d53
commit ea80e8f
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/MySQLdb/_mysql.c b/MySQLdb/_mysql.c
@@ -446,6 +446,7 @@ _mysql_ConnectionObject_Initialize(
 
     Py_BEGIN_ALLOW_THREADS ;
     conn = mysql_init(&(self->connection));
+    self->open = 1;
     if (connect_timeout) {
         unsigned int timeout = connect_timeout;
         mysql_options(&(self->connection), MYSQL_OPT_CONNECT_TIMEOUT,
@@ -496,6 +497,7 @@ _mysql_ConnectionObject_Initialize(
 
     if (!conn) {
         _mysql_Exception(self);
+        self->open = 0;
         return -1;
     }
 
@@ -515,7 +517,6 @@ _mysql_ConnectionObject_Initialize(
       be done here. tp_dealloc still needs to call PyObject_GC_UnTrack(),
       however.
     */
-    self->open = 1;
     return 0;
 }