From 704fb3b656a82519142a5f22488db4f87ae7e1d8 Mon Sep 17 00:00:00 2001 From: beyfendipenguen Date: Mon, 1 May 2023 02:27:53 +0300 Subject: [PATCH] fixed user save method for update and usage of uuid --- custom-user-model-ex1/user/models.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/custom-user-model-ex1/user/models.py b/custom-user-model-ex1/user/models.py index fc9bd77..efecc4d 100644 --- a/custom-user-model-ex1/user/models.py +++ b/custom-user-model-ex1/user/models.py @@ -15,9 +15,10 @@ class Role(models.TextChoices): role = models.CharField(max_length=50, choices=Role.choices) def save(self, *args, **kwargs): - if not self.pk: + if not self.role: self.role = self.base_role return super().save(*args, **kwargs) + return super().save(*args, **kwargs) class StudentManager(BaseUserManager):