zulip/zilencer/management/commands/create_deployment.py

from __future__ import absolute_import
from __future__ import print_function
from optparse import make_option
import sys

from django.core.management.base import BaseCommand

from zerver.models import get_realm
from zerver.lib.create_user import random_api_key
from zerver.management.commands.create_realm import Command as CreateRealm

from zilencer.models import Deployment

class Command(BaseCommand):
    help = """Create a deployment and accompanying realm."""

    option_list = CreateRealm.option_list + (
        make_option('--no-realm',
                    dest='no_realm',
                    action='store_true',
                    default=False,
                    help='Do not create a new realm; associate with an existing one.' + \
                           ' In this case, only the domain and URLs need to be specified.'),
        make_option('-a', '--api-url',
                    dest='api',
                    type='str'),
        make_option('-w', '--web-url',
                    dest='web',
                    type='str'),

        )

    def handle(self, *args, **options):
        if None in (options["api"], options["web"], options["domain"]):
            print("\033[1;31mYou must provide a domain, an API URL, and a web URL.\033[0m\n", file=sys.stderr)
            self.print_help("python manage.py", "create_realm")
            exit(1)

        if not options["no_realm"]:
            CreateRealm().handle(*args, **options)
            print() # Newline

        realm = get_realm(options["domain"])
        if realm is None:
            print("\033[1;31mRealm does not exist!\033[0m\n", file=sys.stderr)
            exit(2)

        dep = Deployment()
        dep.api_key = random_api_key()
        dep.save()
        old_dep = realm.deployment
        if old_dep is not None:
            old_dep.realms.remove(realm)
            old_dep.save()
        dep.realms = [realm]
        dep.base_api_url = options["api"]
        dep.base_site_url = options["web"]
        dep.save()
        print("Deployment %s created." % (dep.id,))
        print("DEPLOYMENT_ROLE_NAME = %s" % (dep.name,))
        print("DEPLOYMENT_ROLE_KEY = %s" % (dep.api_key,))