diff options
| author | Vincent Breitmoser <look@my.amazin.horse> | 2018-01-06 18:13:56 +0100 |
|---|---|---|
| committer | Vincent Breitmoser <look@my.amazin.horse> | 2018-04-10 11:29:52 +0200 |
| commit | 89de9349815b79f55bfff53b7c8c6d43f8c30336 (patch) | |
| tree | 0397dbdbc4f9c3f6258c8cd7dd2faddf8510ff75 /synapse/storage/engines | |
| parent | Use sortedcontainers instead of blist (diff) | |
| download | synapse-89de9349815b79f55bfff53b7c8c6d43f8c30336.tar.xz | |
Use psycopg2cffi module instead of psycopg2 if running on pypy
The psycopg2 package isn't available for PyPy. This commit adds a check if the runtime is PyPy, and if it is uses psycopg2cffi module in favor of psycopg2. This is almost a drop-in replacement, except for one place where an additional cast to string is required.
Diffstat (limited to 'synapse/storage/engines')
| -rw-r--r-- | synapse/storage/engines/__init__.py | 8 |
1 files changed, 7 insertions, 1 deletions
diff --git a/synapse/storage/engines/__init__.py b/synapse/storage/engines/__init__.py index 338b495611..be7f6d6ac3 100644 --- a/synapse/storage/engines/__init__.py +++ b/synapse/storage/engines/__init__.py @@ -18,6 +18,7 @@ from .postgres import PostgresEngine from .sqlite3 import Sqlite3Engine import importlib +import platform SUPPORTED_MODULE = { @@ -31,7 +32,12 @@ def create_engine(database_config): engine_class = SUPPORTED_MODULE.get(name, None) if engine_class: - module = importlib.import_module(name) + needs_pypy_hack = (name == "psycopg2" and + platform.python_implementation() == "PyPy") + if needs_pypy_hack: + module = importlib.import_module("psycopg2cffi") + else: + module = importlib.import_module(name) return engine_class(module, database_config) raise RuntimeError( |