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()})"