from django import forms
from .models import *


class ProjectResourcesForm(forms.ModelForm):
    class Meta:
        model = ProjectResources
        fields = '__all__'
        widgets = {
            'name': forms.TextInput(attrs={'class': 'form-control', 'placeholder': '请输入姓名'}),
            'unit': forms.TextInput(attrs={'class': 'form-control', 'placeholder': '请输入单位'}),
            'position': forms.TextInput(attrs={'class': 'form-control', 'placeholder': '请输入职务'}),
            'contact_phone': forms.TextInput(attrs={'class': 'form-control', 'placeholder': '请输入联系电话'}),
            'contact_address': forms.TextInput(attrs={'class': 'form-control', 'placeholder': '请输入联系地址'}),
            'resource_manager': forms.Select(attrs={'class': 'form-control'}),
        }

    def __init__(self, *args, **kwargs):
        super(ProjectResourcesForm, self).__init__(*args, **kwargs)

        # 添加选择字段的选项
        self.fields['is_enter'] = forms.ChoiceField(
            choices=[('', '---------')] + [('是', '是'), ('否', '否')],
            widget=forms.Select(attrs={'class': 'form-control'}),
            label="是否录入项目信息"
        )

        for field_name in self.fields:
            self.fields[field_name].required = False


class CustomerForm(forms.ModelForm):
    class Meta:
        model = Customer
        fields = '__all__'
        widgets = {
            'relationship_type': forms.Select(attrs={'class': 'form-control'}),
            'detailed_description': forms.TextInput(attrs={'class': 'form-control', 'placeholder': '请输入详细说明'}),
            'name': forms.TextInput(attrs={'class': 'form-control', 'placeholder': '请输入姓名'}),
            'unit': forms.TextInput(attrs={'class': 'form-control', 'placeholder': '请输入单位'}),
            'position': forms.TextInput(attrs={'class': 'form-control', 'placeholder': '请输入职务'}),
            'residence': forms.TextInput(attrs={'class': 'form-control', 'placeholder': '请输入常住地'}),
            'cooperation': forms.Select(attrs={'class': 'form-control'}),
            'project': forms.TextInput(attrs={'class': 'form-control'}),
            'main_organizer': forms.TextInput(attrs={'class': 'form-control'}),
            'assistant': forms.TextInput(attrs={'class': 'form-control'}),
        }

    def __init__(self, *args, **kwargs):
        super(CustomerForm, self).__init__(*args, **kwargs)

        # 添加选择字段的选项
        self.fields['cooperation'] = forms.ChoiceField(
            choices=[('', '---------')] + [('是', '是'), ('否', '否')],
            widget=forms.Select(attrs={'class': 'form-control'}),
            label="是否录入项目信息"
        )

        for field_name in self.fields:
            self.fields[field_name].required = False