charasheet/src/party/models.py

from django.db import models
from django.db.models import F, Q
from django.urls import reverse
from django_extensions.db.models import TimeStampedModel

from character.models import Character
from common.models import UniquelyNamedModel, UniquelyNamedModelManager


class PartyQuerySet(models.QuerySet):
    def managed_by(self, user):
        return self.filter(game_master=user)

    def played_by(self, user):
        return self.filter(characters__in=Character.objects.filter(player=user))

    def played_or_mastered_by(self, user):
        return self.filter(
            Q(game_master=user)
            | Q(characters__in=Character.objects.filter(player=user)),
        ).distinct()

    def related_to(self, user):
        return self.filter(
            Q(game_master=user)
            | Q(characters__in=Character.objects.filter(player=user))
            | Q(invited_characters__in=Character.objects.filter(player=user)),
        ).distinct()

    def invited_to(self, user):
        return self.filter(invited_characters__in=Character.objects.filter(player=user))


class PartyManager(UniquelyNamedModelManager):
    pass


class Party(UniquelyNamedModel, TimeStampedModel, models.Model):  # noqa: DJ008
    game_master = models.ForeignKey(
        "common.User",
        on_delete=models.PROTECT,
        related_name="parties",
        verbose_name="meneur de jeu",
    )
    invited_characters = models.ManyToManyField(
        "character.Character",
        blank=True,
        related_name="invites",
        verbose_name="personnages invités",
    )
    characters = models.ManyToManyField(
        "character.Character",
        blank=True,
        related_name="parties",
        verbose_name="personnages",
    )

    objects = PartyManager.from_queryset(PartyQuerySet)()

    class Meta(UniquelyNamedModel.Meta, TimeStampedModel.Meta):
        verbose_name = "Groupe"
        verbose_name_plural = "Groupes"

    def get_absolute_url(self) -> str:
        return reverse("party:details", kwargs={"pk": self.pk})

    def reset_stats(self):
        for character in self.characters.all():
            character.reset_stats()


class BattleEffectQuerySet(models.QuerySet):
    def increase_rounds(self):
        self.temporary().update(remaining_rounds=F("remaining_rounds") + 1)

    def decrease_rounds(self):
        self.active().update(remaining_rounds=F("remaining_rounds") - 1)

    def active(self):
        return self.filter(remaining_rounds__gt=0)

    def terminated(self):
        return self.filter(remaining_rounds=0)

    def permanent(self):
        return self.filter(remaining_rounds=-1)

    def temporary(self):
        return self.exclude(remaining_rounds=-1)


class BattleEffectManager(models.Manager):
    pass


class BattleEffect(TimeStampedModel, models.Model):
    name = models.CharField(max_length=100, blank=False, null=False, verbose_name="nom")
    target = models.CharField(
        max_length=100,
        blank=False,
        null=False,
        verbose_name="cible",
    )
    description = models.TextField(blank=True, null=False, verbose_name="description")
    remaining_rounds = models.SmallIntegerField(
        blank=False,
        default=-1,
        verbose_name="nombre de tours restants",
        help_text="-1 pour un effet permanent",
    )
    party = models.ForeignKey(
        "party.Party",
        on_delete=models.CASCADE,
        related_name="effects",
        verbose_name="groupe",
    )
    created_by = models.ForeignKey(
        "common.User",
        on_delete=models.CASCADE,
        related_name="effects",
        verbose_name="créé par",
    )

    objects = BattleEffectManager.from_queryset(BattleEffectQuerySet)()

    @property
    def remaining_percent(self) -> float:
        max_display_percent = 5
        if self.remaining_rounds >= max_display_percent or self.remaining_rounds < 0:
            return 100
        return self.remaining_rounds / max_display_percent * 100

    def __str__(self):
        return self.name