Django command to wait for database availability

wait_for_db.py

import time

from django.db import connections
from django.db.utils import OperationalError
from django.core.management.base import BaseCommand
from django.db import DEFAULT_DB_ALIAS


class Command(BaseCommand):
    """Django command to pause execution until database is available"""

    def handle(self, *args, **options):
        """Handle the command"""
        self.stdout.write('Waiting for database...')
        while True:
            try:
                db_conn = connections[DEFAULT_DB_ALIAS]
                db_conn.ensure_connection()
                self.stdout.write(self.style.SUCCESS('Database available!'))
                break  # Sortie de la boucle si la connexion est réussie
            except OperationalError:
                self.stdout.write('Database unavailable, waiting 1 second...')
                time.sleep(1)