Handled more errors during db initialization

This commit is contained in:
Marcin-Ramotowski 2025-06-12 18:42:07 +00:00
parent 301cf5922e
commit cd4ab3fd27

View File

@ -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.")