From 144ab73dc8683781e33a596f6a52415fe7c801e3 Mon Sep 17 00:00:00 2001 From: Chneemann Date: Sun, 30 Mar 2025 19:31:29 +0200 Subject: [PATCH] feat: added generate_random_color function for color field in model, default value set --- user_app/models.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/user_app/models.py b/user_app/models.py index 740f44b..44ca531 100644 --- a/user_app/models.py +++ b/user_app/models.py @@ -1,10 +1,13 @@ -import uuid +import uuid, random from django.contrib.auth.models import AbstractBaseUser, BaseUserManager, PermissionsMixin, Group, Permission from django.db import models def generate_uuid_without_dashes(): return uuid.uuid4().hex +def generate_random_color(): + return "#" + "".join(random.choices("0123456789ABCDEF", k=6)) + class UserManager(BaseUserManager): def create_user(self, email, first_name, password=None): if not email: @@ -34,7 +37,7 @@ class User(AbstractBaseUser, PermissionsMixin): email = models.EmailField(unique=True) phone = models.CharField(max_length=20, blank=True, null=True) initials = models.CharField(max_length=10, blank=True) - color = models.CharField(max_length=20, blank=True) + color = models.CharField(max_length=20, blank=True, default=generate_random_color) is_online = models.BooleanField(default=False) is_contact_only = models.BooleanField(default=False) last_login = models.DateTimeField(blank=True, null=True)