from django import forms

from apps.rims.contacts.models import Contact, ContactGroup


class ContactForm(forms.ModelForm):
    class Meta:
        model = Contact
        fields = [
            "contact_type",
            "first_name",
            "last_name",
            "organisation_name",
            "job_title",
            "email",
            "phone",
            "country",
            "city",
            "address",
            "website",
            "groups",
            "notes",
        ]
        widgets = {
            "address": forms.Textarea(attrs={"rows": 3}),
            "notes": forms.Textarea(attrs={"rows": 3}),
            "groups": forms.CheckboxSelectMultiple(),
        }


class ContactGroupForm(forms.ModelForm):
    class Meta:
        model = ContactGroup
        fields = ["name", "description"]