mirror of
https://github.com/etesync/server
synced 2024-11-24 09:48:09 +00:00
44 lines
1.2 KiB
Python
44 lines
1.2 KiB
Python
import typing as t
|
|
|
|
from django.contrib.auth.models import AbstractUser, UserManager as DjangoUserManager
|
|
from django.contrib.auth.validators import UnicodeUsernameValidator
|
|
from django.db import models
|
|
from django.utils.translation import gettext_lazy as _
|
|
|
|
|
|
class UserManager(DjangoUserManager):
|
|
def get_by_natural_key(self, username: t.Optional[str]):
|
|
return self.get(**{self.model.USERNAME_FIELD + "__iexact": username})
|
|
|
|
|
|
class User(AbstractUser):
|
|
id: int
|
|
username_validator = UnicodeUsernameValidator()
|
|
|
|
objects: UserManager = UserManager()
|
|
|
|
username = models.CharField(
|
|
_("username"),
|
|
max_length=150,
|
|
unique=True,
|
|
help_text=_("Required. 150 characters or fewer. Letters, digits and ./-/_ only."),
|
|
validators=[username_validator],
|
|
error_messages={
|
|
"unique": _("A user with that username already exists."),
|
|
},
|
|
)
|
|
|
|
@classmethod
|
|
def normalize_username(cls, username: str):
|
|
return super().normalize_username(username).lower()
|
|
|
|
|
|
UserType = User
|
|
|
|
|
|
def get_typed_user_model() -> UserType:
|
|
from django.contrib.auth import get_user_model
|
|
|
|
ret: t.Any = get_user_model()
|
|
return ret
|