fix AsyncPostgresqlDatabase

snower · Mar 21, 2017 · b4d9920 · b4d9920
1 parent a7476cb
commit b4d9920
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/torpeewee/postgresql.py b/torpeewee/postgresql.py
@@ -41,7 +41,7 @@ def pop_execution_context(self):
     def execution_context_depth(self):
         raise NotImplementedError
 
-    def execution_context(self, with_transaction=True):
+    def execution_context(self, with_transaction=True, transaction_type=None):
         raise NotImplementedError
 
     def push_transaction(self, transaction):
@@ -54,7 +54,7 @@ def transaction_depth(self):
         raise NotImplementedError
 
     @gen.coroutine
-    def transaction(self):
+    def transaction(self, transaction_type=None):
         raise NotImplementedError
 
     def commit_on_success(self, func):
@@ -63,7 +63,7 @@ def commit_on_success(self, func):
     def savepoint(self, sid=None):
         raise NotImplementedError
 
-    def atomic(self):
+    def atomic(self, transaction_type=None):
         raise NotImplementedError
 
     def _get_pk_sequence(self, model):