Dmitry Belan
18 lines
572 B
Python
18 lines
572 B
Python
from django.contrib.auth.models import AbstractUser
|
|
from django.db import models
|
|
|
|
class CustomUser(AbstractUser):
|
|
ROLE_CHOICES = (
|
|
('ADMIN', 'Администратор'),
|
|
('TEACHER', 'Учитель'),
|
|
('STUDENT', 'Ученик'),
|
|
)
|
|
role = models.CharField('Роль', max_length=10, choices=ROLE_CHOICES, default='STUDENT')
|
|
|
|
class Meta:
|
|
verbose_name = 'Пользователь'
|
|
verbose_name_plural = 'Пользователи'
|
|
|
|
def __str__(self):
|
|
return f"{self.username} ({self.get_role_display()})"
|