diff --git a/api/utils.py b/api/utils.py index 3d51eb3..2f5d69d 100644 --- a/api/utils.py +++ b/api/utils.py @@ -3,7 +3,7 @@ from flask_jwt_extended import get_jwt_identity from models import User, db import os from sqlalchemy import text -from sqlalchemy.exc import DatabaseError +from sqlalchemy.exc import DatabaseError, InterfaceError import time from werkzeug.security import generate_password_hash @@ -41,7 +41,7 @@ def wait_for_db(max_retries): connection.execute(text("SELECT 1")) db_ready = True return - except DatabaseError: + except DatabaseError | InterfaceError: time.sleep(3) raise Exception("Failed to connect to database.")