From cd4ab3fd27b3ce91575e487c450a95737ea99351 Mon Sep 17 00:00:00 2001 From: Marcin-Ramotowski Date: Thu, 12 Jun 2025 18:42:07 +0000 Subject: [PATCH] Handled more errors during db initialization --- api/utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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.")