Merge pull request pytest-dev#61 from mrbox/master

Make db_url work with async databases

Author: rudyryk

peewee_async.py

@@ -21,6 +21,8 @@
 import warnings
 import logging
 
+from playhouse.db_url import register_database
+
 logger = logging.getLogger('peewee.async')
 logger.addHandler(logging.NullHandler())
 
@@ -1172,6 +1174,9 @@ def use_speedups(self, value):
         pass
 
 
+register_database(PostgresqlDatabase, 'postgres+async', 'postgresql+async')
+
+
 class PooledPostgresqlDatabase(AsyncPostgresqlMixin, peewee.PostgresqlDatabase):
     """PosgreSQL database driver providing **single drop-in sync**
     connection and **async connections pool** interface.
@@ -1200,6 +1205,9 @@ def use_speedups(self, value):
         pass
 
 
+register_database(PooledPostgresqlDatabase, 'postgres+pool+async', 'postgresql+pool+async')
+
+
 #########
 # MySQL #
 #########
@@ -1318,6 +1326,9 @@ def use_speedups(self, value):
         pass
 
 
+register_database(MySQLDatabase, 'mysql+async')
+
+
 class PooledMySQLDatabase(MySQLDatabase):
     """MySQL database driver providing **single drop-in sync**
     connection and **async connections pool** interface.
@@ -1339,6 +1350,9 @@ def init(self, database, **kwargs):
         self.max_connections = max_connections
 
 
+register_database(PooledMySQLDatabase, 'mysql+pool+async')
+
+
 ##############
 # Sync utils #
 ##############

peewee_asyncext.py

@@ -13,6 +13,8 @@
 Copyright (c) 2014, Alexey Kinëv <[email protected]>
 
 """
+from playhouse.db_url import register_database
+
 from peewee_async import AsyncPostgresqlMixin
 import playhouse.postgres_ext as ext
 
@@ -47,6 +49,9 @@ def use_speedups(self, value):
         pass
 
 
+register_database(PostgresqlExtDatabase, 'postgresext+async', 'postgresqlext+async')
+
+
 class PooledPostgresqlExtDatabase(AsyncPostgresqlMixin, ext.PostgresqlExtDatabase):
     """PosgreSQL database extended driver providing **single drop-in sync**
     connection and **async connections pool** interface.
@@ -78,3 +83,6 @@ def use_speedups(self):
     @use_speedups.setter
     def use_speedups(self, value):
         pass
+
+
+register_database(PooledPostgresqlExtDatabase, 'postgresext+pool+async', 'postgresqlext+pool+async')