File: with_unique_constraint.py

package info (click to toggle)
python-django 3%3A4.2.28-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 58,896 kB
  • sloc: python: 334,975; javascript: 18,754; xml: 215; makefile: 178; sh: 27
file content (22 lines) | stat: -rw-r--r-- 686 bytes parent folder | download | duplicates (6)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
from django.contrib.auth.models import AbstractBaseUser, BaseUserManager
from django.db import models


class CustomUserWithUniqueConstraintManager(BaseUserManager):
    def create_superuser(self, username, password):
        user = self.model(username=username)
        user.set_password(password)
        user.save(using=self._db)
        return user


class CustomUserWithUniqueConstraint(AbstractBaseUser):
    username = models.CharField(max_length=150)

    objects = CustomUserWithUniqueConstraintManager()
    USERNAME_FIELD = "username"

    class Meta:
        constraints = [
            models.UniqueConstraint(fields=["username"], name="unique_custom_username"),
        ]