Automatically register users
This commit is contained in:
26
lg/models.py
26
lg/models.py
@ -1,12 +1,34 @@
|
||||
from django.contrib.auth.models import User
|
||||
from django.contrib.auth.models import AbstractUser
|
||||
from django.db import models
|
||||
from django.utils.translation import gettext_lazy as _
|
||||
from polymorphic.models import PolymorphicModel
|
||||
|
||||
|
||||
class DiscordUser(AbstractUser):
|
||||
discord_id = models.CharField(
|
||||
max_length=32,
|
||||
primary_key=True,
|
||||
verbose_name=_("discord id"),
|
||||
)
|
||||
|
||||
discriminator = models.CharField(
|
||||
max_length=4,
|
||||
verbose_name=_("discriminator"),
|
||||
)
|
||||
|
||||
avatar_id = models.CharField(
|
||||
max_length=32,
|
||||
verbose_name=_("avatar id"),
|
||||
)
|
||||
|
||||
@property
|
||||
def avatar_url(self):
|
||||
return f"https://cdn.discordapp.com/avatars/{self.discord_id}/{self.avatar_id}.png"
|
||||
|
||||
|
||||
class Player(PolymorphicModel):
|
||||
user = models.OneToOneField(
|
||||
User,
|
||||
DiscordUser,
|
||||
on_delete=models.CASCADE,
|
||||
verbose_name=_("user"),
|
||||
)
|
||||
|
Reference in New Issue
Block a user