added group to custom user

rog/models.py

@@ -35,20 +35,20 @@ db = Db(dbname=env("POSTGRES_DBNAME"), user=env("POSTGRES_USER"), password=env("
 
 class CustomUserManager(BaseUserManager):
     
-    def create_user(self, email, password, **other_fields):
+    def create_user(self, email, password, group,  **other_fields):
 
         if not email:
             raise ValueError(_("You must provide an email address"))
 
         email = self.normalize_email(email)
-        user=self.model(email=email, **other_fields)
+        user=self.model(email=email, group=group, **other_fields)
         user.set_password(password)
         user.save()
 
         return user
 
 
-    def create_superuser(self, email, password, **other_fields):
+    def create_superuser(self, email, password, group, **other_fields):
         other_fields.setdefault('is_staff', True)
         other_fields.setdefault('is_superuser', True)
         other_fields.setdefault('is_active', True)
@@ -58,7 +58,7 @@ class CustomUserManager(BaseUserManager):
         if other_fields.get('is_superuser') is not True:
             raise ValueError(_('Supperuser must assigned to superuser=True'))
 
-        return self.create_user(email, password, **other_fields)
+        return self.create_user(email, password, group, **other_fields)
 
 
 
@@ -138,10 +138,19 @@ class GifuAreas(models.Model):
 
 
 class CustomUser(AbstractBaseUser, PermissionsMixin):
+    class Groups(models.TextChoices):
+        GB1 = 'G1', '大垣-初心者'
+        GB2 = 'G2', '大垣-３時間'
+        GB3 = 'G3', '大垣-5時間'
+
     email = models.EmailField(_("email address"), unique=True)
     is_staff = models.BooleanField(default=False)
     is_active = models.BooleanField(default=True)
+    group = models.CharField(max_length=2,
+                              choices=Groups.choices,
+                              default=Groups.GB1)
     USERNAME_FIELD = 'email'
+    REQUIRED_FIELDS = ['group',]
 
     objects = CustomUserManager()