Merge pull request #47 from insolite/cursor-release-on-error

Prevent connection stuck after leaving cursor unreleased

peewee_async.py

@@ -575,8 +575,9 @@ def select(query):
             yield from result.fetchone()
     except GeneratorExit:
         pass
+    finally:
+        yield from cursor.release
 
-    yield from cursor.release
     return result
 
 
@@ -590,16 +591,18 @@ def insert(query):
 
     cursor = yield from _execute_query_async(query)
 
-    if query.is_insert_returning:
-        if query._return_id_list:
-            result = map(lambda x: x[0], (yield from cursor.fetchall()))
+    try:
+        if query.is_insert_returning:
+            if query._return_id_list:
+                result = map(lambda x: x[0], (yield from cursor.fetchall()))
+            else:
+                result = (yield from cursor.fetchone())[0]
         else:
-            result = (yield from cursor.fetchone())[0]
-    else:
-        result = yield from query.database.last_insert_id_async(
-            cursor, query.model_class)
+            result = yield from query.database.last_insert_id_async(
+                cursor, query.model_class)
+    finally:
+        yield from cursor.release
 
-    yield from cursor.release
     return result