Misago/misago/threads/models/threadparticipant.py

from django.db import models

from misago.conf import settings


class ThreadParticipantManager(models.Manager):
    def set_owner(self, thread, user):
        ThreadParticipant.objects.filter(
            thread=thread,
            is_owner=True
        ).update(is_owner=False)

        self.remove_participant(thread, user)

        ThreadParticipant.objects.create(
            thread=thread,
            user=user,
            is_owner=True
        )

    def add_participants(self, thread, users):
        bulk = []
        for user in users:
            bulk.append(ThreadParticipant(
                thread=thread,
                user=user,
                is_owner=False
            ))

        ThreadParticipant.objects.bulk_create(bulk)

    def remove_participant(self, thread, user):
        ThreadParticipant.objects.filter(
            thread=thread,
            user=user
        ).delete()


class ThreadParticipant(models.Model):
    thread = models.ForeignKey('misago_threads.Thread')
    user = models.ForeignKey(settings.AUTH_USER_MODEL)
    is_owner = models.BooleanField(default=False)

    objects = ThreadParticipantManager()