From 5b5dfdad3edb77469ed3ff9f1cc8f11e5b8a0c19 Mon Sep 17 00:00:00 2001 From: Han Date: Wed, 12 Feb 2025 08:04:02 +0700 Subject: [PATCH] "Add user management feature". --- users/models.py | 12 ++++++++++++ 1 file changed, 12 insertions(+) create mode 100644 users/models.py diff --git a/users/models.py b/users/models.py new file mode 100644 index 0000000..405cb41 --- /dev/null +++ b/users/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}"