from django.db import models from django.db.models import Manager from typing import Dict, Optional, Text import zerver.models import datetime def get_remote_server_by_uuid(uuid): # type: (Text) -> RemoteZulipServer return RemoteZulipServer.objects.get(uuid=uuid) class RemoteZulipServer(models.Model): uuid = models.CharField(max_length=36, unique=True) # type: Text api_key = models.CharField(max_length=64) # type: Text hostname = models.CharField(max_length=128, unique=True) # type: Text contact_email = models.EmailField(blank=True, null=False) # type: Text last_updated = models.DateTimeField('last updated', auto_now=True) # type: datetime.datetime # Variant of PushDeviceToken for a remote server. class RemotePushDeviceToken(zerver.models.AbstractPushDeviceToken): server = models.ForeignKey(RemoteZulipServer) # type: RemoteZulipServer # The user id on the remote server for this device device this is user_id = models.BigIntegerField() # type: int class Deployment(models.Model): realms = models.ManyToManyField(zerver.models.Realm, related_name="_deployments") # type: Manager is_active = models.BooleanField(default=True) # type: bool # 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) # type: Optional[Text] base_api_url = models.CharField(max_length=128) # type: Text base_site_url = models.CharField(max_length=128) # type: Text @property def endpoints(self): # type: () -> Dict[str, Text] return {'base_api_url': self.base_api_url, 'base_site_url': self.base_site_url} @property def name(self): # type: () -> Text # 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