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 from models import User, db
import os import os
from sqlalchemy import text from sqlalchemy import text
from sqlalchemy.exc import DatabaseError from sqlalchemy.exc import DatabaseError, InterfaceError
import time import time
from werkzeug.security import generate_password_hash from werkzeug.security import generate_password_hash
@ -41,7 +41,7 @@ def wait_for_db(max_retries):
connection.execute(text("SELECT 1")) connection.execute(text("SELECT 1"))
db_ready = True db_ready = True
return return
except DatabaseError: except DatabaseError | InterfaceError:
time.sleep(3) time.sleep(3)
raise Exception("Failed to connect to database.") raise Exception("Failed to connect to database.")