Merge pull request #49 from kammala/patch-2

Compatibility fixes
05bit · Mar 5, 2017 · b16e3fe · b16e3fe
2 parents 9ba1e55 + 28e6b49
commit b16e3fe
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/peewee_async.py b/peewee_async.py
@@ -24,6 +24,12 @@
 logger = logging.getLogger('peewee.async')
 logger.addHandler(logging.NullHandler())
 
+
+def get_peewee_version_tuple():
+    return tuple(peewee.__version__.split('.'))
+
+PEEWEE_VERSION = get_peewee_version_tuple()
+
 try:
     import aiopg
 except ImportError:
@@ -1485,7 +1491,11 @@ def _run_sql(database, operation, *args, **kwargs):
     """
     logger.debug((operation, args, kwargs))
 
-    with database.exception_wrapper():
+    exception_wrapper = database.exception_wrapper
+    if PEEWEE_VERSION <= (2, 8, 5):
+        exception_wrapper = exception_wrapper()
+
+    with exception_wrapper:
         cursor = yield from database.cursor_async()
 
         try: