mirror of
https://github.com/Crocmagnon/charasheet.git
synced 2024-05-19 01:12:51 +02:00
118 lines
3.5 KiB
Python
118 lines
3.5 KiB
Python
from django import forms
|
|
from django.core.exceptions import ValidationError
|
|
from django.db.models import Q
|
|
|
|
from character.models import Character, Path, RacialCapability
|
|
from character.models.pet import Pet
|
|
|
|
|
|
class EquipmentForm(forms.ModelForm):
|
|
class Meta:
|
|
model = Character
|
|
fields = ["equipment", "money_pp", "money_po", "money_pa", "money_pc"]
|
|
|
|
|
|
class AddPathForm(forms.Form):
|
|
character_path = forms.ModelChoiceField(
|
|
Path.objects.none(),
|
|
required=False,
|
|
empty_label="----- Voies liées au personnage -----",
|
|
)
|
|
other_path = forms.ModelChoiceField(
|
|
Path.objects.none(),
|
|
required=False,
|
|
empty_label="----- Autres voies -----",
|
|
)
|
|
|
|
def __init__(self, character: Character, *args, **kwargs) -> None:
|
|
super().__init__(*args, **kwargs)
|
|
paths = {cap.path_id for cap in character.capabilities.all()}
|
|
paths = (
|
|
Path.objects.exclude(pk__in=paths)
|
|
.order_by("profile__name", "race__name")
|
|
.select_related("profile", "race")
|
|
)
|
|
character_paths = paths.filter(
|
|
Q(profile=character.profile) | Q(race=character.race),
|
|
)
|
|
self.fields["character_path"].queryset = character_paths
|
|
self.fields["character_path"].widget.attrs["class"] = "form-select"
|
|
self.fields["other_path"].queryset = paths.exclude(
|
|
pk__in={path.pk for path in character_paths},
|
|
)
|
|
self.fields["other_path"].widget.attrs["class"] = "form-select"
|
|
|
|
def clean(self):
|
|
cleaned_data = super().clean()
|
|
values = [cleaned_data.get("character_path"), cleaned_data.get("other_path")]
|
|
if len(list(filter(None, values))) != 1:
|
|
msg = "Vous devez sélectionner une seule valeur."
|
|
raise ValidationError(msg)
|
|
return cleaned_data
|
|
|
|
|
|
class CharacterForm(forms.ModelForm):
|
|
def __init__(self, *args, **kwargs) -> None:
|
|
super().__init__(*args, **kwargs)
|
|
self.fields["racial_capability"].queryset = (
|
|
RacialCapability.objects.select_related("race")
|
|
)
|
|
self.fields["damage_reduction"].widget.attrs.update({"rows": 2})
|
|
self.fields["equipment"].widget.attrs.update({"rows": 3})
|
|
|
|
class Meta:
|
|
model = Character
|
|
fields = [
|
|
"name",
|
|
"race",
|
|
"profile",
|
|
"profile_picture",
|
|
"private",
|
|
"level",
|
|
"gender",
|
|
"age",
|
|
"height",
|
|
"weight",
|
|
"value_strength",
|
|
"value_dexterity",
|
|
"value_constitution",
|
|
"value_intelligence",
|
|
"value_wisdom",
|
|
"value_charisma",
|
|
"health_max",
|
|
"racial_capability",
|
|
"weapons",
|
|
"armor",
|
|
"shield",
|
|
"defense_misc",
|
|
"equipment",
|
|
"money_pp",
|
|
"money_po",
|
|
"money_pa",
|
|
"money_pc",
|
|
"damage_reduction",
|
|
"notes",
|
|
]
|
|
|
|
|
|
class PetForm(forms.ModelForm):
|
|
class Meta:
|
|
model = Pet
|
|
fields = [
|
|
"name",
|
|
"health_max",
|
|
"health_remaining",
|
|
"modifier_strength",
|
|
"modifier_dexterity",
|
|
"modifier_constitution",
|
|
"modifier_intelligence",
|
|
"modifier_wisdom",
|
|
"modifier_charisma",
|
|
"damage",
|
|
"initiative",
|
|
"defense",
|
|
"attack",
|
|
"recovery",
|
|
"notes",
|
|
]
|