test_nov/app1/management/commands/wait_for_db.py

"""
Django command to wait until DB is started

"""

import time

from psycopg2 import OperationalError as Psycopg2OpError
from django.db.utils import OperationalError

from django.core.management.base import BaseCommand

class Command(BaseCommand):
    """ Django wait for DB"""

    def handle(self, *args, **options):
        self.stdout.write('Waiting for Database ...')
        db_up = False
        while db_up is False:
            try:
                self.check(databases=['default'])
                db_up = True
            except (Psycopg2OpError, OperationalError):
                self.stdout.write('Database unavailable, waiting 1 sec...')
                time.sleep(1)
    
        self.stdout.write(self.style.SUCCESS('database ready !!'))