Added healthcheck

This commit is contained in:
Marcin-Ramotowski
2025-06-11 22:04:35 +00:00
parent 3f40a6126c
commit 479ec4f917
4 changed files with 40 additions and 0 deletions

View File

@ -7,6 +7,7 @@ from sqlalchemy.exc import DatabaseError
import time
from werkzeug.security import generate_password_hash
db_ready = False
def admin_required(user_id, message='Access denied.'):
"Check if common user try to make administrative action."
@ -33,10 +34,12 @@ def get_user_or_404(user_id):
def wait_for_db(max_retries):
"Try to connect with database <max_retries> times."
global db_ready
for _ in range(max_retries):
try:
with db.engine.connect() as connection:
connection.execute(text("SELECT 1"))
db_ready = True
return
except DatabaseError:
time.sleep(3)