Skip to content

Commit 90e80a4

Browse files
committed
Merge pull request pytest-dev#28 from dgelvin/master
Fix for Proxy database support
2 parents ee93457 + b65b4d1 commit 90e80a4

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

peewee_async.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -347,7 +347,9 @@ def _swap_database(self, query):
347347
If query database can't be swapped and differs from manager's
348348
database, it's **WRONG AND DANGEROUS**, so assertion is raised.
349349
"""
350-
if query.database == self.database:
350+
if query.database == self.database or \
351+
(isinstance(query.database, peewee.Proxy) and
352+
query.database.obj == self.database):
351353
return query
352354
elif self._subclassed(peewee.PostgresqlDatabase,
353355
query.database,

0 commit comments

Comments
 (0)