HHHMHA/recipe-app-api

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


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

    def handle(self, *args, **kwargs):
        self.stdout.write('[INFO] waiting for database to start...')
        db_conn = None
        while not db_conn:
            try:
                db_conn = connections['default']
            except OperationalError:
                self.stdout.write(self.style.ERROR('[ERROR] database '
                                                   'unavailable waiting for 1'
                                                   ' second...'))
                time.sleep(1)

        self.stdout.write(self.style.SUCCESS('[INFO] connected to database '
                                             'successfully!'))