from django import forms
from django.contrib.auth.forms import UserCreationForm
from django.contrib.auth.models import User

from .models import ClientSite


class SignUpForm(UserCreationForm):
    email = forms.EmailField(required=True)

    class Meta:
        model = User
        fields = ("username", "email", "password1", "password2")


class ClientSiteForm(forms.ModelForm):
    class Meta:
        model = ClientSite
        fields = [
            "business_name",
            "business_address",
            "business_phone",
            "business_description",
            "old_website",
            "template_option",
        ]
        widgets = {
            "business_description": forms.Textarea(attrs={"rows": 4}),
        }