Postgres db_wait fix

This commit is contained in:
Andrew Morgan 2020-04-02 11:29:00 +01:00
parent 677d0edbac
commit db2652cc5e

View file

@ -14,6 +14,7 @@
# limitations under the License. # limitations under the License.
import logging import logging
import psycopg2.extras
from ._base import BaseDatabaseEngine, IncorrectDatabaseSetup from ._base import BaseDatabaseEngine, IncorrectDatabaseSetup
@ -24,6 +25,7 @@ class PostgresEngine(BaseDatabaseEngine):
def __init__(self, database_module, database_config): def __init__(self, database_module, database_config):
super().__init__(database_module, database_config) super().__init__(database_module, database_config)
self.module.extensions.register_type(self.module.extensions.UNICODE) self.module.extensions.register_type(self.module.extensions.UNICODE)
self.module.extensions.set_wait_callback(psycopg2.extras.wait_select)
# Disables passing `bytes` to txn.execute, c.f. #6186. If you do # Disables passing `bytes` to txn.execute, c.f. #6186. If you do
# actually want to use bytes than wrap it in `bytearray`. # actually want to use bytes than wrap it in `bytearray`.