Explorar o código

Fix pep8 error on psycopg2cffi hack

Richard van der Hoff %!s(int64=6) %!d(string=hai) anos
pai
achega
d1e56cfcd1
Modificáronse 1 ficheiros con 5 adicións e 6 borrados
  1. 5 6
      synapse/storage/engines/__init__.py

+ 5 - 6
synapse/storage/engines/__init__.py

@@ -32,12 +32,11 @@ def create_engine(database_config):
     engine_class = SUPPORTED_MODULE.get(name, None)
 
     if engine_class:
-        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)
+        # pypy requires psycopg2cffi rather than psycopg2
+        if (name == "psycopg2" and
+                platform.python_implementation() == "PyPy"):
+            name = "psycopg2cffi"
+        module = importlib.import_module(name)
         return engine_class(module, database_config)
 
     raise RuntimeError(