mirror of
https://github.com/zulip/zulip.git
synced 2025-10-24 16:43:57 +00:00
365fe0b3d5
Fixes #2665. Regenerated by tabbott with `lint --fix` after a rebase and change in parameters. Note from tabbott: In a few cases, this converts technical debt in the form of unsorted imports into different technical debt in the form of our largest files having very long, ugly import sequences at the start. I expect this change will increase pressure for us to split those files, which isn't a bad thing. Signed-off-by: Anders Kaseorg <anders@zulip.com>
84 lines
3.3 KiB
Python
84 lines
3.3 KiB
Python
import datetime
|
|
|
|
from django.db import models
|
|
|
|
from analytics.models import BaseCount
|
|
from zerver.models import AbstractPushDeviceToken, AbstractRealmAuditLog
|
|
|
|
|
|
def get_remote_server_by_uuid(uuid: str) -> 'RemoteZulipServer':
|
|
return RemoteZulipServer.objects.get(uuid=uuid)
|
|
|
|
class RemoteZulipServer(models.Model):
|
|
UUID_LENGTH = 36
|
|
API_KEY_LENGTH = 64
|
|
HOSTNAME_MAX_LENGTH = 128
|
|
|
|
uuid: str = models.CharField(max_length=UUID_LENGTH, unique=True)
|
|
api_key: str = models.CharField(max_length=API_KEY_LENGTH)
|
|
|
|
hostname: str = models.CharField(max_length=HOSTNAME_MAX_LENGTH)
|
|
contact_email: str = models.EmailField(blank=True, null=False)
|
|
last_updated: datetime.datetime = models.DateTimeField('last updated', auto_now=True)
|
|
|
|
def __str__(self) -> str:
|
|
return f"<RemoteZulipServer {self.hostname} {self.uuid[0:12]}>"
|
|
|
|
def format_requestor_for_logs(self) -> str:
|
|
return "zulip-server:" + self.uuid
|
|
|
|
# Variant of PushDeviceToken for a remote server.
|
|
class RemotePushDeviceToken(AbstractPushDeviceToken):
|
|
server: RemoteZulipServer = models.ForeignKey(RemoteZulipServer, on_delete=models.CASCADE)
|
|
# The user id on the remote server for this device device this is
|
|
user_id: int = models.BigIntegerField(db_index=True)
|
|
|
|
class Meta:
|
|
unique_together = ("server", "user_id", "kind", "token")
|
|
|
|
def __str__(self) -> str:
|
|
return f"<RemotePushDeviceToken {self.server} {self.user_id}>"
|
|
|
|
class RemoteRealmAuditLog(AbstractRealmAuditLog):
|
|
"""Synced audit data from a remote Zulip server, used primarily for
|
|
billing. See RealmAuditLog and AbstractRealmAuditLog for details.
|
|
"""
|
|
server: RemoteZulipServer = models.ForeignKey(RemoteZulipServer, on_delete=models.CASCADE)
|
|
realm_id: int = models.IntegerField(db_index=True)
|
|
# The remote_id field lets us deduplicate data from the remote server
|
|
remote_id: int = models.IntegerField(db_index=True)
|
|
|
|
def __str__(self) -> str:
|
|
return f"<RemoteRealmAuditLog: {self.server} {self.event_type} {self.event_time} {self.id}>"
|
|
|
|
class RemoteInstallationCount(BaseCount):
|
|
server: RemoteZulipServer = models.ForeignKey(RemoteZulipServer, on_delete=models.CASCADE)
|
|
# The remote_id field lets us deduplicate data from the remote server
|
|
remote_id: int = models.IntegerField(db_index=True)
|
|
|
|
class Meta:
|
|
unique_together = ("server", "property", "subgroup", "end_time")
|
|
index_together = [
|
|
["server", "remote_id"],
|
|
]
|
|
|
|
def __str__(self) -> str:
|
|
return f"<InstallationCount: {self.property} {self.subgroup} {self.value}>"
|
|
|
|
# We can't subclass RealmCount because we only have a realm_id here, not a foreign key.
|
|
class RemoteRealmCount(BaseCount):
|
|
server: RemoteZulipServer = models.ForeignKey(RemoteZulipServer, on_delete=models.CASCADE)
|
|
realm_id: int = models.IntegerField(db_index=True)
|
|
# The remote_id field lets us deduplicate data from the remote server
|
|
remote_id: int = models.IntegerField(db_index=True)
|
|
|
|
class Meta:
|
|
unique_together = ("server", "realm_id", "property", "subgroup", "end_time")
|
|
index_together = [
|
|
["property", "end_time"],
|
|
["server", "remote_id"],
|
|
]
|
|
|
|
def __str__(self) -> str:
|
|
return f"{self.server} {self.realm_id} {self.property} {self.subgroup} {self.value}"
|