Release cursor in raw_query() call after exception

05bit · Nov 30, 2016 · 917daf7 · insolite · Apr 20, 2017 · 917daf7
1 parent c2236f4
commit 917daf7
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/peewee_async.py b/peewee_async.py
@@ -578,6 +578,7 @@ def select(query):
     finally:
         yield from cursor.release
 
+    yield from cursor.release
     return result
 
 
@@ -689,8 +690,9 @@ def raw_query(query):
             yield from result.fetchone()
     except GeneratorExit:
         pass
+    finally:
+        yield from cursor.release
 
-    yield from cursor.release
     return result