load_db_state_to_k8s.py

import logging
from django.core.management.base import BaseCommand
from django.shortcuts import get_object_or_404

from api.models.key import Key
from api.models.app import App
from api.models.domain import Domain
from api.models.certificate import Certificate
from api.models.service import Service
from api.models.volume import Volume
from api.models.gateway import Route, Gateway
from api.exceptions import DryccException, AlreadyExists


logger = logging.getLogger(__name__)


class Command(BaseCommand):
    """Management command for publishing Drycc platform state from the database
    to k8s.
    """

    def _deploy(self, app):
        try:
            app.state_to_k8s()
        except AlreadyExists as error:
            logger.debug(error)
            print('WARNING: {} has a deployment in progress. '
                  'Skipping deployment...'.format(app))
        except DryccException as error:
            logger.exception(error)
            print('ERROR: There was a problem deploying {} '
                  'due to {}'.format(app, str(error)))

    def handle(self, *args, **options):
        """Publishes Drycc platform state from the database to kubernetes."""
        print("Publishing DB state to kubernetes...")

        self.save_apps()

        # certificates have to be attached to domains to create k8s secrets
        for cert in Certificate.objects.all():
            for domain in cert.domains:
                domain = get_object_or_404(Domain, domain=domain)
                cert.attach_in_kubernetes(domain)

        # deploy apps
        print("Deploying available applications.")
        for app in App.objects.all():
            self._deploy(app)
        print("Done Publishing DB state to kubernetes.")

    def save_apps(self):
        """Saves important Django data models to the database."""
        for app in App.objects.all():
            try:
                app.save()
                app.config_set.latest().save()
                tls = app.tls_set.latest()
                tls.refresh_issuer_to_k8s()
                tls.refresh_certificate_to_k8s()
            except DryccException as error:
                print('ERROR: Problem saving to model {} for {}'
                      'due to {}'.format(str(App.__name__), str(app), str(error)))
        for model in (Volume, Route, Gateway, Key, Domain, Certificate, Service):
            for obj in model.objects.all():
                try:
                    obj.save()
                except DryccException as error:
                    print('ERROR: Problem saving to model {} for {}'
                          'due to {}'.format(str(model.__name__), str(obj), str(error)))