Updating registration functionality

TAS-scorchedshadow · Sep 17, 2023 · 5f16708 · 5f16708
1 parent 6bac411
commit 5f16708
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/config.py b/config.py
@@ -14,7 +14,9 @@ class Config(object):
     FORCE_HTTPS = True
     SQLALCHEMY_TRACK_MODIFICATIONS = False
     LOGIN_DISABLED = False  # Enable this to avoid having to login
-    SQLALCHEMY_DATABASE_URI = os.environ.get('DATABASE_URL')
+    SQLALCHEMY_DATABASE_URI = os.getenv("DATABASE_URL")  # or other relevant config var
+    if SQLALCHEMY_DATABASE_URI.startswith("postgres://"):
+        SQLALCHEMY_DATABASE_URI = SQLALCHEMY_DATABASE_URI.replace("postgres://", "postgresql://", 1)
     SESSION_COOKIE_SECURE = True
 
     MAIL_SERVER = os.environ.get('MAIL_SERVER')