aboutsummaryrefslogtreecommitdiffstats
path: root/pydis_site/apps/api/models/bot
diff options
context:
space:
mode:
authorGravatar Akarys42 <[email protected]>2019-11-20 18:21:39 +0100
committerGravatar Akarys42 <[email protected]>2019-11-20 18:21:39 +0100
commitb23edc0704618db7f17bd3438a1931c1e37cb16d (patch)
treee3d8d59107aead98422180bdef75fad8069b0ffb /pydis_site/apps/api/models/bot
parentTest for the attachment image to be in the staff log (diff)
parentUpdate migration history (diff)
Merge branch 'bot#549-show-attachments-staff' of https://github.com/python-discord/site into bot#549-show-attachments-staff
Diffstat (limited to 'pydis_site/apps/api/models/bot')
-rw-r--r--pydis_site/apps/api/models/bot/infraction.py7
-rw-r--r--pydis_site/apps/api/models/bot/message.py1
-rw-r--r--pydis_site/apps/api/models/bot/user.py2
3 files changed, 9 insertions, 1 deletions
diff --git a/pydis_site/apps/api/models/bot/infraction.py b/pydis_site/apps/api/models/bot/infraction.py
index dfb32a97..108fd3a2 100644
--- a/pydis_site/apps/api/models/bot/infraction.py
+++ b/pydis_site/apps/api/models/bot/infraction.py
@@ -71,3 +71,10 @@ class Infraction(ModelReprMixin, models.Model):
"""Defines the meta options for the infraction model."""
ordering = ['-inserted_at']
+ constraints = (
+ models.UniqueConstraint(
+ fields=["user", "type"],
+ condition=models.Q(active=True),
+ name="unique_active_infraction_per_type_per_user"
+ ),
+ )
diff --git a/pydis_site/apps/api/models/bot/message.py b/pydis_site/apps/api/models/bot/message.py
index 72b0b61a..8b18fc9f 100644
--- a/pydis_site/apps/api/models/bot/message.py
+++ b/pydis_site/apps/api/models/bot/message.py
@@ -55,6 +55,7 @@ class Message(ModelReprMixin, models.Model):
models.URLField(
max_length=512
),
+ blank=True,
help_text="Attachments attached to this message."
)
diff --git a/pydis_site/apps/api/models/bot/user.py b/pydis_site/apps/api/models/bot/user.py
index 21617dc4..5140d2bf 100644
--- a/pydis_site/apps/api/models/bot/user.py
+++ b/pydis_site/apps/api/models/bot/user.py
@@ -50,7 +50,7 @@ class User(ModelReprMixin, models.Model):
def __str__(self):
"""Returns the name and discriminator for the current user, for display purposes."""
- return f"{self.name}#{self.discriminator}"
+ return f"{self.name}#{self.discriminator:0>4}"
@property
def top_role(self) -> Role: