zulip/zilencer/models.py

from django.db import models
import zerver.models

def get_deployment_by_domain(domain):
    return Deployment.objects.get(realms__domain=domain)

class Deployment(models.Model):
    realms = models.ManyToManyField(zerver.models.Realm, related_name="_deployments")
    is_active = models.BooleanField(default=True)

    # TODO: This should really become the public portion of a keypair, and
    # it should be settable only with an initial bearer "activation key"
    api_key = models.CharField(max_length=32, null=True)

    base_api_url = models.CharField(max_length=128)
    base_site_url = models.CharField(max_length=128)

    @property
    def endpoints(self):
        return {'base_api_url': self.base_api_url, 'base_site_url': self.base_site_url}

    @property
    def name(self):
        # TODO: This only does the right thing for prod because prod authenticates to
        # staging with the zulip.com deployment key, while staging is technically the
        # deployment for the zulip.com realm.
        # This also doesn't necessarily handle other multi-realm deployments correctly.
        return self.realms.order_by('pk')[0].domain