"""Wartet, bis die DB Connections annimmt."""
import time

from django.core.management.base import BaseCommand
from django.db import OperationalError, connections


class Command(BaseCommand):
    help = "Wartet, bis die Standard-DB verfügbar ist."

    def add_arguments(self, parser):
        parser.add_argument("--timeout", type=int, default=60)

    def handle(self, *args, **opts):
        deadline = time.time() + opts["timeout"]
        while time.time() < deadline:
            try:
                connections["default"].ensure_connection()
                self.stdout.write(self.style.SUCCESS("DB ist bereit."))
                return
            except OperationalError:
                self.stdout.write("warte auf DB...")
                time.sleep(2)
        self.stderr.write(self.style.ERROR("DB nach Timeout nicht erreichbar."))
        raise SystemExit(1)