25 lines
654 B
Python
25 lines
654 B
Python
from django.contrib.auth.models import AbstractUser
|
|
from django.db import models
|
|
|
|
from accounts.manager import UserManager
|
|
|
|
ROLE_CHOICES = (
|
|
('Admin', 'Admin'),
|
|
('Manager', 'Manager'),
|
|
('Staff', 'Staff'),
|
|
)
|
|
|
|
|
|
class User(AbstractUser):
|
|
phone_number = models.CharField(max_length=15, blank=True, null=True)
|
|
role = models.CharField(max_length=20, choices=ROLE_CHOICES, default='Staff')
|
|
|
|
is_staff = models.BooleanField(default=False)
|
|
is_customer = models.BooleanField(default=False)
|
|
is_user = models.BooleanField(default=False)
|
|
|
|
objects = UserManager()
|
|
|
|
def __str__(self):
|
|
return self.username
|