zulip/zilencer/models.py

38 lines
1.5 KiB
Python

import datetime
from typing import Text
from django.db import models
from zerver.models import AbstractPushDeviceToken, Realm
def get_remote_server_by_uuid(uuid: 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
def __str__(self) -> str:
return "<RemoteZulipServer %s %s>" % (self.hostname, self.uuid[0:12])
# Variant of PushDeviceToken for a remote server.
class RemotePushDeviceToken(AbstractPushDeviceToken):
server = models.ForeignKey(RemoteZulipServer, on_delete=models.CASCADE) # type: RemoteZulipServer
# The user id on the remote server for this device device this is
user_id = models.BigIntegerField(db_index=True) # type: int
token = models.CharField(max_length=4096, db_index=True) # type: bytes
class Meta:
unique_together = ("server", "token")
def __str__(self) -> str:
return "<RemotePushDeviceToken %s %s>" % (self.server, self.user_id)
class Customer(models.Model):
stripe_customer_id = models.CharField(max_length=255, unique=True)
realm = models.OneToOneField(Realm, on_delete=models.CASCADE)