diff --git a/models.py b/models.py new file mode 100644 index 0000000..405cb41 --- /dev/null +++ b/models.py @@ -0,0 +1,12 @@ +from django.contrib.auth.models import AbstractUser +from django.db import models + +class User(AbstractUser): + ROLE_CHOICES = [ + ('admin', 'Admin'), + ('customer', 'Customer'), + ] + role = models.CharField(max_length=10, choices=ROLE_CHOICES, default='customer') + + def __str__(self): + return f"{self.username} - {self.role}"