Tried to change DB_URL again.

NYUCCL · Dec 16, 2022 · 9c46a0b · 9c46a0b
1 parent ed1b547
commit 9c46a0b
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/psiturk/db.py b/psiturk/db.py
@@ -20,6 +20,10 @@
     # internally use `mysql+pymysql://` so sqlalchemy talks to
     # the pymysql package
     DATABASE = DATABASE.replace('mysql://', 'mysql+pymysql://')
+if 'postgres://' in DATABASE.lower():
+    # internally use `mysql+pymysql://` so sqlalchemy talks to
+    # the pymysql package
+    DATABASE = DATABASE.replace('postgres://', 'postgresql://')
 
 engine = create_engine(DATABASE, echo=False, pool_recycle=3600)
 db_session = scoped_session(sessionmaker(autocommit=False,