Merge pull request #28 from dgelvin/master

Fix for Proxy database support
05bit · May 28, 2016 · 90e80a4 · 90e80a4
2 parents ee93457 + b65b4d1
commit 90e80a4
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/peewee_async.py b/peewee_async.py
@@ -347,7 +347,9 @@ def _swap_database(self, query):
         If query database can't be swapped and differs from manager's
         database, it's **WRONG AND DANGEROUS**, so assertion is raised.
         """
-        if query.database == self.database:
+        if query.database == self.database or \
+            (isinstance(query.database, peewee.Proxy) and
+             query.database.obj == self.database):
             return query
         elif self._subclassed(peewee.PostgresqlDatabase,
                               query.database,