charasheet/src/party/forms.py

24 lines
701 B
Python

from django import forms
from django.core.exceptions import ValidationError
from party.models import Party
class PartyForm(forms.ModelForm):
class Meta:
model = Party
fields = ["name", "invited_characters"]
def clean_invited_characters(self):
invited = self.cleaned_data["invited_characters"]
if not self.instance:
return invited
members = self.instance.characters.all()
for character in invited:
if character in members:
self.add_error(
"invited_characters",
ValidationError(f"{character} is already a group member."),
)
return invited