from django import forms
from .models import ContactMessage, Customer, Product, PriceRecord

class ContactForm(forms.ModelForm):
    class Meta:
        model = ContactMessage
        fields = ['name', 'phone', 'email', 'subject', 'message']
        widgets = {
            'name': forms.TextInput(attrs={'class': 'form-control', 'placeholder': '您的姓名'}),
            'phone': forms.TextInput(attrs={'class': 'form-control', 'placeholder': '联系电话'}),
            'email': forms.EmailInput(attrs={'class': 'form-control', 'placeholder': '电子邮箱'}),
            'subject': forms.TextInput(attrs={'class': 'form-control', 'placeholder': '留言主题'}),
            'message': forms.Textarea(attrs={'class': 'form-control', 'rows': 5, 'placeholder': '请输入您的留言内容'}),
        }

class CustomerForm(forms.ModelForm):
    class Meta:
        model = Customer
        fields = ['name', 'contact_person', 'phone', 'email', 'address', 'company_info']
        widgets = {
            'name': forms.TextInput(attrs={'class': 'form-control'}),
            'contact_person': forms.TextInput(attrs={'class': 'form-control'}),
            'phone': forms.TextInput(attrs={'class': 'form-control'}),
            'email': forms.EmailInput(attrs={'class': 'form-control'}),
            'address': forms.Textarea(attrs={'class': 'form-control', 'rows': 2}),
            'company_info': forms.Textarea(attrs={'class': 'form-control', 'rows': 3}),
        }

class PriceRecordForm(forms.ModelForm):
    class Meta:
        model = PriceRecord
        fields = ['product', 'platform', 'price', 'url', 'supplier', 'remark']
        widgets = {
            'product': forms.Select(attrs={'class': 'form-control'}),
            'platform': forms.Select(attrs={'class': 'form-control'}),
            'price': forms.NumberInput(attrs={'class': 'form-control', 'step': '0.01'}),
            'url': forms.URLInput(attrs={'class': 'form-control'}),
            'supplier': forms.TextInput(attrs={'class': 'form-control'}),
            'remark': forms.Textarea(attrs={'class': 'form-control', 'rows': 2}),
        }