Merge pull request #21 from Downtownapp/master

Hotfix: savepoints
05bit · Feb 24, 2016 · ea038fa · ea038fa
2 parents f2a3ef9 + 3121797
commit ea038fa
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/peewee_async.py b/peewee_async.py
@@ -781,15 +781,15 @@ def __init__(self, db, sid=None):
 
     @asyncio.coroutine
     def commit(self):
-        yield from _run_sql('RELEASE SAVEPOINT %s;' % self.quoted_sid)
+        yield from _run_sql(self.db, 'RELEASE SAVEPOINT %s;' % self.quoted_sid)
 
     @asyncio.coroutine
     def rollback(self):
-        yield from _run_sql('ROLLBACK TO SAVEPOINT %s;' % self.quoted_sid)
+        yield from _run_sql(self.db, 'ROLLBACK TO SAVEPOINT %s;' % self.quoted_sid)
 
     @asyncio.coroutine
     def __aenter__(self):
-        yield from _run_sql('SAVEPOINT %s;' % self.quoted_sid)
+        yield from _run_sql(self.db, 'SAVEPOINT %s;' % self.quoted_sid)
         return self
 
     @asyncio.coroutine