Add custom fields to the built-in Group model

Author:: jmoppel
Posted:: December 19, 2022
Language:: Python
Version:: 3.2
Score:: 1 (after 1 ratings)

Download
Raw

Add fields and extend Django's built-in Group model using a OneToOneField (i.e. a profile model). In this example, we add a UUIDField. Whenever a new group is created, we automatically (via signals) create a corresponding Role record referencing the newly created group. Whenever a Group is deleted, the corresponding Role is deleted as well.

from uuid import uuid4
from django.db import models
from django.db.models.signals import post_save
from django.dispatch import receiver

class Role(models.Model):
    """Extension of Django's built-in Group model

    Adds a UUID field to Django's built-in Group model.
    """

    uuid = models.UUIDField(
        auto_created=True,
        default=uuid4,
        editable=False,
        help_text="object ID",
        primary_key=True,
        unique=True,
    )
    group = models.OneToOneField(Group, on_delete=models.CASCADE)

    def __str__(self):
        return f"{self.group.name}"


@receiver(post_save, sender=Group)
def create_role(sender, instance, created, **kwargs):
    if created:
        Role.objects.create(group=instance)

Comments

Please login first before commenting.

Add custom fields to the built-in Group model

More like this

Comments