zulip/zilencer/models.py

52 lines
2.1 KiB
Python

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